Initial commit.

.project

@@ -0,0 +1,17 @@
<?xml version="1.0" encoding="UTF-8"?>
<projectDescription>
	<name>pyvtr</name>
	<comment></comment>
	<projects>
	</projects>
	<buildSpec>
		<buildCommand>
			<name>org.python.pydev.PyDevBuilder</name>
			<arguments>
			</arguments>
		</buildCommand>
	</buildSpec>
	<natures>
		<nature>org.python.pydev.pythonNature</nature>
	</natures>
</projectDescription>

.pydevproject

@@ -0,0 +1,10 @@
<?xml version="1.0" encoding="UTF-8" standalone="no"?>
<?eclipse-pydev version="1.0"?>
<pydev_project>
<pydev_property name="org.python.pydev.PYTHON_PROJECT_INTERPRETER">Default</pydev_property>
<pydev_property name="org.python.pydev.PYTHON_PROJECT_VERSION">python 2.7</pydev_property>
<pydev_pathproperty name="org.python.pydev.PROJECT_SOURCE_PATH">
<path>/pyvtr/src</path>
</pydev_pathproperty>
</pydev_project>

src/pyvtr/Enum.py

@@ -0,0 +1,10 @@
'''
Created on Nov 8, 2009
@author: Alec Thomas
'''
def Enum(*sequential, **named):
    enums = dict(zip(sequential, range(len(sequential))), **named)
    return type('Enum', (), enums)
        

src/pyvtr/net/Hop.py

@@ -0,0 +1,21 @@
'''
Created on Jun 8, 2011
@author: BMAllred
'''
class Hop:
    '''
    Trace route hop.
    '''
    def __init__(self, address, roundTrip, hop):
        '''
        Initializes a new instance of the Hop class.
        '''
        
        self.Address = address
        self.RoundTrip = roundTrip
        self.HopCount = hop
        self.HostName = address

src/pyvtr/net/TraceRoute.py

@@ -0,0 +1,90 @@
'''
Created on Jun 8, 2011
@author: BMAllred
'''
import socket
from Hop import Hop
class TraceRoute:
    '''
    Trace route class.
    '''
    def __init__(self, destination, maxHops):
        '''
        Initializes a new instance of the TraceRoute class.
        '''
        
        self._maxHops = maxHops
        self._listenPort = 33434
        self.Destination = destination
        self.Hops = []
        
    def Execute(self):
        '''
        Executes the trace route operation.
        '''
        
        # Clear any previous results.
        self.Hops = []
        
        # Find the IP address of the host.
        destinationAddress = socket.gethostbyname(self.Destination)
        
        # Find the protocols.
        icmp = socket.getprotobyname('icmp')
        udp = socket.getprotobyname('udp')
        
        # Set the current hop count (this is relative to TTL).
        currentHop = 1
        
        # Set the operation flag.
        workInProgress = True
        
        while workInProgress:
            # Create sockets.
            receiveSocket = socket.socket(socket.AF_INET, socket.SOCK_RAW, icmp)
            sendSocket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, udp)
            
            # Bind sockets.
            receiveSocket.bind(("", self._listenPort))
            sendSocket.setsockopt(socket.SOL_IP, socket.IP_TTL, currentHop)
            sendSocket.sendto("", (destinationAddress, self._listenPort))
            
            # Create a new hop object.
            hop = Hop("*", 0, currentHop)
            
            try:
                # Return the first address (discard the data for now).
                # TODO: Retrieve round trip time.
                data = receiveSocket.recvfrom(512)
                hop.Address = data[0]
                
                # TODO: Remove after debugging.
                print data
                
                # Attempt to resolve the host name.
                try:
                    hop.HostName = socket.gethostbyaddr(hop.Address)[0]
                except socket.error:
                    hop.HostName = hop.Address
                
            except socket.error:
                pass
            finally:
                # Clean-up operations.
                sendSocket.close()
                receiveSocket.close()
            
            # Add the hop to the collection.
            self.Hops.append(hop)
            
            # Decide whether we need to continue.
            if hop.Address == destinationAddress or currentHop > self._maxHops:
                workInProgress = False
            else:
                currentHop += 1
        
        return self.Hops

src/pyvtr/pyvtr.py

@@ -0,0 +1,22 @@
#!/usr/bin/python
'''
Created on Jun 8, 2011
@author: BMAllred
'''
from net.TraceRoute import TraceRoute
from net.Hop import Hop
def traceRoute(address):
    # Initialize trace route object.
    trace = TraceRoute(address, 30)
    
    # Display all hops.
    for hop in trace.Execute():
        print hop.HopCount + ": " + hop.Address
if __name__ == '__main__':
    traceRoute("www.google.com")
    pass

src/pyvtr/text/BlockType.py

@@ -0,0 +1,3 @@
from pyvtr.Enum import Enum
BlockType = Enum(Plain=0, Route=1, Hop=2)

src/pyvtr/text/Parser.py

@@ -0,0 +1,61 @@
'''
Created on Jun 8, 2011
@author: BMAllred
'''
from pyvtr.text.TextBlock import TextBlock
from pyvtr.text.BlockType import BlockType
class Parser:
    '''
    Text parser for writing output to various data structures.
    '''
    def __init__(self, templateFile):
        '''
        Initializes a new instance of the Parser class.
        '''
        
        self._Blocks = []
        self.TemplateFile = templateFile
        
    def ReadTemplate(self):
        self._Blocks = []
        
        with open(self.TemplateFile) as file:
            newBlockContext = ""
            
            # Read the next line in the file.
            for line in file.readlines():
            
                # Check for any parsing to be done.
                if line.upper() == "{ROUTES}":
                    self._Blocks.append(TextBlock(newBlockContext))
                    newBlockContext = ""
                
                elif line.upper() == "{/ROUTES}":
                    idx = line.find("{/ROUTES}")
                    trashLength = idx + 9
                    postText = line[trashLength:len(line)]
                    self._Blocks.append(TextBlock(newBlockContext, BlockType.Route, postText))
                    newBlockContext = ""
                
                elif line.upper() == "{HOPS}":
                    self._Blocks.append(TextBlock(newBlockContext, BlockType.Route))
                    newBlockContext = ""
                
                elif line.upper() == "{/HOPS}":
                    idx = line.find("{/HOPS}")
                    trashLength = idx + 7
                    postText = line[trashLength:len(line)]
                    self._Blocks[-1].InnerBlocks.append(TextBlock(newBlockContext, BlockType.Hop, postText))
                    newBlockContext = ""
                
                else:
                    newBlockContext += line
                
                # If there is content then we want to add it to the collection then clear it.
                if len(newBlockContext) > 0:
                    self._Blocks.append(TextBlock(newBlockContext))
                    newBlockContext = ""
            

src/pyvtr/text/TextBlock.py

@@ -0,0 +1,21 @@
'''
Created on Jun 8, 2011
@author: BMAllred
'''
from pyvtr.text.BlockType import BlockType
class TextBlock:
    '''
    Text block class.
    '''
    def __init__(self, text, blockType = BlockType.Plain, postText = ""):
        '''
        Initializes a new instance of the TextBlock class.
        '''
        
        self.InnerBlocks = []
        self.Text = text
        self.Type = blockType
        self.PostText = postText