TCP通信—client与server

    TCP

        client -> 1.初始化套接字 -> 2.连接 -> 数据收发

        server -> 1.初始化套接字 -> 2.绑定 -> 3.监听 -> 4.接收客户端连接 -> 数据收发

client：

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

// 函数声明
void *ThreadRecvData(void *);
char *GetTimeHMS();

// 主线程
int main(int argc, char **argv){
	
	int sockfd;							// 存储套接字
	pthread_t pt;
	unsigned short dft_port = 60000;
	struct sockaddr_in server_info;		// 存储服务端协议地址信息
	char buffer[1024] = {0};
	char *server_ip;

	if( argc < 2 ){
		fprintf(stderr, "<usage> : %s [IPv4] [port]\n" \
						"          Port default value is 60000 if you don't specify.\n", argv[0]);
		return -1;
	}
	server_ip = argv[1];
	if( argv[2] ){	// 如果制定了第二个参数
		dft_port = atoi(argv[2]);
	}

	/* 1. 初始化客户端套接字 */
	if(-1 == (sockfd = socket(AF_INET, SOCK_STREAM, 0))){
		perror("socket");
		return -1;
	}

	/* 2. 连接 */
	memset(&server_info, 0x00, sizeof(server_info));		// 清空协议地址结构
	server_info.sin_family      = AF_INET;					// 协议族
	server_info.sin_port        = htons(dft_port);			// 端口号
	server_info.sin_addr.s_addr = inet_addr(server_ip);		// IPv4地址
	
	if( -1 == connect(sockfd, (const struct sockaddr *)&server_info, sizeof(server_info)) ){
		perror("connect");
		return -1;
	}else{
		system("clear");
		printf("Connect successed\n");
	}

	// 创建线程用来接收数据
	if( pthread_create(&pt, NULL, ThreadRecvData, &sockfd) ){
		perror("connect");
		return -1;
	}

	// 发送数据
	while(1){
		fgets(buffer, sizeof(buffer), stdin);
		if(-1 == send(sockfd, buffer, strlen(buffer), 0) ){
			perror("send");
			break;
		}
	}

	/* 4. 关闭套接字 退出进程 */
	close(sockfd);

	return 0;
}

// 接收数据 - 线程
void *ThreadRecvData(void *arg){
	int  sockfd = *(int *)arg;		// 获取客户端的套接字
	char buffer[1024] = {0};
	ssize_t recvlen;
	
	while(1){
		memset(buffer, 0x00, sizeof(buffer));
		recvlen = recv(sockfd, buffer, sizeof(buffer), 0);
		if( !recvlen ){ break; }	// 连接被服务端关闭
		else if( -1 == recvlen ){
			perror("recv");
			close(sockfd);
			exit(-1);
		}
		printf("[%s]\n\t%s\n", GetTimeHMS(), buffer);
	}
	printf(
"Connection closed by foreign host.\n");
	close(sockfd);
	exit(0);
}

// 获取系统时间
char *GetTimeHMS(){
	static char buffer[64] = {0};
    time_t timep;
    struct tm *p = NULL;
    timep = time(NULL);    	// 获取时间戳
    if(NULL == (p = localtime(&timep)) ){
            perror("localtime");
            return NULL;
    }	// 解析时间数据并组装
    sprintf(buffer, "%02d:%02d:%02d", p->tm_hour, p->tm_min, p->tm_sec);
    return buffer;
}


/*
================================================================================================

	int socket(int domain, int type, int protocol);
	函数功能 : 初始化套接字
	函数参数 : domain       - 协议族
			   type     - 套接字类型 SOCK_STREAM / SOCK_DGRAM
			   protocol - 如果已经通过参数domain/type指定了套接字的使用方式 则该参数写0
	返 回 值 : 成功返回初始化好的新套接字, 失败返回-1
 
================================================================================================

	int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
	函数功能 : 连接到目标服务端
	函数参数 : sockfd       - 初始化好的客户端套接字
			   addr     - 指针, 指向的内存中的数据有服务端使用的协议族/IP地址/端口号
			   addrlen  - 长度(字节), 存储目标协议地址信息结构的大小


	in_addr_t inet_addr(const char *cp);
	函数功能 : 将cp指向的字符串内容(点分十进制形式的IP地址)转换位网络字节序无符号32位整型值

================================================================================================
*/

server：

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

void *ThreadHandler(void *arg);

int main(){

	int sockfd;		// 仅用来监听
	struct sockaddr_in server_info;

	if(-1 == (sockfd = socket(AF_INET, SOCK_STREAM, 0)) ){
		perror("socket");
		return -1;
	}

	memset(&server_info, 0x00, sizeof(server_info));
	server_info.sin_family      = AF_INET;
	server_info.sin_port        = htons(50000);
	server_info.sin_addr.s_addr = INADDR_ANY;

	if(-1 == bind(sockfd, (const struct sockaddr *)&server_info, sizeof(server_info)) ){
		perror("bind");
		return -1;
	}

	if(-1 == listen(sockfd, 1024)){
		perror("listen");
		return -1;
	}

	struct sockaddr_in client_info;
	socklen_t info_size = sizeof(client_info);
	int client_fd;
	pthread_t pt;

	while(1){
		// accept会阻塞程序运行 直到有客户端发起连接请求
		if(-1 == (client_fd = accept(sockfd, NULL, NULL)) ){
			perror("accept");
			return -1;
		}
		pthread_create(&pt , NULL, ThreadHandler, &client_fd);
	}

	
	close(sockfd);

	return 0;
}

void *ThreadHandler(void *arg){
	int client_fd = *(int *)arg;
	char buffer[1024];
	printf("New client %d connected.\n", client_fd);
	while(1){
		if( !recv(client_fd, buffer, sizeof(buffer), 0) ){
			close(client_fd);
			pthread_exit(NULL);
		}
	}
}
	

/*
	int bind  (int sockfd, const struct sockaddr *addr, socklen_t addrlen);

	int getpeername(int sockfd, struct sockaddr *addr, socklen_t *addrlen);
	int accept     (int sockfd, struct sockaddr *addr, socklen_t *addrlen);
	接收一个客户端的连接
		sockfd - 监听套机字
		addr - 指针，指向的内存用来存储连接上来的客户端的协议地址数据
		addrlen - 指针，指向的内存中的数据用来描述协议地址结构的大小
	accept函数成功将会返回一个新套接字，这个套接字用来和刚刚连接上来的客户端进行通信

*/