第二章 套接字类型与协议设置

套接字类型与协议设置

套接字协议及其数据传输特性

关于协议
计算机间对话必备通信规则

一、 创建套接字

#include <sys/socket.h>

int socket(int domain, int type, int protocol);
成功返回文件描述符，失败返回-1

domain：套接字中使用的协议族（Protocol Family）信息
type：套接字数据传输类型信息
protocol：计算机通信中使用的协议信息

• 协议族
  
  套接字实际使用协议信息通过socket第三个参数传递，在指定协议族范围内通过第一个参数决定第三个参数

AF PF
AF 表示ADDRESS FAMILY 地址族
PF 表示PROTOCOL FAMILY 协议族

在windows系统中，这两个宏定义是一样的，所以使用哪个都没有关系，在Winsock2.h中：

#define AF_INET 2
#define PF_INET AF_INET

在Unix/Linux系统中，在不同的版本中这两者有微小差别：
对于BSD，是AF
对于POSIX，是PF

• 套接字类型：套接字数据传输方式，比如PF_INET协议族也存在多种数据传输方式

面向连接的套接字(SOCK_STREAM)
可靠的、按序传递的、基于字节的面向连接的数据传输方式的套接字

1. 连接可靠传输，除非特殊情况不会发生数据丢失
2. 按序传输数据，按先后顺序传输到缓冲区
3. read和recv有缓冲(buffer)，buffer可以被写满或者不写满然后被读取，read和recv的次数并不对应
4. 缓冲区满不会导致数据丢失。套接字会停止传输，面向连接的套接字根据接收端的状态传输数据，传输出错还会提供重传服务
5. 套接字连接必须一一对应

面向消息的套接字(SOCK_DGRAM)
不可靠的、不按序传递到、以数据的高速传输为目的的套接字
特点：

1. 强调快速传输而非传输顺序
2. 传输的数据可能丢失也可能损毁
3. 传输的数据有数据边界,接受数据的次数应和传输次数相同
4. 限制每次传输的数据大小
   例子：
5. 两辆摩托车送两个包裹给一个客户，无需保证顺序，可能会丢失，包裹大小有限制，接收者要2次接受
6. 不存在连接的概念，通过广播实现

• 协议的最终选择：socket第三个参数

1. 同一个协议族中存在多个数据传输方式相同的协议，用第三个参数具体制定协议信息

2. 以前面的内容为例子，创建“IPv4协议族中面向连接的套接字”，PF_INET、SOCK_STREAM，满足这两个条件的协议只有IPPROTO_TCP

int tcp_socket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); // TCP套接字

3. 创建"IPv4协议族中面向消息的套接字"

int udp_socket = socket(PF_INT, SOCK_DRAM, IPPROTO_UDP); // UDP Socket

二、面向连接的套接字: TCP套接字示例

• Linux示例
  tcp_server.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
void error_handling(char *message);
int main(int argc, char* argv){
	int serv_sock;
	int clnt_sock;
	struct sockaddr_in serv_addr;
	struct sockaddr_in clnt_addr;
	scoklen_t clnt_addr_size;
	
	char message[] = "Hello World!";
	
	if(argc!=2){
		printf("Usage : %s <port>\n", argv[0]);
		exit(1);
	}
	
	serv_sock=socket(PF_INET, SOCK_STREAM, 0);
	if(serv_sock == -1)
		error_handling("socket() error");
	
	memset(&serv_addr, 0, sizeof(serv_addr));
	serv_addr.sin_family=AF_INET;	    												 									 
	serv_addr.sin_addr.s_addr
=htonl(INADDR_ANY);
	serv_addr.sin_port
=htons(atoi(argv[1]));
	if(bind(serv_sock, (struct sockaddr*)&serv_addr,sizeof(serv_addr))==-1)                          
		error_handing("bind() error");
	if(listen(serv_sock, 5)==-1)
		error_handling("listen() error");
	
	clnt_adder_size=sizeof(clnt_addr);
	clnt_sock=accept(serv_sock, (struct sockaddr*)&clnt_addr, &clnt_addr_size);
	if(clnt_sock==-1)
		error_handling("accept() error");
	
	write(clnt_sock, message, sizeof(message));
	close(clnt_sock);
	close(serv_sock);
	return 0;
}
void error_handling(char *message){
	fputs(message, stderr);
	fputc('\n', stderr);
	exit(1);
}

tcp_client.c

tcp_client.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
void error_handling(char *message);
int main(int argc, char* argv[]){
	int sock;
	struct sockaddr_in serv_addr;
	char message[30];
	int str_len;
	int idx=0, read_len=0;

	if(argc!=3){
		printf("Usage : %s <IP> <port>\n", argv[0]);
		exit(1);
	}
    // 创建TCP套接字。如果前面两个参数传递PF_INET、SOCK_STREAM，可省略第三个参数
	sock=socket(PF_INET, SOCK_STREAM, 0);
	if(sock==-1)
		error_handling("socket() error");

	memset(&serv_addr, 0, sizeof(serv_addr));
	serv_addr.sin_family=AF_INET;
		serv_addr.sin_addr.s_addr=inet_addr(argv[1]);
	serv_addr.sin_port=htons(atoi(argv[2]));
	if(connect(sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr))==-1)
		error_handling("connect() error!");
	
	// 每次读取1个字节，如果read返回0，循环结束，读完之后str_len存有读取的总字节数
	while(read_len=read(sock, &message[idx++], 1)){
		if(read_len==-1)
			error_handling("read() error!");
		str_len+=read_len;
	}
	printf("Message from server: %s \n", message);
	printf("Function read call count: %d \n", str_len);
	close(sock);
	return 0;
}
void error_handling(char *message){
	fputs(message, stderr);
	fputc('\n', stderr);
	exit(1);
}

运行：

gcc tcp_client.c -o hclient
./hclient 127.0.0.1 9190
Message from server: Hello World!
Function read call count: 13

• Windows示例
  windows操作系统socket函数函数名和参数与Linux平台相同，返回值类型不同

#include <winsock2.h>
SOCKET socket(int af, int type, int protocol);
成功返回socket句柄，失败返回INVALID_SOCKET

返回值SOCKET，保存整数型套接字句柄值，int型，但建议用SOCKET类型接收

微软建议使用SOCKET，同样的对于INVALID_SOCKET也是
INVALID_SOCKET表示提示错误的常数，其值为-1

SOCKET soc = socket(PF_INET, SOCKET_STREAM, IPPROTO_TCP);
if (soc == -1)
    ErrorHanding("...");

应该这样写：

SOCKET soc = socket(PF_INET, SOCKET_STREAM, IPPROTO_TCP);
if (soc == INVALID_SOCKET)
    ErrorHanding("...");

tcp_server_win.c
和 hello_server_win.c一样

tcp_client_win.c

#include <stdio.h>
#include <stdlib.h>
#include <winsock2.h>
#include <WS2tcpip.h>
#pragma comment(lib, "ws2_32.lib")
void ErrorHandling(const char* message);
int main(int argc, char* argv[]) {
	WSADATA wsaData;
	SOCKET hSocket; // 声明SOCKET变量以保存socket函数返回值
	SOCKADDR_IN servAddr;

	char message[30];
	int strLen;
	int idx=0, readLen=0;
	if (argc != 3) {
		printf("Usage : %s <IP> <port>\n", argv[0]);
		exit(1);
	}
	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
		ErrorHandling("WSAStartup() error!");
    // 调用socket函数创建TCP套接字    
	hSocket = socket(PF_INET, SOCK_STREAM, 0);
	if (hSocket == INVALID_SOCKET)
		ErrorHandling("socket() error");
	memset(&servAddr, 0, sizeof(servAddr));
	servAddr.sin_family = AF_INET;
	//servAddr.sin_addr.s_addr = inet_pton(argv[1]); // 老版本函数
	inet_pton(AF_INET, argv[1], (void*)&servAddr.sin_addr.s_addr);
	servAddr.sin_port = htons(atoi(argv[2]));
	// 调用connect向服务器发送连接请求，成功时返回0，失败时返回SOCKET_ERROR
	if (connect(hSocket, (SOCKADDR*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR)
		ErrorHandling("connect() error!");
	
    // while循环中调用recv函数读取数据，每次读取1个字节
	while(readLen=recv(hSocket, &message[idx++], 1, 0)){
		if(readLen==-1)
			ErrorHanding("read() error!");	
        // 每次读取1个字节，变量strlen加1,与recv函数调用次数相同    
		strLen+=readLen;
	}
	printf("Message from server: %s \n", message);
	printf("Function read call count: %d \n", strLen);

	closesocket(hSocket);
	WSACleanup(); // 注销第20行中初始化的Winsock库
	return 0;
}
void ErrorHandling(const char* message) {
	fputs(message, stderr);
	fputs((char*)'\n', stderr);
	exit(1);
}

运行：

hTCPClientWin 127.0.0.1 9190
Message from server: Hello World!
Function read call count: 13