Python原始套接字编程

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

#!/usr/bin/env python

#-------------------------------------------------------------------------------

# Name:     raw_http.py

# Purpose:       construct a raw http get packet

#

# Author:    Yangjun

#

# Created:       08/02/2014

# Copyright:   (c) Yangjun 2014

# Licence:       <your licence>

#-------------------------------------------------------------------------------

import sys

import socket

from impacket import ImpactDecoder, ImpactPacket

def main():

    if len(sys.argv) < 3:

        print "Use: %s <src ip> <dst ip>" % sys.argv[0]

        print "Use: %s <src ip> <dst ip> <cnt>" % sys.argv[0]

        sys.exit(1)

    elif len(sys.argv) == 3:

        src = sys.argv[1]

        dst = sys.argv[2]

        cnt = 1

    elif len(sys.argv) ==4:

        src = sys.argv[1]

        dst = sys.argv[2]

        cnt = sys.argv[3]

    else:

        print "Input error!"

        sys.exit(1)

#print src, dst

    ip = ImpactPacket.IP()

    ip.set_ip_src(src)

    ip.set_ip_dst(dst)

    # Create a new ICMP packet of type ECHO.

    icmp = ImpactPacket.ICMP()

    tcp = ImpactPacket.TCP()

    tcp.set_th_sport(55968)

    tcp.set_th_dport(80)

    tcp.set_th_seq(1)

    tcp.set_th_ack(1)

    tcp.set_th_flags(0x18)

    tcp.set_th_win(64)

    tcp.contains( ImpactPacket.Data("GET /att/DIYLife/41264/528 HTTP/1.1\r\nHost: 192.168.111.1\r\nAccept-Encoding: identity\r\n\r\n"))

    ip.contains(tcp)

    # Open a raw socket. Special permissions are usually required.

    s = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_TCP)

    s.setsockopt(socket.IPPROTO_IP, socket.IP_HDRINCL, 1)

    seq_id = 0

    while cnt >= 1:

        # Calculate its checksum.

        seq_id = seq_id + 1

        tcp.set_th_seq(seq_id)

        tcp.calculate_checksum()

        # Send it to the target host.

        s.sendto(ip.get_packet(), (dst,80))

        cnt= cnt -1

if __name__ == ‘__main__‘:

    main()