首页 > 编程语言 >Go socket 编程源码解析(下)

Go socket 编程源码解析(下)

时间:2022-12-17 23:34:07浏览次数:68  
标签:return socket err syscall nil 源码 fd Go


在上一节中介绍了 socket 的 Listen 方法,这里进一步介绍 AcceptReadWrite 方法。

1. Accept

Accept 的核心逻辑在于:

func (ln *TCPListener) accept() (*TCPConn, error) {
	fd, err := ln.fd.accept()
	if err != nil {
		return nil, err
	}
	tc := newTCPConn(fd)
	if ln.lc.KeepAlive >= 0 {
		setKeepAlive(fd, true)
		ka := ln.lc.KeepAlive
		if ln.lc.KeepAlive == 0 {
			ka = defaultTCPKeepAlive
		}
		setKeepAlivePeriod(fd, ka)
	}
	return tc, nil
}

通过 socket 返回的 fd 调用 accept 方法从 socket 上接收数据。accept 返回新 fd,通过该新 fd 建立 tcp 连接。并且通过 setKeepAlivesetKeepAlivePeriod 函数添加对应该新 fd 的 KeepAlive 属性:tcp_keepalive_time, tcp_keepalive_intvltcp_keepalive_probes

在 KeepAlive 函数中有一段函数 runtime.KeepAlive 比较有意思:

func setKeepAlive(fd *netFD, keepalive bool) error {
	err := fd.pfd.SetsockoptInt(syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, boolint(keepalive))
	runtime.KeepAlive(fd)
	return wrapSyscallError("setsockopt", err)
}

它的存在是为了让 fd 不会被 GC 回收,更多信息可参考 issue_21402go 变量逃逸分析

继续看 accept 方法:

func (fd *netFD) accept() (netfd *netFD, err error) {
	d, rsa, errcall, err := fd.pfd.Accept()
	...
	if netfd, err = newFD(d, fd.family, fd.sotype, fd.net); err != nil {
		poll.CloseFunc(d)
		return nil, err
	}

	netfd.setAddr(netfd.addrFunc()(lsa), netfd.addrFunc()(rsa))
	return netfd, nil
}

netFD 包的是 pfd poll.FD,调用 pfd 的 Accept 方法返回 socket 上的系统文件描述符 d。将 d 包装成 netfd,接着通过 setAddr 设置 netfd 的本地地址 laddr 和 client 端地址 raddr。

poll.FDAccept 是重头戏了,接着看:

func (fd *FD) Accept() (int, syscall.Sockaddr, string, error) {
	for {
		s, rsa, errcall, err := accept(fd.Sysfd)
		if err == nil {
			return s, rsa, "", err
		}
		switch err {
		case syscall.EINTR:
			continue
		case syscall.EAGAIN:
			if fd.pd.pollable() {
				if err = fd.pd.waitRead(fd.isFile); err == nil {
					continue
				}
			}
		...
	}
}

func accept(s int) (int, syscall.Sockaddr, string, error) {
	ns, sa, err := Accept4Func(s, syscall.SOCK_NONBLOCK|syscall.SOCK_CLOEXEC)
	switch err {
	case nil:
		return ns, sa, "", nil
	...
    }
}

func accept4(s int, rsa *RawSockaddrAny, addrlen *_Socklen, flags int) (fd int, err error) {
	r0, _, e1 := Syscall6(SYS_ACCEPT4, uintptr(s), uintptr(unsafe.Pointer(rsa)), uintptr(unsafe.Pointer(addrlen)), uintptr(flags), 0, 0)
	fd = int(r0)
	if e1 != 0 {
		err = errnoErr(e1)
	}
	return
}

poll.FD 的 accept 方法中做了下面几件事:

  1. accept 函数经过 Accept4Func, accept4 到系统调用,通过系统调用号 SYS_ACCEPT4 和文件描述符 fd.Sysfd 返回作用在 socket 上的系统文件描述符和远端 socket 地址。
  2. 这里 accept 是非阻塞的,意味着即使没有 client 连接也会返回。此时返回的 err 类型为 syscall.EAGAIN
  3. 进入到 EAGAIN 错误类型中,会通过 fd.pd.pollable 方法判断是否为 true。如果为 true 阻塞当前 goroutine 直到有新的可读数据。

Accept 的实现简单介绍基本告一段落了,下面继续看 socket 的 ReadWrite 实现。

2. Read 和 Write

2.1 Read

Read 经过层层调用到 poll.FD 的 Read 方法:

func (fd *FD) Read(p []byte) (int, error) {
	...
	if err := fd.pd.prepareRead(fd.isFile); err != nil {
		return 0, err
	}
	if fd.IsStream && len(p) > maxRW {
		p = p[:maxRW]
	}
	for {
		n, err := ignoringEINTRIO(syscall.Read, fd.Sysfd, p)
		if err != nil {
			n = 0
			if err == syscall.EAGAIN && fd.pd.pollable() {
				if err = fd.pd.waitRead(fd.isFile); err == nil {
					continue
				}
			}
		}
		err = fd.eofError(n, err)
		return n, err
	}
}

从上述代码可以发现:

  • 网络处理逻辑通过层层封装走到 poll 的 Read,poll 是不区分文件还是网络数据的。因此,在 prepareRead 中需要通过 fd.isFile 判断。
  • maxRW 是能读取数据的最大字节,这里是 1G。原因分析在注释中:
// Darwin and FreeBSD can't read or write 2GB+ files at a time,
// even on 64-bit systems.
// The same is true of socket implementations on many systems.
// See golang.org/issue/7812 and golang.org/issue/16266.
// Use 1GB instead of, say, 2GB-1, to keep subsequent reads aligned.
  • ignoringEINTRIO 中通过 syscall.Read 函数,作用在系统调用上,通过系统调用号和文件描述符 fd.Sysfd 读取 socket 的数据到 p。
  • 类似 Accept,如果 ignoringEINTRIO 返回错误 syscall.EAGAIN,并且 fd.pd.pollable 是 true 的话,会阻塞当前 goroutine 等待读取数据。
  • 进入到 eofError 逻辑。对于文件,如果读到 EOF 则说明文件结束。对于网络数据,err 返回为 nil。

Write

类似于 ReadWrite 的核心逻辑在:

// Write implements io.Writer.
func (fd *FD) Write(p []byte) (int, error) {
	...
	for {
		max := len(p)
		if fd.IsStream && max-nn > maxRW {
			max = nn + maxRW
		}
		n, err := ignoringEINTRIO(syscall.Write, fd.Sysfd, p[nn:max])
		if n > 0 {
			nn += n
		}
		if nn == len(p) {
			return nn, err
		}
		if err == syscall.EAGAIN && fd.pd.pollable() {
			if err = fd.pd.waitWrite(fd.isFile); err == nil {
				continue
			}
		}
		if err != nil {
			return nn, err
		}
		if n == 0 {
			return nn, io.ErrUnexpectedEOF
		}
	}
}

通过 syscall.Write 函数进入系统调用,执行 Write 调用作用于系统文件描述符 fd.Sysfd 写数据到 p。如果返回 EAGAINpollabletrue 的话则阻塞当前 goroutine 进入 waitWrite。直到数据写完,跳出 for 循环。


标签:return,socket,err,syscall,nil,源码,fd,Go
From: https://www.cnblogs.com/xingzheanan/p/16989839.html

相关文章

  • Go socket 源码解析(上)
    0.socket介绍Liunx中一切皆文件。通过文件描述符和系统调用号可以实现对任何设备的访问。同样的,socket也是一种文件描述符。通过socket可以建立网络传输。对于TCP......
  • unbutu系统wireshark源码编译与安装
    官网:https://www.wireshark.org/官方文档:Wireshark·Documentation介绍wireshark[1]是一款抓包工具。wireshark的GUI(用户界面)框架从开发版本1.11.0(2013.11.15)开......
  • C++学习---cstdio的源码学习分析07-刷新文件流函数fflush
    cstdio中的文件访问函数stdio.h中定义了一系列文件访问函数(fopen,fclose,fflush,freopen,setbuf,setvbuf),接下来我们一起来分析一下fflush对应的源码实现。fopen:打开文件fclose:......
  • 利用Gdb探索GoRoutine的奥秘!
    主协程(runtime.main)的创建gdbmaininfofiles查看程序入口在程序入口打断点run单步调试查看下这两个参数ax=di,bx=si然后分配32字节的空间让sp对齐16......
  • abp vnext blog模块用户信息同步 源码解析
    先看一下Volo.Blogging.Domain引用的关于用户的项目只有Volo.Abp.Users.Domain,再看BlogUser定义:publicclassBlogUser:AggregateRoot<Guid>,IUser,IUpdateUserD......
  • C# Socket 使用简单测试示例
    引用Newtonsoft.Json.dllusingNewtonsoft.Json;进行对象序列化和反序列化。服务端:usingSystem;usingSystem.Collections.Generic;usingSystem.Linq;usingSyst......
  • Java超市系统超市自提超市多商家系统源码超市自提网站
    简介Ssm多商家超市自提系统。用户注册申请开店成为商家,普通注册用户下单时选择离自己较近的自提点次日取货。管理员进行店铺审核、用户、分类管理等。演示视频https://w......
  • Python+QT美颜工具源码
    OverridetheentrypointofanimageIntroducedinGitLabandGitLabRunner9.4.Readmoreaboutthe extendedconfigurationoptions.Beforeexplainingtheav......
  • [OpenCV实战]15 基于深度学习的目标跟踪算法GOTURN
    目录​​1什么是对象跟踪和GOTURN​​​​2在OpenCV中使用GOTURN​​​​3GOTURN优缺点​​​​4参考​​在这篇文章中,我们将学习一种基于深度学习的目标跟踪算法GOTURN......
  • 【SpringBoot】使用WebSocket做消息对话
    Http协议只能客户端发送---服务器回复,无法做到服务器主动向客户端发送消息,所以可以使用websocket来进行双向通道发消息 研究了一下抖音斗鱼的弹幕也是用的websocket,......