forked from openwpm/OpenWPM
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Initial commit of socket interface. Tested and working with proxy
- Loading branch information
1 parent
4f12fcb
commit f2379dd
Showing
8 changed files
with
148 additions
and
34 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,9 @@ | ||
| *.pyc | ||
| *~ | ||
| .*.sw* | ||
|
|
||
| # A bug in selenium creates this on unix systems | ||
| C:\\nppdf32Log\\debuglog.txt | ||
|
|
||
| #MAC filesystem explorer drops this | ||
| .DS_Store |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,97 @@ | ||
| import Queue | ||
| import threading | ||
| import socket | ||
| import struct | ||
| import cPickle | ||
|
|
||
| #TODO - Implement a kill command for the server socket | ||
| #TODO - Add in thread joins and make sure everything exits cleanly | ||
|
|
||
| class serversocket: | ||
| ''' | ||
| A server socket to recieve and process string messages | ||
| from client sockets to a central queue | ||
| ''' | ||
| def __init__(self, verbose=False): | ||
| self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | ||
| self.sock.bind(('localhost',0)) | ||
| self.sock.listen(10) #queue a max of n connect requests | ||
| self.verbose = verbose | ||
| self.queue = Queue.Queue() | ||
| if self.verbose: | ||
| print "Server bound to: " + str(self.sock.getsockname()) | ||
|
|
||
| def start_accepting(self): | ||
| ''' Start the listener thread ''' | ||
| thread = threading.Thread(target=self._accept, args=()) | ||
| thread.start() | ||
|
|
||
| def _accept(self): | ||
| ''' Listen for connections and pass handling to a new thread ''' | ||
| while True: | ||
| (client, address) = self.sock.accept() | ||
| thread = threading.Thread(target=self._handle_conn, args=(client,address)) | ||
| thread.start() | ||
|
|
||
| def _handle_conn(self, client, address): | ||
| ''' | ||
| Recieve messages and pass to queue. Messages are prefixed with | ||
| a 4-byte integer to specify the message length. | ||
| ''' | ||
| if self.verbose: | ||
| print "Thread: " + str(threading.current_thread()) + " connected to: " + str(address) | ||
| try: | ||
| while True: | ||
| msg = self.receive_msg(client, 5) | ||
| msglen, is_pickled = struct.unpack('>I?', msg) | ||
| msg = self.receive_msg(client, msglen) | ||
| if is_pickled: | ||
| msg = cPickle.loads(msg) | ||
| self.queue.put(msg) | ||
| except RuntimeError: | ||
| if self.verbose: | ||
| print "Client socket: " + str(address) + " closed" | ||
|
|
||
| def receive_msg(self, client, msglen): | ||
| msg = '' | ||
| while len(msg) < msglen: | ||
| chunk = client.recv(msglen-len(msg)) | ||
| if chunk == '': | ||
| raise RuntimeError("socket connection broken") | ||
| msg = msg + chunk | ||
| return msg | ||
|
|
||
| class clientsocket: | ||
| def __init__(self): | ||
| self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | ||
|
|
||
| def connect(self, host, port): | ||
| self.sock.connect((host,port)) | ||
|
|
||
| def send(self, msg, is_pickled=False): | ||
| #prepend with message length | ||
| msg = struct.pack('>I?', len(msg), is_pickled) + msg | ||
| totalsent = 0 | ||
| while totalsent < len(msg): | ||
| sent = self.sock.send(msg[totalsent:]) | ||
| if sent == 0: | ||
| raise RuntimeError("socket connection broken") | ||
| totalsent = totalsent + sent | ||
|
|
||
| def send_pickled(self, obj): | ||
| pickled = cPickle.dumps(obj) | ||
| self.send(pickled,is_pickled=True) | ||
|
|
||
| def close(self): | ||
| self.sock.close() | ||
|
|
||
| if __name__ == '__main__': | ||
| import sys | ||
|
|
||
| #Just for testing | ||
| if sys.argv[1] == 's': | ||
| sock = serversocket(verbose=True) | ||
| sock.start_accepting() | ||
| elif sys.argv[1] == 'c': | ||
| sock = clientsocket() | ||
| import ipdb; ipdb.set_trace() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters