Linux网络编程（1）简单的TCP客户端

TCP协议是传输层的协议，它提供了两个点之间可靠的基于数据流的全双工的传输方式。使用TCP进行网络通信，需要有服务器和客户端。下面简单介绍TCP客户端的工作流程。

首先，TCP客户端需要创建一个套接字（socket），然后使用这个套接字向服务器发起连接，这时会发生TCP的三次握手，连接成功后，就可以利用socket进行通信了。在Linux下既可以使用Linux万能的read()和write()函数，也可以使用socket提供的send()和recv()函数进行收发数据。通信结束后，一般使用close()函数关闭socket。

下面将会实现一个简单的TCP客户端，它向服务器发送一段消息，然后读取服务器返回的一段消息，最后关闭程序。

首先，简单介绍一下使用到的函数和数据结构：

#include <sys/socket.h>
int socket(int family, int type, int protocol);
int connect(int sockfd, const struct sockaddr *servaddr, socklen_t addrlen);
ssize_t recv(int sockfd, void *buff, size_t nbytes, int flags);
ssize_t send(int sockfd, const void *buff, size_t nbytes, int flags);
int shutdown(int sockfd, int howto);

#include <arpa/inet.h>
in_addr_t inet_addr(const char *strptr);
uint16_t htons(uint16_t hostshrot);

#include <unistd.h>
int close(int fd);

#include <netinet/in.h>
struct sockaddr_in
{
	uint8_t sin_len;
	sa_family_t sin_family;
	in_port_t sin_port;
	struct in_addr sin_addr;
	char sin_zero[8];
};

struct in_addr
{
	in_addr_t s_addr;
};

#include <sys/socket.h>
struct sockaddr
{
	uint8_t sa_len;
	sa_family_t sa_family;
	char sa_data[14];
};

socket()函数用来创建一个套接字，返回-1表示出错，成功时返回以创建的套接字的描述符。第一个参数是地址族，一般都填写AF_INET，代表这个网络套接字地址，使用IPv4协议。其他的选项还有AF_INET6、AF_LOCAL、AF_ROUTE、AF_KEY等，我就不详细介绍了（因为我也不太会用-_-）。第二个参数是socket的类型，如果要使用TCP协议的话填写SOCK_STREAM，其他的选项还有SOCK_DGREAM（使用UDP），SOCK_SEQPACKET（使用SCTP），SOCK_RAW（使用IP）。

connect()函数用来向服务器建立连接，就是进行三次握手。返回-1表示出错，成功时返回0。第一个参数是socket的描述符，也就是socket()函数返回的描述符。第二个参数是服务器的地址，第三个参数是服务器地址的大小。关于socket的地址在下面详细说明。

send()函数用来发送数据。失败时返回-1，成功时返回发送出去数据的长度。事实上，它只是把需要发送的数据复制socket的发送缓冲区。也就是说send()函数成功返回时，并不能保证对方已经收到了数据。第一个参数是socket描述符，第二个参数是缓冲区的地址，第三个参数是缓冲区的大小，第四个参数是发送选项。如果没有特殊需求的话一般都填写0。

recv()函数用来接收数据。失败时返回-1，成功时返回收到数据的长度。需要注意的是，如果返回0，并不代表收到了长度为0的数据，而是表示对方发送数据结束了。这是需要做的就是关闭socket。

send()和recv()函数也可以使用系统提供的read()和write()代替。他们之间的区别是，send()和recv()可以提供多种选项，而read()和write()则不能。send()和recv()默认是阻塞模式，也就是说如果数据没有复制到发送缓冲区或者没有收到数据（和TCP的实现细节有关），它们是不会返回的，程序就停在那里一直等待。

inet_addr()函数用来把一个点分十进制表示的IP（XXX.XXX.XXX.XXX）地址转换成socket地址（不需要再变换字节序，关于字节序的问题在下面介绍）。具有类似功能的函数还有inet_aton()。还有把socket地址转换成点分十进制地址的函数inet_ntoa()。这几个函数只适用于IPv4。inet_pton()和inet_ntop()这两个函数兼容Pv4、IPv6，提供了类似的功能。只是接口略有不同。我就不详细介绍了，详细内容可以百度一下，或者查看Linux Manual。

htons()函数用来提供2字节整数的字节序转换。字节序分为大端和小端，网络字节序是大端字节序，而x86机器的字节序通常都是小端。以2字节的整数0x1234为例，用大端字节序表示，高位0X12放在内存的低地址，低位0x34放在内存的高地址，在内存里的顺序是：0x12，0x34,。小端字节序高位0x12放在内存的高地址，低地址0x34放在内存的低地址，在内存里的顺序呢是：0x34，0x12。有相似功能函数还有htonl()，ntohs()，ntohl()。其实这四个函数都是宏函数，根据本机CPU字节序的宏定义来决定是否替换成改变字节序的函数。有兴趣的话可以查看一下改变字节序函数的实现。

shutdown()用来关闭套接字。它的行为和参数howto有关。howto可选的值为：SHUT_RD（值为0，只关闭读半部），SHUT_WR（值为1，之关闭写半部）,SHUT_RDWR（值为2，读半部和写半部都关闭）

close()函数也可以用来关闭一个描述符。这个函数既可以关闭套接字描述符，也可以关闭文件描述符。成功返回0，失败返回-1。也可以使用socket提供的shutdown()来关闭套接字。我觉得close()函数比较简单。关于shutdown()的使用细节可以查看Linux Manual。

shutdown()和close()的区别是，shutdown可以选择性的关闭读半部或者写半部或者都关闭，而close()只能都关闭。close()的原理是吧描述符的技术减1，只有在技术为0的时候才真正关闭了套接字。如果使用shutdown()的话则直接关闭。

下面简单介绍一下socket的地址。struct sockaddr_in是适用于IPv4的网络套接字的地址。其中需要填写的主要是sin_family，sin_addr和sin_port。它们分别代表地址族，IP地址和端口号。而sin_addr又是一个结构体，它的成员只有一个，就是s_addr，代表IP地址。sin_port和s_addr都是网络字节序的。但是struct sockaddr_in类型的地址不能直接使用，必须强制转换成struct sockaddr *类型。所以在socket提供的涉及到网络地址的函数里，网络地址的参数都是struct sockaddr *类型，同时还需要指明这个地址结构体的大小。

下面就是简单地TCP客户端的代码了：

#include <stdio.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>

#define BUFFER_SIZE 64

int main(int argc, char** argv)
{
    if(argc != 3)
    {
        printf("Usage:\n %s <ServerIp> <ServerPort>\n", argv[0]);
        return 0;
    }

    char buffer[BUFFER_SIZE];
    int socketFd;
    struct sockaddr_in serverAddr;

    memset(&serverAddr, 0, sizeof(serverAddr));
    serverAddr.sin_family = AF_INET;
    serverAddr.sin_port = htons(atoi(argv[2]));
    serverAddr.sin_addr.s_addr = inet_addr(argv[1]);

    socketFd = socket(AF_INET, SOCK_STREAM, 0);
    connect(socketFd, (struct sockaddr*)&serverAddr, sizeof(serverAddr));
    send(socketFd, "Hello World!", 13, 0);
    recv(socketFd, buffer, BUFFER_SIZE, 0);

    printf("Received Message:\n%s\n", buffer);

    close(socketFd);
    return 0;
}

运行程序看一下效果。首先用netcat来启动一个简单的TCP服务器：nc -l 2333 （启动TCP服务器，绑定2333端口）

假设上面的程序的名字为TcpClient，在编译好的程序所在的目录下输入：./TcpClient 127.0.0.1 2333

程序启动后，netcat会收到"Hello World!"的消息，然后在netcat里输入一些字符，按回车，TcpClient会收到netcat发来的消息，然后程序关闭。