tcp_client.ipynb

{
 "metadata": {
  "name": ""
 },
 "nbformat": 3,
 "nbformat_minor": 0,
 "worksheets": [
  {
   "cells": [
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "import socket\n",
      "import sys\n",
      "import numpy as np\n",
      "import struct\n",
      "import scipy\n",
      "import time\n",
      "import datetime\n",
      "\n",
      "#########################################################\n",
      "#Sets up a tcp_client to receive data from USRP tcp_sink \n",
      "#and dump data to file\n",
      "#########################################################\n",
      "\n",
      "###GLOBALS\n",
      "file_size = 9999999 #Number of data points in single traces\n",
      "no_of_files = 100000000 #Number of traces to be taken\n",
      "\n",
      "# Create a TCP/IP socket\n",
      "sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)\n",
      "\n",
      "# Bind the socket to the port\n",
      "server_address = ('127.0.0.1', 10000)\n",
      "print >>sys.stderr, 'starting up on %s port %s' % server_address\n",
      "sock.bind(server_address)\n",
      "\n",
      "# Listen for incoming connections\n",
      "sock.listen(1)\n",
      "\n",
      "#Set client address and port number\n",
      "client_address=('127.0.0.1',10000)\n",
      "\n",
      "while True:\n",
      "    # Wait for a connection\n",
      "\tprint >>sys.stderr, 'waiting for a connection'\n",
      "\tconnection, client_address = sock.accept()\n",
      "\ttry:\n",
      "        print >>sys.stderr, 'connection from', client_address\n",
      "\t\tnum_of_points = 0  #counter for no. of points\n",
      "\t\tfile_no = 0 #counter for no. of files\n",
      "        \n",
      "        \n",
      "        #Name data dump file\n",
      "        fname=\"Trace_0000\"\n",
      "        \n",
      "        ##########################################################\n",
      "        #Uncheck these lines for timestamping dump files\n",
      "        ##########################################################\n",
      "#       st=time.mktime(datetime.datetime.now().timetuple())\n",
      "# \t\tfname='./Trace_'+str(st)\n",
      "\n",
      "\t\tf = open(fname, 'a')\n",
      "\t        # Receive the data in small chunks and retransmit it\n",
      "\t    while True:\n",
      "            data = connection.recv(8);\n",
      "            f.write(data)\t\t    \n",
      "            num_of_points += 1\n",
      "            if(num_of_points == file_size):\n",
      "                f.close()\n",
      "\t\t\t\tprint \"Done.: \"+fname\n",
      "\t\t\t\tif(file_no == no_of_files):\n",
      "\t\t\t\t\tbreak\n",
      "\t\t\t\tnum_of_points = 0\n",
      "\t\t\t\tfile_no += 1\n",
      "                \n",
      "                #New dump file name\n",
      "\t\t\t\tfname=\"Trace_0000\"\n",
      "        \n",
      "                ##########################################################\n",
      "                #Uncheck these lines for timestamping dump files\n",
      "                ##########################################################\n",
      "        #       st=time.mktime(datetime.datetime.now().timetuple())\n",
      "        # \t\tfname='./Trace_'+str(st)\n",
      "\t\t\t\t\n",
      "                f = open(fname, 'a')\n",
      "\t\t\t\tprint \"Opening new file : \"+fname\n",
      "                    \n",
      "\t\tprint \"Done creating files..\"\n",
      "\tfinally:\n",
      "        \t# Clean up the connection\n",
      "        \tconnection.close()\n"
     ],
     "language": "python",
     "metadata": {},
     "outputs": []
    }
   ],
   "metadata": {}
  }
 ]
}