NAME = 'de.cryptobitch.muelli.Pwnitter'
class Pwnitter(dbus.service.Object):
- def __init__(self, bus, object_name, device='mon0'):
+ def __init__(self, bus, object_name, device='mon0', run_once_only=False):
super(Pwnitter, self).__init__(bus, object_name)
self.device = device
self.status = status
self.is_running = False
+ self.run_once_only = run_once_only
def setup_monitor(device='mon0'):
# FIXME: Replace hardcoded interface
log.debug('Issueing connection')
+ if self.run_once_only: # If we wanted to run once only, we make it stop now
+ self.is_running = False
if host == 'mobile.twitter.com':
params = urllib.urlencode({
log.error("FAILED to tweet as %s, debug follows:", name)
log.error("%s, %s", response.status, response.reason)
log.error("%s", response.read())
- return self.is_running # Execute next time, we're idle
+
+ return self.is_running # Execute next time, we're idle or stop if we wanted to run once and have processed a message successfully
# FIXME: Ideally, check whether Pcap has got data for us
def main():
parser.add_option("-s", "--session", dest="use_session_bus",
action="store_true", default=False,
help="Bind Pwnitter to the SessionBus instead of the SystemBus")
+ parser.add_option("-1", "--single", dest="run_once_only",
+ action="store_true", default=False,
+ help="Make it send a single message only")
(options, args) = parser.parse_args()
loglevel = {'debug': logging.DEBUG, 'info': logging.INFO,
'warn': logging.WARN, 'error': logging.ERROR,
else:
session_bus = dbus.SystemBus()
name = dbus.service.BusName(NAME, session_bus)
- pwnitter = Pwnitter(session_bus, '/Pwnitter')
+ pwnitter = Pwnitter(session_bus, '/Pwnitter', run_once_only=options.run_once_only)
#object.Start()
loop = gobject.MainLoop()