[网络协议] socket入门之TCP服务端

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 网络协议 -> socket入门之TCP服务端 -> 正文阅读

[网络协议]socket入门之TCP服务端

#include <stdio.h>

#ifdef WIN32
#include <WinSock2.h>
#include <Windows.h>
#else
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <string.h>
#endif

#include <thread>

#ifdef WIN32
#pragma comment(lib, "ws2_32.lib")
#endif

class TcpThread
{
public:
	void Main()
	{
		char buf[1024] = { 0 };

		while (1)
		{
			memset(buf, 0, 1024);
			int len1 = recv(client, buf, 1023, 0);
			printf("recv: %s\n", buf);

			if (len1 <= 0)
				break;

			if (strstr(buf, "quit") != NULL)
				break;

			int sendlen = send(client, "ok", 3, 0);
		}

#ifdef WIN32
		closesocket(client);
#else 
		close(client);
#endif

		delete this;

	}

public:
	int client = 0;
};

int main(int argc, char* argv[])
{
#ifdef WIN32
	WSADATA wsadata = { 0 };
	WSAStartup(MAKEWORD(2, 2), &wsadata);
#endif
	int s = socket(AF_INET, SOCK_STREAM, 0);
	printf("%d ", s);
	if (s == -1)
	{
		printf("Create socket failed.\n");
		return -1;
	}

	unsigned short port = 8080;
	if (argc > 1)
	{
		port = atoi(argv[1]);
	}

	sockaddr_in saddr;
	saddr.sin_family = AF_INET;
	saddr.sin_port = htons(port);
	saddr.sin_addr.s_addr = htonl(0);	// 接受任意IP地址发送的数据

	if (bind(s, (sockaddr*)&saddr, sizeof(saddr)) != 0)
	{
		printf("bind port %d failed\n", port);
		return -2;
	}
	printf("bind port %d success!\n", port);

	listen(s, 10);

	while (1)
	{
		sockaddr_in caddr = { 0 };
#ifdef WIN32
		int len = sizeof(sockaddr_in);
#else
		socklen_t len = sizeof(sockaddr_in);
#endif

		int client = accept(s, (sockaddr*)&caddr, &len);


		if (client == -1)
		{
			printf("accpet failed.\n");
			return -3;
		}

		printf("accept socket:%d\n", client);

		char *ip = inet_ntoa(caddr.sin_addr);
		printf("ip:%s\n", ip);

		unsigned short cport = ntohs(caddr.sin_port);
		printf("port:%d\n", cport);

		TcpThread *th = new TcpThread();
		th->client = client;
		std::thread sth(&TcpThread::Main, th);
		sth.detach();
	}

#ifdef WIN32
	closesocket(s);
#else
	close(s);
#endif

#ifdef WIN32
	WSACleanup();
#endif
	getchar();
	return 0;
}