《TCP/IP网络编程》第7章 优雅地断开套接字连接

close或closesocket，完全断开，既不能传输数据，也不能接收数据。

半关闭shutdown函数

#include <sys/socket.h>
//成功0，失败-1
int shutdown(int sock, int howto);
//SHUT_RD，断开输入流
//SHUT_WR，断开输出流
//SHUT_RDWR，断开输入输出流

断开输出流时向对方主机传输EOF，同时保留输入流，可以接收对方数据。

半关闭文件传输服务器端

07.file_server_linux.c

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

#define PORT 9999
#define BUF_SIZE 30

void error_handling(const char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

int main(int argc, char *argv[])
{
    int serv_sock = socket(PF_INET, SOCK_STREAM, 0);
    if (serv_sock == -1)
        error_handling("socket() error");

    int opt = 1;
    if (setsockopt(serv_sock, SOL_SOCKET, SO_REUSEADDR, (const void *)&opt, sizeof(opt)) == -1)
        error_handling("setsockopt() error");

    socklen_t addr_size = sizeof(struct sockaddr_in);

    struct sockaddr_in serv_addr;
    memset(&serv_addr, 0, addr_size);
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_addr.sin_port = htons(PORT);

    if (bind(serv_sock, (struct sockaddr *)&serv_addr, addr_size) == -1)
        error_handling("bind() error");

    if (listen(serv_sock, 5) == -1)
        error_handling("listen() error");

    struct sockaddr_in clnt_addr;
    int clnt_sock = accept(serv_sock, (struct sockaddr *)&clnt_addr, &addr_size);
    if (clnt_sock == -1)
        error_handling("accept() error");

    FILE *fp = fopen("07.file_server_linux.c", "rb");
    char buf[BUF_SIZE];
    int read_cnt;
    while (read_cnt = fread((void *)buf, 1, BUF_SIZE, fp))
        write(clnt_sock, buf, read_cnt);
    fclose(fp);

    // 关闭输出流，不再向客户端发送数据
    shutdown(clnt_sock, SHUT_WR);

    while (read_cnt = read(clnt_sock, buf, BUF_SIZE))
    {
        buf[read_cnt] = 0;
        printf("Message from client: %s\n", buf);
    }

    // close(clnt_sock);
    shutdown(clnt_sock, SHUT_RD);

    close(serv_sock);

    return 0;
}

// gcc 07.file_server_linux.c -o 07.file_server_linux && ./07.file_server_linux

半关闭文件传输客户端

07.file_client_linux.c

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

#define IP "127.0.0.1"
#define PORT 9999
#define BUF_SIZE 30

void error_handling(const char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

int main(int argc, char *argv[])
{
    int sock = socket(PF_INET, SOCK_STREAM, 0);
    if (sock == -1)
        error_handling("socket() error");

    socklen_t addr_size = sizeof(struct sockaddr_in);

    struct sockaddr_in addr;
    memset(&addr, 0, addr_size);
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = inet_addr(IP);
    addr.sin_port = htons(PORT);

    if (connect(sock, (struct sockaddr *)&addr, addr_size) == -1)
        error_handling("connect() error");

    FILE *fp = fopen("receive.dat", "wb");
    char buf[BUF_SIZE];
    int read_cnt;
    while ((read_cnt = read(sock, buf, BUF_SIZE)) != 0)
        fwrite((void *)buf, 1, read_cnt, fp);
    fclose(fp);

    puts("Received file data");

    write(sock, "Thank you", 10);

    close(sock);

    return 0;
}

// gcc 07.file_client_linux.c -o 07.file_client_linux && ./07.file_client_linux

Windows实现

#include <winsock2.h>
//成功0，失败SOCKET_ERROR
int shutdown(int sock, int howto);
//SD_RECEIVE，断开输入流
//SD_SEND，断开输出流
//SD_BOTH，断开输入输出流

07.file_server_win.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <winsock2.h>

#define PORT 9999
#define BUF_SIZE 30

void ErrorHanding(const char *message)
{
	fputs(message, stderr);
	fputc('\n', stderr);
	exit(1);
}

int main(int argc, char *argv[])
{
	WSADATA wsaData;
	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
		ErrorHanding("WSAStartup() error!");

	SOCKET serv_sock = socket(PF_INET, SOCK_STREAM, 0);
	if (serv_sock == -1)
		ErrorHanding("socket() error");

	int opt = 1;
	if (setsockopt(serv_sock, SOL_SOCKET, SO_REUSEADDR, (const char *)&opt, sizeof(opt)) < 0)
		ErrorHanding("setsockopt() error");

	int addr_size = sizeof(SOCKADDR_IN);

	SOCKADDR_IN serv_addr;
	memset(&serv_addr, 0, addr_size);
	serv_addr.sin_family = AF_INET;
	serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	serv_addr.sin_port = htons(PORT);

	if (bind(serv_sock, (SOCKADDR *)&serv_addr, addr_size) == SOCKET_ERROR)
		ErrorHanding("bind() error");

	if (listen(serv_sock, 5) == SOCKET_ERROR)
		ErrorHanding("listen() error");

	SOCKADDR_IN clnt_addr;
	SOCKET clnt_sock = accept(serv_sock, (SOCKADDR *)&clnt_addr, &addr_size);
	if (clnt_sock == INVALID_SOCKET)
		ErrorHanding("accept() error");

	FILE *fp = fopen("07.file_server_win.c", "rb");
	int read_cnt;
	char buf[BUF_SIZE];
	while (read_cnt = fread((void *)buf, 1, BUF_SIZE, fp))
		send(clnt_sock, buf, read_cnt, 0);
	fclose(fp);

	//关闭输出流，不再向客户端发送数据
	shutdown(clnt_sock, SD_SEND);

	//recv(clnt_sock, buf, BUF_SIZE, 0);
	while (read_cnt = recv(clnt_sock, buf, BUF_SIZE - 1, 0))
	{
		buf[read_cnt] = 0;
		printf("Message from client: %s\n", buf);
	}

	// close(clnt_sock);
	shutdown(clnt_sock, SD_RECEIVE);

	closesocket(serv_sock);

	WSACleanup();

	return 0;
}

// gcc 07.file_server_win.c -o 07.file_server_win -lws2_32 && 07.file_server_win

07.file_client_win.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <winsock2.h>

#define IP "127.0.0.1"
#define PORT 9999
#define BUF_SIZE 30

void ErrorHanding(const char *message)
{
	fputs(message, stderr);
	fputc('\n', stderr);
	exit(1);
}

int main(int argc, char *argv[])
{
	WSADATA wsaData;
	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
		ErrorHanding("WSAStartup() error!");

	SOCKET sock = socket(PF_INET, SOCK_STREAM, 0);
	if (sock == -1)
		ErrorHanding("socket() error");

	int addr_size = sizeof(SOCKADDR_IN);

	SOCKADDR_IN addr;
	memset(&addr, 0, addr_size);
	addr.sin_family = AF_INET;
	addr.sin_addr.s_addr = inet_addr(IP);
	addr.sin_port = htons(PORT);

	if (connect(sock, (SOCKADDR *)&addr, addr_size) == SOCKET_ERROR)
		ErrorHanding("connect() error");

	FILE *fp = fopen("receive.dat", "wb");
	char buf[BUF_SIZE];
	int read_cnt;
	while ((read_cnt = recv(sock, buf, BUF_SIZE, 0)) != 0)
		fwrite((void *)buf, 1, read_cnt, fp);
	fclose(fp);

	puts("Received file data");

	send(sock, "Thank you", 10, 0);

	closesocket(sock);

	WSACleanup();

	return 0;
}

// gcc 07.file_client_win.c -o 07.file_client_win -lws2_32 && 07.file_client_win