zoukankan      html  css  js  c++  java
  • golang 创建 tun 设备

    源码:

    package main
    
    import (
        "flag"
        "fmt"
        "github.com/pkg/errors"
        "net"
        "os"
        "syscall"
        "unsafe"
    )
    var (
        HostName, _ = os.Hostname()
        ETH_P_ARP   = 0x0806
        AF_INET     = int32(2)
        AF_INET6    = int32(10)
        AF_BRIDGE   = int32(7)
    )
    
    func Htons(i uint16) uint16 {
        return (i<<8)&0xff00 | i>>8
    }
    
    type intfReq struct {
        name  [16]byte
        flags uint16
    }
    
    // sendIOCtlMessage ioctl system call
    func sendIOCtlMessage(fd uintptr, request uintptr, ifReq uintptr) error {
        fmt.Printf("syscall fd %+v, request %+v, ifReq %+v", fd, request, ifReq)
        _, _, errno := syscall.Syscall(syscall.SYS_IOCTL, fd, request, ifReq)
        if errno != 0 {
            fmt.Printf("Fail to execute ioctl, the errno is %+v", errno)
            return errors.New("failed execute ioctl")
        }
        return nil
    }
    
    // sendFCtlMessage fcntl system call
    func sendFCtlMessage(fd int, cmd int, arg int) error {
        _, _, err := syscall.Syscall(syscall.SYS_FCNTL, uintptr(fd), uintptr(cmd), uintptr(arg))
        if err != 0 {
            fmt.Printf("Fail to execute fctl, the error is %+v", err)
            return errors.New("failed execute fctl")
        }
        return nil
    }
    
    var tunDevicePath = "/dev/net/tun"
    func main() {
    var DeviceName string flag.StringVar(&DeviceName, "name", "", "pkt0") flag.Parse() flag.Usage() fd, err := syscall.Open(tunDevicePath, os.O_RDWR, 0) if err != nil { fmt.Printf("failed to open tun device: %+v", err) return } var req intfReq req.flags = syscall.IFF_TAP | syscall.IFF_NO_PI copy(req.name[:], DeviceName) if err := sendIOCtlMessage(uintptr(fd), uintptr(syscall.TUNSETIFF), uintptr(unsafe.Pointer(&req))); err != nil { return } if err := sendFCtlMessage(fd, syscall.F_SETFD, syscall.FD_CLOEXEC); err != nil { return } if err := sendIOCtlMessage(uintptr(fd), syscall.TUNSETPERSIST, 1); err != nil { return } raw, err := syscall.Socket(syscall.AF_PACKET, syscall.SOCK_RAW, int(Htons(syscall.ETH_P_ALL))) if err != nil { fmt.Printf("failed to create RAW socket: %+v", err) return } var reqRaw intfReq copy(reqRaw.name[:], DeviceName) if err := sendIOCtlMessage(uintptr(raw), syscall.SIOCGIFINDEX, uintptr(unsafe.Pointer(&reqRaw))); err != nil { return } ifi, err := net.InterfaceByName(DeviceName) if err != nil { fmt.Printf("failed to get device: %+v", err) return } if err := syscall.Bind(raw, &syscall.SockaddrLinklayer{ Protocol: uint16(Htons(syscall.ETH_P_ALL)), Ifindex: ifi.Index, }); err != nil { fmt.Printf("failed to bind: %+v", err) return } if err := sendIOCtlMessage(uintptr(raw), syscall.SIOCSIFTXQLEN, uintptr(unsafe.Pointer(&reqRaw))); err != nil { return } if err := sendIOCtlMessage(uintptr(raw), syscall.SIOCGIFFLAGS, uintptr(unsafe.Pointer(&reqRaw))); err != nil { return } reqRaw.flags = syscall.IFF_UP if err := sendIOCtlMessage(uintptr(raw), syscall.SIOCSIFFLAGS, uintptr(unsafe.Pointer(&reqRaw))); err != nil { return } var r intfReq copy(r.name[:], DeviceName) if err := sendIOCtlMessage(uintptr(raw), syscall.SIOCGIFFLAGS, uintptr(unsafe.Pointer(&r))); err != nil { return } if err := syscall.Close(raw); err != nil { fmt.Printf("failed to close RAW socket: %+v", err) return } return }

    运行结果:

    [root@wangjq tun]# ./main -name demo
    Usage of ./main:
      -name string
            pkt0
    syscall fd 3, request 1074025674, ifReq 824634130180
    syscall fd 3, request 1074025675, ifReq 1
    syscall fd 5, request 35123, ifReq 824634130162
    syscall fd 5, request 35139, ifReq 824634130162
    syscall fd 5, request 35091, ifReq 824634130162
    syscall fd 5, request 35092, ifReq 824634130162
    syscall fd 5, request 35091, ifReq 824634130198
    [root@wangjq tun]# ip link show demo
    10: demo: <NO-CARRIER,BROADCAST,UP> mtu 1500 qdisc pfifo_fast state DOWN mode DEFAULT group default qlen 10
        link/ether 7e:e7:77:ad:f7:9e brd ff:ff:ff:ff:ff:ff

     test2:

        fd, err := syscall.Open(tunDevicePath, os.O_RDWR, 0)
        if err != nil {
            fmt.Printf("failed to open tun device: %+v", err)
            return nil, err
        }
    
        var req intfReq
        req.flags = syscall.IFF_TAP | syscall.IFF_NO_PI
        copy(req.name[:], DeviceName)
    
        if err := sendIOCtlMessage(uintptr(fd),
            uintptr(syscall.TUNSETIFF),
            uintptr(unsafe.Pointer(&req))); err != nil {
            return nil, err
        }
        if err := sendFCtlMessage(fd, syscall.F_SETFD, syscall.FD_CLOEXEC); err != nil {
            return nil, err
        }
        if err := sendIOCtlMessage(uintptr(fd), syscall.TUNSETPERSIST, 1); err != nil {
            return nil, err
        }
    
        raw, err := syscall.Socket(syscall.AF_PACKET, syscall.SOCK_RAW, int(Htons(syscall.ETH_P_ALL)))
        if err != nil {
            fmt.Printf("failed to create RAW socket: %+v", err)
            return nil, err
        }
    
        var reqRaw intfReq
        copy(reqRaw.name[:], DeviceName)
        reqRaw.flags = syscall.IFF_UP
        if err := sendIOCtlMessage(uintptr(raw),
            syscall.SIOCSIFFLAGS, uintptr(unsafe.Pointer(&reqRaw))); err != nil {
            return nil, err
        }
        if err := syscall.Close(raw); err != nil {
            fmt.Printf("failed to close RAW socket: %+v", err)
            return nil, err
        }
    
        return &device{fd: fd}, nil
  • 相关阅读:
    DBCC修复不同情况下的损坏
    Python单例模式的4种实现方法
    osx 安装redis
    tornado系列文章
    Python中__init__和__new__的区别详解
    Tornado源码分析之http服务器篇
    tornado.ioloop.IOLoop相关文章
    How to use POST method in Tornado?
    https://github.com/diogobaeder/pycket
    Frequently Asked Questions
  • 原文地址:https://www.cnblogs.com/wangjq19920210/p/11760820.html
Copyright © 2011-2022 走看看