185 lines
6.5 KiB
Python
Executable File
185 lines
6.5 KiB
Python
Executable File
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import base64
|
|
import collections
|
|
import logging
|
|
import math
|
|
import sys
|
|
import threading
|
|
import time
|
|
import tweepy
|
|
|
|
sys.path.append("../../../")
|
|
|
|
from conf import Conf
|
|
from ether2any import Ether2Any
|
|
|
|
class TwittStreamHandler(tweepy.StreamListener):
|
|
def __init__(self, dev, coder, debug=False):
|
|
super(TwittStreamHandler, self).__init__()
|
|
self.dev = dev
|
|
self.fragments = collections.defaultdict(str)
|
|
self.debug = debug
|
|
self.coder = coder
|
|
|
|
def on_status( self, status ):
|
|
""" On statis, decode and reassemble packet-status-texts. If complete, write them to the tun-dev. """
|
|
|
|
sourcePacket = status.text
|
|
if self.debug:
|
|
print "in uni:", repr(sourcePacket)
|
|
(isFragment, packetLen, packetId, packet) = None, None, None, None
|
|
try:
|
|
(isFragment, packetLen, packetId, packet) = self.coder.decode(sourcePacket)
|
|
except ValueError, e:
|
|
print "Could not decode tweet, omitting (Error was: %s).\n\tText was: %s" % (e, repr(sourcePacket))
|
|
raise
|
|
return
|
|
if isFragment:
|
|
self.fragments[packetId] += packet
|
|
print " >+ Added fragment with id", packetId
|
|
else:
|
|
toSend = None
|
|
if self.fragments.has_key(packetId):
|
|
toSend = self.fragments[packetId] + packet
|
|
else:
|
|
toSend = packet
|
|
if toSend == '':
|
|
print " >! Received packet with id", packetId, "is BROKEN"
|
|
else:
|
|
print " >> Received packet with id", packetId
|
|
if self.debug:
|
|
print repr(toSend)
|
|
self.dev.write(toSend)
|
|
|
|
def on_limit(self, track):
|
|
print "We got limited ", track
|
|
print "At the moment there is no error-handling for this, so we just kill everything. Remember: This software doesn't even deserve the label 'alpha' ;)"
|
|
sys.exit(1)
|
|
|
|
def on_error(self, status_code):
|
|
print "We got an error code: ", status_code
|
|
if status_code == 401:
|
|
print "Better check 'yer twitter-credentials."
|
|
sys.exit(2)
|
|
else:
|
|
print "At the moment there is no error-handling for this, so we just kill everything. Remember: This software doesn't even deserve the label 'alpha' ;)"
|
|
sys.exit(1)
|
|
|
|
def on_timeout(self, status_code):
|
|
print "Got an timeout: ", status_code
|
|
print "At the moment there is no error-handling for this, so we just kill everything. Remember: This software doesn't even deserve the label 'alpha' ;)"
|
|
sys.exit(1)
|
|
|
|
# TODO: Thread is not needed, tweepy has its own threading. remove it
|
|
class DownstreamThread(threading.Thread):
|
|
def __init__(self, dev, coder, auth, endpoint, debug=False):
|
|
threading.Thread.__init__(self)
|
|
self.debug = debug
|
|
self.auth = auth
|
|
self.coder = coder
|
|
self.api = tweepy.API(self.auth)
|
|
self.endpoint = endpoint
|
|
self.dev = dev
|
|
self.daemon = True
|
|
|
|
def run(self):
|
|
stream = tweepy.Stream(auth=self.auth, listener=TwittStreamHandler(self.dev, self.coder, self.debug))
|
|
user = self.api.get_user(self.endpoint)
|
|
print "Endpoint is", self.endpoint, "with id", user.id
|
|
stream.filter([user.id])
|
|
|
|
class RFC1149(Ether2Any):
|
|
def __init__(self, debug=False):
|
|
Ether2Any.__init__(self, tap=False)
|
|
|
|
self.debug = debug
|
|
network = Conf.get("network", {'mtu': 1400})
|
|
self.coder = Conf.get("coder", None)
|
|
self.twitterConf = Conf.get("twitter", None)
|
|
self.endpoint = self.twitterConf['endpoint']
|
|
if not self.endpoint:
|
|
print "No endpoint in configuration, please add one."
|
|
sys.exit(1)
|
|
self.dev.ifconfig(**network)
|
|
self.dev.up()
|
|
|
|
self._setupTwitter()
|
|
self.downstream = DownstreamThread(dev=self.dev, coder=self.coder, auth=self.auth, endpoint=self.endpoint, debug=self.debug)
|
|
self.downstream.start()
|
|
|
|
def _requestTwitterTokens(self):
|
|
auth = tweepy.OAuthHandler(self._dec(self.ck), self._dec(self.cs))
|
|
pass
|
|
|
|
def _setupTwitter(self):
|
|
ck = [17, 39, 65, 39, 25, 22, 38, 30, 20, 38, 33, 69, 27, 0, 61,
|
|
31, 61, 34, 42, 32]
|
|
cs = [71, 37, 71, 37, 57, 36, 18, 2, 59, 64, 63, 58, 23, 61, 74,
|
|
30, 34, 68, 38, 33, 56, 1, 3, 74, 29, 9, 41, 32, 33, 7, 52,
|
|
8, 70, 20, 30, 25, 38, 51, 57, 66, 53, 0, 42]
|
|
self.auth = tweepy.OAuthHandler(self._dec(ck), self._dec(cs))
|
|
if not self.twitterConf['ACCESS_KEY']:
|
|
# request tokens, get access, write tokens down.
|
|
auth_url = self.auth.get_authorization_url()
|
|
print "We have no access token for a twitter account. Please visit the"
|
|
print "url printed down below, login and report back with the PIN."
|
|
print
|
|
print " Authorization URL: %s" % auth_url
|
|
print
|
|
verifier = raw_input('PIN: ').strip()
|
|
self.auth.get_access_token(verifier)
|
|
self.twitterConf['ACCESS_KEY'] = self.auth.access_token.key
|
|
self.twitterConf['ACCESS_SECRET'] = self.auth.access_token.secret
|
|
authConf = open("conf_auth.py", "w")
|
|
authConf.write("""from conf import Conf
|
|
|
|
# WARNING: This config was overwritten! If you change it be sure
|
|
# that you know, what you are doing.
|
|
if not Conf['twitter']['ACCESS_KEY']:
|
|
Conf['twitter']['ACCESS_KEY'] = "%s"
|
|
Conf['twitter']['ACCESS_SECRET'] = "%s"
|
|
""" % (self.twitterConf['ACCESS_KEY'], self.twitterConf['ACCESS_SECRET']))
|
|
authConf.close()
|
|
self.auth.set_access_token(self.twitterConf['ACCESS_KEY'], self.twitterConf['ACCESS_SECRET'])
|
|
self.api = tweepy.API(self.auth)
|
|
self.me = self.api.me()
|
|
print "Logged in as %s" % (self.me.screen_name,)
|
|
|
|
def _dec(self, s):
|
|
""" ... """
|
|
return "".join(map(lambda x: chr(x+48), reversed(s)))
|
|
|
|
def sendToNet(self, packet):
|
|
fragments = self.coder.encode(packet)
|
|
if self.debug:
|
|
print "out raw:", repr(packet)
|
|
print "out frag:", repr(fragments)
|
|
print " >> Sending out %d bytes in %d tweet%s" % (len(packet), len(fragments), len(fragments)!=1 and "s" or "")
|
|
for fragment in fragments:
|
|
try:
|
|
self.api.update_status(fragment)
|
|
except tweepy.error.TweepError, e:
|
|
# TODO: Proper handling of
|
|
# - over 140 chars limit (bug in this software)
|
|
# - connection refused (retransmit)
|
|
# - limit reached (terminate or wait)
|
|
print " >! ERROR - Either connection refused or limited. Not sent."
|
|
print " >! ERROR - Either connection refused or limited. Not sent."
|
|
print " >! ERROR - Either connection refused or limited. Not sent."
|
|
print " >! ERROR - Either connection refused or limited. Not sent."
|
|
print "YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYY"
|
|
print repr(fragment)
|
|
print "YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYY"
|
|
print e
|
|
print " >! ERROR - Either connection refused or limited. Not sent."
|
|
print " >! ERROR - Either connection refused or limited. Not sent."
|
|
print " >! ERROR - Either connection refused or limited. Not sent."
|
|
|
|
if __name__ == '__main__':
|
|
rfc = RFC1149(debug=True)
|
|
print "Starting RFC1149 ip-over-twitter service..."
|
|
rfc.run()
|
|
|