source4/scripting/swig/rpcclient

   1 #!/usr/bin/python
   2
   3 import sys, os, string
   4 from cmd import Cmd
   5 from optparse import OptionParser
   6
   7
   8 import dcerpc, samr
   9
  10 class rpcclient(Cmd):
  11
  12     prompt = 'rpcclient$ '
  13
  14     def __init__(self, binding, domain, username, password):
  15         Cmd.__init__(self)
  16         self.binding = binding
  17         self.domain = domain
  18         self.username = username
  19         self.password = password
  20
  21     def emptyline(self):
  22
  23         # Default for empty line is to repeat last command - yuck
  24
  25         pass
  26
  27     def onecmd(self, line):
  28
  29         # Override the onecmd() method so we can trap error returns
  30
  31         try:
  32             Cmd.onecmd(self, line)
  33         except dcerpc.NTSTATUS, arg:
  34             print 'The command returned an error: %s' % arg[1]
  35
  36     # Command handlers
  37
  38     def do_help(self, line):
  39         """Displays on-line help for rpcclient commands."""
  40         Cmd.do_help(self, line)
  41
  42     def do_shell(self, line):
  43
  44         status = os.system(line)
  45
  46         if os.WIFEXITED(status):
  47             if os.WEXITSTATUS(status) != 0:
  48                 print 'Command exited with code %d' % os.WEXITSTATUS(status)
  49         else:
  50             print 'Command exited with signal %d' % os.WTERMSIG(status)
  51
  52     def do_EOF(self, line):
  53         """Exits rpcclient."""
  54         print
  55         sys.exit(0)
  56
  57     # SAMR pipe commands
  58
  59     def do_SamrEnumDomains(self, line):
  60         """Enumerate domain names."""
  61
  62         usage = 'usage: SamrEnumDomains'
  63
  64         if line != '':
  65             print usage
  66             return
  67
  68         pipe = dcerpc.pipe_connect(
  69             self.binding,
  70             dcerpc.DCERPC_SAMR_UUID, dcerpc.DCERPC_SAMR_VERSION,
  71             self.domain, self.username, self.password)
  72
  73         connect_handle = samr.Connect(pipe)
  74
  75         for i in connect_handle.EnumDomains():
  76             print i
  77
  78     def do_SamrLookupDomain(self, line):
  79         """Return the SID for a domain."""
  80
  81         usage = 'SamrLookupDomain DOMAIN'
  82
  83         parser = OptionParser(usage)
  84         options, args = parser.parse_args(string.split(line))
  85
  86         if len(args) != 1:
  87             print 'usage:', usage
  88             return
  89
  90         pipe = dcerpc.pipe_connect(
  91             self.binding,
  92             dcerpc.DCERPC_SAMR_UUID, dcerpc.DCERPC_SAMR_VERSION,
  93             self.domain, self.username, self.password)
  94
  95         connect_handle = samr.Connect(pipe)
  96
  97         print connect_handle.LookupDomain(args[0])
  98
  99 if __name__ == '__main__':
 100
 101     # Parse command line
 102
 103     usage = 'rpcclient BINDING [options]'
 104
 105     if len(sys.argv) == 1:
 106         print usage
 107         sys.exit(1)
 108
 109     binding = sys.argv[1]
 110     del(sys.argv[1])
 111
 112     if string.find(binding, ':') == -1:
 113         binding = 'ncacn_np:' + binding
 114
 115     parser = OptionParser(usage)
 116
 117     parser.add_option('-U', '--username', action='store', type='string',
 118                       help='Use given credentials when connecting',
 119                       metavar='DOMAIN\\username%password',
 120                       dest='username')
 121
 122     parser.add_option('-c', '--command', action='store', type='string',
 123                       help='Execute COMMAND', dest='command')
 124
 125     options, args = parser.parse_args()
 126
 127     # Break --username up into domain, usernamd and password
 128
 129     if not options.username:
 130         options.username = '%'
 131
 132     domain = ''
 133     if string.find(options.username, '\\') != -1:
 134         domain, options.username = string.split(options.username, '\\')
 135
 136     password = ''
 137     if string.find(options.username, '%') != -1:
 138         options.username, password  = string.split(options.username, '%')
 139
 140     username = options.username
 141
 142     # Run command loop
 143
 144     c = rpcclient(binding, domain, username, password)
 145
 146     if options.command:
 147         c.onecmd(options.command)
 148         sys.exit(0)
 149
 150     while 1:
 151         try:
 152             c.cmdloop()
 153         except KeyboardInterrupt:
 154             print 'KeyboardInterrupt'