# Server code
import SimpleXMLRPCServer
import string,base64

class AuthentificateXMLRPCRequestHandler(SimpleXMLRPCServer.SimpleXMLRPCRequestHandler):
    def do_POST(self):
        try:
            header =  self.headers['Authorization']
            type, user_passwd = header.split()
            username, password = string.split(base64.decodestring(user_passwd), ':')
            if self.testUser(username,password):
                SimpleXMLRPCServer.SimpleXMLRPCRequestHandler.do_POST(self)
            else:
                self.report_error(401)  
        except:
            self.report_error(401)

    def report_error (self,code):
        '''
        Send back an errorcode
        '''
        #it is important to read out the complete sended request , 
        # but throw the data away, because an error should be send back
        try:
            max_chunk_size = 10*1024*1024
            size_remaining = int(self.headers["content-length"])
            while size_remaining:
                chunk_size = min(size_remaining, max_chunk_size)
                size_remaining -= len(self.rfile.read(chunk_size))
        except:
            pass

        #now just send the error back
        special_errors={401:self.report_401, 
                                    404:self.report_404}
        if special_errors.has_key(code):
             special_errors[code]()
        else:
            self.send_response(code)
            self.end_headers()
            self.connection.shutdown(1)
    
    def report_401(self):
        self.send_response(401)
        self.send_header("WWW-Authenticate", 'Basic realm="%s"'% self.server.relam)
        response = 'Unauthorised'
        self.send_header("Content-type", "text/plain")
        self.send_header("Content-length", str(len(response)))
        self.end_headers()
        self.wfile.write(response)
        # shut down the connection
        self.wfile.flush()
        self.connection.shutdown(1)
        
    def testUser(self,username,password):
        """
        Function for testing authentification
        """
        if username=="test" and password=="test":
                return True

        return False



def test():
    server = SimpleXMLRPCServer.SimpleXMLRPCServer(("localhost", 8000),AuthentificateXMLRPCRequestHandler)
    server.relam="xmlrpc"
    server.register_introspection_functions()
    server.register_function(lambda x: x*x, 'square')
    server.serve_forever()

if __name__ == '__main__':
    test()