root / trunk / gdhostc

#!/usr/bin/python
## -*- coding: utf-8 -*-
##
## gdhostc -- Gerf Dynamic Host Client
## 
## Copyright (C) 2001-2002,2005 Nick Borko & Christian H�e
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
## 
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
## 
## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##

import sys
import socket
import gdhost.client as client
from gdhost.version import version_string

## Version Check
if sys.version_info >= (2, 2) and \
   sys.version_info <  (2,3):
    try:
        import optik as optparse
    except ImportError:
        try:
            import optparse
        except ImportError:
            print >> sys.stderr, "You must either install optik (http://optik.sf.net) or upgrade to python 2.3"
elif sys.version_info >= (2,3):
    import optparse
else:
    print >> sys.stderr, "This program requires python 2.2 with the optik module (http://optik.sf.net) or python 2.3 or greater"
    sys.exit(1)

parser = optparse.OptionParser(version=version_string,
                               usage="%prog [options] command")
parser.add_option( '-k', '--key', action="store",
                        dest="key",
                        help="Encryption Key (required)")
parser.add_option( '-n', '--name', action="store",
                   dest="name", default= socket.gethostname(),
                   help="The dynamic host shortname" )
parser.add_option( '-i', '--ip', action="store",
                   dest='ip', default='auto',
                   help="Your IP address, 'auto' or 'server'. By default, this is auto.  'auto' means that your ip is discovered locally before talking to the server.  'server' means that the server will discover your ip. If you are behind a NAT firewall, 'server' will discover the firewall's ip address." )
parser.add_option( '-S', '--server', action="store",
                   dest="server", default="gerf.net",
                   help="The GDH server's dns name or ip address" )
parser.add_option( '-P', '--port', action="store",
                   dest="port", default=8888,
                   help="The GDH server's port number" )
parser.add_option( '-q', '--quiet', action="store_true",
                   dest="quiet", default=False,
                   help="Run quietly" )
parser.add_option( '-d', '--debug', action="store_true",
                   dest="debug", default=False,
                   help="Turn on debugging" )

if "__main__" == __name__:
    options, command = parser.parse_args()
    if command:
        command = command[0].strip()
        if command not in ('update','remove') :
            parser.error('Command must be one of update or remove; not %s' % \
                         repr(command))
            sys.exit(1)

    # Verify that ip is legit
    if options.ip not in ('auto','server') and \
       not options.ip.replace('.','').isdigit():
        
        parser.error("IP address must be an ip number, 'auto', or 'server'; not '%s'" % options.ip)

    # Helpful display
    if not command:
        parser.print_help()
        if options.ip == 'auto':
            ip = "discover locally"
        elif options.ip == 'server':
            ip = "server will discover the ip"
        else:
            ip = repr(options.ip)
        print 
        print "current values:"
        print "  KEY                = '%s'" % (options.key or '-required-')
        print "  NAME               = '%s'" % options.name
        print "  IP                 = %s" % ip
        print "  SERVER             = '%s'" % options.server
        print "  PORT               = '%s'" % options.port
        sys.exit(1)

    # Required values
    if not options.key:
        parser.error('You must set the encryption key (-k)')
        sys.exit(1)

    try:
        connection = client.Connection( server=options.server,
                                        port=options.port )
        if command == 'update':
            if options.ip == 'auto':
                ip = client.LOCAL_IP_DISCOVER
            elif options.ip == 'server':
                ip = client.REMOTE_IP_DISCOVER
            else:
                ip = options.ip
            connection.update( key=options.key,
                               name=options.name,
                               ip=ip )
        else:
            connection.delete( key=options.key,
                               name=options.name )
    except client.ErrorConnection,err:
        if options.debug:
            raise
        else:
            print >> sys.stderr, "Unable to communicate with %s:%s:\n\t%s" % (
                options.server, options.port, str(err))
            sys.exit(10)
    except client.ErrorDenied:
        if options.debug:
            raise
        else:
            print >> sys.stderr, "The server rejected your request"
            sys.exit(10)
    except client.ErrorNoName:
        if options.debug:
            raise
        else:
            print >> sys.stderr, "The server didn't recognize the name %s" % options.name
            sys.exit(10)
    except:
        if options.debug:
            raise
        print >> sys.stderr, "Unexpected Error!"
        if options.debug:
            print >> sys.stderr, "Plase file a bug at http://trac.gerf.org/gdhost with the above output"
        else:
            print >> sys.stderr, "Please rerun this command with --debug and file a bug at http://trac.gerf.org/gdhost"

    if not options.quiet:
        print "Successfully did a %s for %s!" % (command, options.name)
    sys.exit(0)