Skip to content
Snippets Groups Projects
Commit 16eb4262 authored by Phil's avatar Phil
Browse files

Missing import inet6 in inet

parent e51a3172
No related branches found
No related tags found
No related merge requests found
...@@ -6,6 +6,7 @@ ...@@ -6,6 +6,7 @@
import os,time,struct,re,socket,new import os,time,struct,re,socket,new
from select import select from select import select
from scapy.utils import checksum
from scapy.layers.l2 import * from scapy.layers.l2 import *
from scapy.fields import * from scapy.fields import *
from scapy.packet import * from scapy.packet import *
...@@ -278,7 +279,7 @@ class TCP(Packet): ...@@ -278,7 +279,7 @@ class TCP(Packet):
ck=checksum(psdhdr+p) ck=checksum(psdhdr+p)
p = p[:16]+struct.pack("!H", ck)+p[18:] p = p[:16]+struct.pack("!H", ck)+p[18:]
elif isinstance(self.underlayer, inet6.IPv6) or isinstance(self.underlayer, inet6._IPv6ExtHdr): elif isinstance(self.underlayer, inet6.IPv6) or isinstance(self.underlayer, inet6._IPv6ExtHdr):
ck = in6_chksum(socket.IPPROTO_TCP, self.underlayer, p) ck = inet6.in6_chksum(socket.IPPROTO_TCP, self.underlayer, p)
p = p[:16]+struct.pack("!H", ck)+p[18:] p = p[:16]+struct.pack("!H", ck)+p[18:]
else: else:
warning("No IP underlayer to compute checksum. Leaving null.") warning("No IP underlayer to compute checksum. Leaving null.")
...@@ -332,7 +333,7 @@ class UDP(Packet): ...@@ -332,7 +333,7 @@ class UDP(Packet):
ck=checksum(psdhdr+p) ck=checksum(psdhdr+p)
p = p[:6]+struct.pack("!H", ck)+p[8:] p = p[:6]+struct.pack("!H", ck)+p[8:]
elif isinstance(self.underlayer, inet6.IPv6) or isinstance(self.underlayer, inet6._IPv6ExtHdr): elif isinstance(self.underlayer, inet6.IPv6) or isinstance(self.underlayer, inet6._IPv6ExtHdr):
ck = in6_chksum(socket.IPPROTO_UDP, self.underlayer, p) ck = inet6.in6_chksum(socket.IPPROTO_UDP, self.underlayer, p)
p = p[:6]+struct.pack("!H", ck)+p[8:] p = p[:6]+struct.pack("!H", ck)+p[8:]
else: else:
warning("No IP underlayer to compute checksum. Leaving null.") warning("No IP underlayer to compute checksum. Leaving null.")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment