diff --git a/tools/pamusb-conf b/tools/pamusb-conf index 47954b9..aaae15e 100755 --- a/tools/pamusb-conf +++ b/tools/pamusb-conf @@ -22,178 +22,178 @@ import os from xml.dom import minidom class Device: - def __init__(self, udi): - self.__udi = udi - self.__findStorageDevice() - deviceObj = bus.get_object('org.freedesktop.Hal', - udi) - deviceProperties = deviceObj.GetAllProperties( - dbus_interface = 'org.freedesktop.Hal.Device') - self.vendor = deviceProperties['usb_device.vendor'] - self.product = deviceProperties['info.product'] - self.serialNumber = deviceProperties['usb_device.serial'] - - def __findStorageDevice(self): - for child in halManager.FindDeviceByCapability('storage'): - obj = bus.get_object('org.freedesktop.Hal', - child) - properties = obj.GetAllProperties( - dbus_interface = 'org.freedesktop.Hal.Device') - if properties['storage.physical_device'] == self.__udi + '_if0': - self.__storageUdi = child - return - raise Exception, '%s is not a storage device.' % self.__udi - - def __repr__(self): - return "%s %s (%s)" % (self.vendor, self.product, self.serialNumber) - - def volumes(self): - vols = [] - for volume in halManager.FindDeviceByCapability('volume'): - deviceObj = bus.get_object('org.freedesktop.Hal', - volume) - deviceProperties = deviceObj.GetAllProperties( - dbus_interface = 'org.freedesktop.Hal.Device') - if deviceProperties['block.storage_device'] != self.__storageUdi: - continue - vols.append({'uuid' : deviceProperties['volume.uuid'], - 'device' : deviceProperties['block.device']}) - return vols + def __init__(self, udi): + self.__udi = udi + self.__findStorageDevice() + deviceObj = bus.get_object('org.freedesktop.Hal', + udi) + deviceProperties = deviceObj.GetAllProperties( + dbus_interface = 'org.freedesktop.Hal.Device') + self.vendor = deviceProperties['usb_device.vendor'] + self.product = deviceProperties['info.product'] + self.serialNumber = deviceProperties['usb_device.serial'] + + def __findStorageDevice(self): + for child in halManager.FindDeviceByCapability('storage'): + obj = bus.get_object('org.freedesktop.Hal', + child) + properties = obj.GetAllProperties( + dbus_interface = 'org.freedesktop.Hal.Device') + if properties['storage.physical_device'] == self.__udi + '_if0': + self.__storageUdi = child + return + raise Exception, '%s is not a storage device.' % self.__udi + + def __repr__(self): + return "%s %s (%s)" % (self.vendor, self.product, self.serialNumber) + + def volumes(self): + vols = [] + for volume in halManager.FindDeviceByCapability('volume'): + deviceObj = bus.get_object('org.freedesktop.Hal', + volume) + deviceProperties = deviceObj.GetAllProperties( + dbus_interface = 'org.freedesktop.Hal.Device') + if deviceProperties['block.storage_device'] != self.__storageUdi: + continue + vols.append({'uuid' : deviceProperties['volume.uuid'], + 'device' : deviceProperties['block.device']}) + return vols def listOptions(question, options, force = False): - if force == False and len(options) == 1: - return 0 - while True: - try: - print question - for i in range(len(options)): - print "%d) %s" % (i, options[i]) - print - sys.stdout.write('[%s-%s]: ' % (0, len(options) - 1)) - optionId = int(sys.stdin.readline()) - print - if optionId not in range(len(options)): - raise Exception - return optionId - except KeyboardInterrupt: sys.exit() - except Exception: pass - else: break + if force == False and len(options) == 1: + return 0 + while True: + try: + print question + for i in range(len(options)): + print "%d) %s" % (i, options[i]) + print + sys.stdout.write('[%s-%s]: ' % (0, len(options) - 1)) + optionId = int(sys.stdin.readline()) + print + if optionId not in range(len(options)): + raise Exception + return optionId + except KeyboardInterrupt: sys.exit() + except Exception: pass + else: break def addDevice(options): - devices = [] - - for udi in halManager.FindDeviceStringMatch('info.bus', 'usb_device'): - try: - devices.append(Device(udi)) - except Exception, ex: - pass - - if len(devices) == 0: - print 'No devices detected.' - sys.exit() - - device = devices[listOptions("Please select the device you wish to add.", - devices, force = options['force'])] - volumes = device.volumes() - volume = volumes[listOptions("Which volume would you like to use for " \ - "storing data ?", - ["%s (UUID: %s)" % (volume['device'], - volume['uuid']) - for volume in volumes], - force = options['force'])] - - print 'Name\t\t: %s' % options['deviceName'] - print 'Vendor\t\t: %s' % device.vendor - print 'Model\t\t: %s' % device.product - print 'Serial\t\t: %s' % device.serialNumber - if volume['uuid'] != '': - print 'Volume UUID\t: %s (%s)' % (volume['uuid'], volume['device']) - else: - print - print 'WARNING: No UUID detected for device %s. One time pads will be disabled.' % volume['device'] - print - print 'Save device to %s ?' % options['configFile'] - - sys.stdout.write('[y/n] ') - if sys.stdin.readline().strip() != 'y': - sys.exit(1) - - try: - doc = minidom.parse(options['configFile']) - except Exception, err: - print 'Unable to read %s: %s' % (options['configFile'], err) - sys.exit(1) - - devs = doc.getElementsByTagName('devices') - dev = doc.createElement('device') - dev.attributes['id'] = options['deviceName'] - devs[0].appendChild(dev) - - for name, value in (('vendor', device.vendor), - ('model', device.product), - ('serial', device.serialNumber), - ('volume_uuid', volume['uuid'])): - e = doc.createElement(name) - t = doc.createTextNode(value) - e.appendChild(t) - dev.appendChild(e) - - # Disable one time pads if there's no device UUID - if volume['uuid'] == '': - e = doc.createElement('option') - e.setAttribute('name', 'one_time_pad') - e.appendChild(doc.createTextNode('false')) - dev.appendChild(e) - - try: - f = open(options['configFile'], 'w') - f.write(doc.toxml()) - f.close() - except Exception, err: - print 'Unable to save %s: %s' % (options['configFile'], err) - sys.exit(1) - else: - print 'Done.' + devices = [] + + for udi in halManager.FindDeviceStringMatch('info.bus', 'usb_device'): + try: + devices.append(Device(udi)) + except Exception, ex: + pass + + if len(devices) == 0: + print 'No devices detected.' + sys.exit() + + device = devices[listOptions("Please select the device you wish to add.", + devices, force = options['force'])] + volumes = device.volumes() + volume = volumes[listOptions("Which volume would you like to use for " \ + "storing data ?", + ["%s (UUID: %s)" % (volume['device'], + volume['uuid']) + for volume in volumes], + force = options['force'])] + + print 'Name\t\t: %s' % options['deviceName'] + print 'Vendor\t\t: %s' % device.vendor + print 'Model\t\t: %s' % device.product + print 'Serial\t\t: %s' % device.serialNumber + if volume['uuid'] != '': + print 'Volume UUID\t: %s (%s)' % (volume['uuid'], volume['device']) + else: + print + print 'WARNING: No UUID detected for device %s. One time pads will be disabled.' % volume['device'] + print + print 'Save device to %s ?' % options['configFile'] + + sys.stdout.write('[y/n] ') + if sys.stdin.readline().strip() != 'y': + sys.exit(1) + + try: + doc = minidom.parse(options['configFile']) + except Exception, err: + print 'Unable to read %s: %s' % (options['configFile'], err) + sys.exit(1) + + devs = doc.getElementsByTagName('devices') + dev = doc.createElement('device') + dev.attributes['id'] = options['deviceName'] + devs[0].appendChild(dev) + + for name, value in (('vendor', device.vendor), + ('model', device.product), + ('serial', device.serialNumber), + ('volume_uuid', volume['uuid'])): + e = doc.createElement(name) + t = doc.createTextNode(value) + e.appendChild(t) + dev.appendChild(e) + + # Disable one time pads if there's no device UUID + if volume['uuid'] == '': + e = doc.createElement('option') + e.setAttribute('name', 'one_time_pad') + e.appendChild(doc.createTextNode('false')) + dev.appendChild(e) + + try: + f = open(options['configFile'], 'w') + f.write(doc.toxml()) + f.close() + except Exception, err: + print 'Unable to save %s: %s' % (options['configFile'], err) + sys.exit(1) + else: + print 'Done.' def usage(): - print 'Usage: %s [--config file] --add-device [--no-autodetect]' % os.path.basename(__file__) - sys.exit(1) + print 'Usage: %s [--config file] --add-device [--no-autodetect]' % os.path.basename(__file__) + sys.exit(1) import getopt try: - opts, args = getopt.getopt(sys.argv[1:], "ha:nc:", - ["help", "add-device=", "no-autodetect", - "config="]) + opts, args = getopt.getopt(sys.argv[1:], "ha:nc:", + ["help", "add-device=", "no-autodetect", + "config="]) except getopt.GetoptError: - usage() + usage() if len(args) != 0: - usage() + usage() options = { 'force' : False, 'deviceName' : None, - 'configFile' : '/etc/pamusb.conf' } + 'configFile' : '/etc/pamusb.conf' } for o, a in opts: - if o in ("-h", "--help"): - usage() - if o in ("-a", "--add-device"): - options['deviceName'] = a - if o in ("-n", "--no-autodetect"): - options['force'] = True - if o in ("-c", "--config"): - options['configFile'] = a + if o in ("-h", "--help"): + usage() + if o in ("-a", "--add-device"): + options['deviceName'] = a + if o in ("-n", "--no-autodetect"): + options['force'] = True + if o in ("-c", "--config"): + options['configFile'] = a if options['deviceName'] is None: - usage() + usage() bus = dbus.SystemBus() halService = bus.get_object('org.freedesktop.Hal', - '/org/freedesktop/Hal/Manager') + '/org/freedesktop/Hal/Manager') halManager = dbus.Interface(halService, 'org.freedesktop.Hal.Manager') try: - addDevice(options) + addDevice(options) except KeyboardInterrupt: - sys.exit(1) + sys.exit(1)