Go语言linux下构造tcp数据包(不使用net包和Cgo)

原创 2014年03月03日 19:54:07

Golang可以通过syscall包中的相关socket函数(Windows和Linux中syscall中网络编程的函数不太一样)来进行网络编

程,可以用我们熟悉的方法使用raw socket,因为最近在搞一些协议的学习,所以特意试了试,和普通的linux+C的

socket编程基本没什么区别。


代码如下:

package main

import (
    "bytes"
    "encoding/binary"
    . "fmt"
    "strconv"
    "strings"
    "syscall"
    "unsafe"
)

type TCPHeader struct {
    SrcPort   uint16
    DstPort   uint16
    SeqNum    uint32
    AckNum    uint32
    Offset    uint8
    Flag      uint8
    Window    uint16
    Checksum  uint16
    UrgentPtr uint16
}

type PsdHeader struct {
    SrcAddr   uint32
    DstAddr   uint32
    Zero      uint8
    ProtoType uint8
    TcpLength uint16
}

func inet_addr(ipaddr string) uint32 {
    var (
        segments []string = strings.Split(ipaddr, ".")
        ip       [4]uint64
        ret      uint64
    )
    for i := 0; i < 4; i++ {
        ip[i], _ = strconv.ParseUint(segments[i], 10, 64)
    }
    ret = ip[3]<<24 + ip[2]<<16 + ip[1]<<8 + ip[0]
    return uint32(ret)
}

func htons(port uint16) uint16 {
    var (
        high uint16 = port >> 8
        ret  uint16 = port<<8 + high
    )
    return ret
}

func CheckSum(data []byte) uint16 {
	var (
		sum    uint32
		length int = len(data)
		index  int
	)
	for length > 1 {
		sum += uint32(data[index])<<8 + uint32(data[index+1])
		index += 2
		length -= 2
	}
	if length > 0 {
		sum += uint32(data[index])
	}
	sum += (sum >> 16)

	return uint16(^sum)
}

func main() {
    var (
        msg       string
        psdheader PsdHeader
        tcpheader TCPHeader
    )

    Printf("Input the content: ")
    Scanf("%s", &msg)

    /*填充TCP伪首部*/
    psdheader.SrcAddr = inet_addr("127.0.0.1")
    psdheader.DstAddr = inet_addr("127.0.0.1")
    psdheader.Zero = 0
    psdheader.ProtoType = syscall.IPPROTO_TCP
    psdheader.TcpLength = uint16(unsafe.Sizeof(TCPHeader{})) + uint16(len(msg))

    /*填充TCP首部*/
    tcpheader.SrcPort = htons(3000)
    tcpheader.DstPort = htons(8080)
    tcpheader.SeqNum = 0
    tcpheader.AckNum = 0
    tcpheader.Offset = uint8(uint16(unsafe.Sizeof(TCPHeader{}))/4) << 4
    tcpheader.Flag = 2 //SYN
    tcpheader.Window = 60000
    tcpheader.Checksum = 0

    /*buffer用来写入两种首部来求得校验和*/
    var (
        buffer bytes.Buffer
    )
    binary.Write(&buffer, binary.BigEndian, psdheader)
    binary.Write(&buffer, binary.BigEndian, tcpheader)
    tcpheader.Checksum = CheckSum(buffer.Bytes())

    /*接下来清空buffer,填充实际要发送的部分*/
    buffer.Reset()
    binary.Write(&buffer, binary.BigEndian, tcpheader)
    binary.Write(&buffer, binary.BigEndian, msg)

    /*下面的操作都是raw socket操作,大家都看得懂*/
    var (
        sockfd int
        addr   syscall.SockaddrInet4
        err    error
    )
    if sockfd, err = syscall.Socket(syscall.AF_INET, syscall.SOCK_RAW, syscall.IPPROTO_TCP); err != nil {
        Println("Socket() error: ", err.Error())
        return
    }
    defer syscall.Shutdown(sockfd, syscall.SHUT_RDWR)
    addr.Addr[0], addr.Addr[1], addr.Addr[2], addr.Addr[3] = 127, 0, 0, 1
    addr.Port = 8080
    if err = syscall.Sendto(sockfd, buffer.Bytes(), 0, &addr); err != nil {
        Println("Sendto() error: ", err.Error())
        return
    }
    Println("Send success!")
}


如果转载请注明文章出处:http://blog.csdn.net/gophers/article/details/20393601

版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

用Golang自己构造ICMP数据包

ICMP是用来对网络状况进行反馈的协议,可以用来侦测网络状态或检测网路错误。 限于当前Golang在网络编程方面的代码稀缺,资料甚少,所以分享一个用Golang来构造ICMP数据包并发送ping程...

Linux下GO语言内存共享,CGO实现

package ce import ( "errors" "fmt" "os" "os/exec" ) /* #include #include #include #include ...

Golang在windows下的socket编程(不使用net包和Cgo)

go也可以想普通的socket编程那样: 创建套接字 -> 绑定 -> 监听 … 当然,go中有net包可以使用,但是如果要实现一些比较底层的操作,例如自己构造数据包,就可以通过这种比较原始的方式来进...

go语言的cgo简单教程

目前Go语言有2套编译器:GC和gccgo。其中GC提供的cgo支持C语言,gccgo支持C/C++。 此外,SWIG从2.0.1之后也对go语言提供支持,可以支持C++的类和回调。Go官方提供...

linux下的go语言的websocket库

  • 2017-01-12 17:50
  • 7.53MB
  • 下载

详解C语言开发unix/Linux下安全套接字(SSL)的TCP服务端程序

本文详细说明了用C语言开发UNIX/Linux下安全套接字(TCP SSL) 的服务端程序,采用IO多路复用,单进程单线程,自定义并实现 缓存,链表等数据结构,代码由本人亲自编写,在IBM AIX...

linux下 socket tcp Server c语言编写(分别实现单进程,多进程,多线程)

TCP 迭代服务器接受一个客户端的连接,然后处理,完成了这个客户的所有请求后,断开连接。TCP 迭代服务器一次只能处理一个客户端的请求,只有在这个客户的所有请求满足后,服务器才可以继续后面的请求。如果...

Linux下C语言TCP编程01

TCP01简单的TCP编程 1、在线客户端数量为1; 2、服务端只支持接收和回复,不能输入后进行发送; 3、客户端能输入发送并接收服务器的回复。
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)