golang 获取tcp连接的文件描述符fd

Posted 2023-05-08

参考技术A 有个通过代理进来的tcp连接，通过Conn.RemoteAddr获取到的是代理点的ip地址，为了获取实际客户端的ip，找到了syscall.Getpeername的方法，而这个方法需要的是连接的fd。

使用read write 读写socket

一旦，我们建立好了tcp连接之后，我们就可以把得到的fd当作文件描述符来使用。

由此网络程序里最基本的函数就是read和write函数了。

写函数：

ssize_t write(int fd, const void*buf,size_t nbytes);

write函数将buf中的nbytes字节内容写入文件描述符fd.成功时返回写的字节数.失败时返回-1. 并设置errno变量. 在网络程序中,当我们向套接字文件描述符写时有两可能. 

1)write的返回值大于0,表示写了部分或者是全部的数据. 这样我们用一个while循环来不停的写入，但是循环过程中的buf参数和nbyte参数得由我们来更新。也就是说，网络写函数是不负责将全部数据写完之后在返回的。

2)返回的值小于0,此时出现了错误.我们要根据错误类型来处理. 如果错误为EINTR表示在写的时候出现了中断错误. 如果为EPIPE表示网络连接出现了问题(对方已经关闭了连接). 为了处理以上的情况,我们自己编写一个写函数来处理这几种情况. 

int my_write(int fd,void *buffer,int length) 

{ 

int bytes_left; 

int written_bytes; 

char *ptr; 

ptr=buffer; 

bytes_left=length; 

while(bytes_left>0) 

{ 

         /* 开始写*/ 

         written_bytes=write(fd,ptr,bytes_left); 

         if(written_bytes<=0) /* 出错了*/ 

         {        

                 if(errno==EINTR) /* 中断错误 我们继续写*/ 

                         written_bytes=0; 

                 else             /* 其他错误 没有办法,只好撤退了*/ 

                         return(-1); 

         } 

         bytes_left-=written_bytes; 

         ptr+=written_bytes;     /* 从剩下的地方继续写   */ 

} 

return(0); 

} 

读函数read 

ssize_t read(int fd,void *buf,size_t nbyte) 

read函数是负责从fd中读取内容.当读成功 时,read返回实际所读的字节数,如果返回的值是0 表示已经读到文件的结束了,小于0表示出现了错误.如果错误为EINTR说明读是由中断引起 的, 如果是ECONNREST表示网络连接出了问题. 和上面一样,我们也写一个自己的读函数. 

int my_read(int fd,void *buffer,int length) 

{ 

int bytes_left; 

int bytes_read; 

char *ptr; 

bytes_left=length; 

while(bytes_left>0) 

{ 

     bytes_read=read(fd,ptr,bytes_read); 

     if(bytes_read<0) 

     { 

       if(errno==EINTR) 

          bytes_read=0; 

       else 

          return(-1); 

     } 

     else if(bytes_read==0) 

         break; 

      bytes_left-=bytes_read; 

      ptr+=bytes_read; 

} 

return(length-bytes_left); 

}