Something that is hopefully a DHCP server package
[invirt/packages/invirt-dhcp.git] / code / pydhcplib / scripts / pydhcp
1 #!/usr/bin/python
2
3
4 from pydhcplib import dhcp_constants
5 from pydhcplib import dhcp_packet
6 from pydhcplib import dhcp_network
7
8 from pydhcplib import type_hw_addr
9 from pydhcplib import type_ipv4
10 from pydhcplib import type_strlist
11
12 import sys
13
14 from optparse import OptionParser
15
16
17 parser = OptionParser()
18
19 """ Action options """
20 parser.add_option("-L", "--listen", action="store_true",dest="listen", help="",default=False)
21 parser.add_option("-E", "--emit",  action="store_true",dest="emit", help="",  default=False)
22 parser.add_option("-R", "--readable-conversion",  action="store_true",dest="readable", help="", default=False)
23 parser.add_option("-B", "--binary-conversion",  action="store_true",dest="binary", help="", default=False)
24 parser.add_option("-s", "--source-file", action="store",dest="source", help="", default=False, type="string")
25 parser.add_option("-d", "--destination-file", action="store",dest="destination", help="", default=False, type="string")
26 parser.add_option("-p", "--port", action="store",dest="port", help="", default="67", type="int")
27 parser.add_option("-a", "--address", action="store",dest="address", help="", default="0.0.0.0", type="string")
28 parser.add_option("-r", "--raw", action="store",dest="raw", help="", default=False,type="string")
29 parser.add_option("-n", "--number", action="store",dest="number", help="", default="0", type="int")
30
31
32
33 (options, args) = parser.parse_args()
34
35 print options
36
37
38
39 def main() :
40     ActionSum = 0
41     for Action in (options.listen,options.emit,options.readable,options.binary) :
42         if Action == True : ActionSum += 1
43     if ActionSum > 1 :
44         print "Command line error : [-L -E -R -B] Only one of these actions can be taken."
45         sys.exit(0)
46
47     if options.readable == True : r_conversion()
48
49     if options.binary == True : b_conversion()
50
51     if options.listen == True :
52         if options.raw == False:
53             listen_address(options.address,options.port,int(options.number))
54         else  :
55             print "Listen RAW : ",options.raw
56             listen_packet(options.raw,options.number)
57
58     if options.emit == True : emit(options.address,options.port)
59
60
61 def listen_address(address,port,number) :
62     listener = dhcp_network.DhcpClient(address,port,port)
63
64     if  (number == 0 ) :
65         while True :
66             packet = listener.GetNextDhcpPacket()
67             packet.PrintHeaders()
68             packet.PrintOptions()
69
70     else :
71         while number > 0 :
72             packet = listener.GetNextDhcpPacket()
73             packet.PrintHeaders()
74             packet.Print()
75
76             number -= 1
77
78 def listen_packet(interface,number) :
79     listener = dhcp_network.DhcpRawClient(mysocket)
80
81     if  (number == 0 ) :
82         while True :
83             packet = dhcp_packet.DhcpPacket()
84             
85             packet.DecodePacket(listener.Receive(1024))
86             packet.PrintHeaders()
87             packet.PrintOptions()
88
89     else :
90         while number > 0 :
91             packet = dhcp_packet.DhcpPacket()
92             
93             packet.DecodePacket(listener.Receive(1024))
94             packet.PrintHeaders()
95             packet.PrintOptions()
96
97             number -= 1
98     
99
100 def emit(address,port) :
101     pass
102
103 def r_conversion() :
104     rawdata = sys.stdin.read()
105     while ( len(rawdata)>0 ) :
106         readdata = dhcp_packet.DhcpPacket()
107         readdata.DecodePacket(rawdata)
108         readdata.PrintHeaders()
109         readdata.PrintOptions()
110         rawdata = sys.stdin.read()
111
112 def b_conversion() :
113     """
114     pythondata =  sys.stdin.read()
115     while ( len(pythondata)>0 ) :
116         data = dhcp_packet.DhcpPacket()
117         data.DecodePacket(rawdata)
118         
119         pythondata = sys.stdin.read()
120     """
121
122 main()