C语言socket 案例

#include "stdafx.h"
#include <stdio.h>
#include <stdlib.h>
#include <WinSock2.h>			// windows套接字头文件
#pragma comment(lib,"ws2_32.lib")

char cName[50];		// 存放名称 主副线程公用的变量
int iFlag = 0;		// 服务器模式	判断是不是自己发出数据信息（1是，0不是）

struct sClient		// 声明客户端结构体类型
{
	SOCKET s;
	sockaddr_in sin;
};

DWORD WINAPI Server_Send_ThreadFunc(LPVOID lp);		// 服务器发送消息线程函数
DWORD WINAPI Client_Receive_ThreadFunc(LPVOID lp);	// 客户端接收信息线程函数

// 设置控制台颜色输出函数
void SetColorFunc(int a)
{
	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), a);		// 控制台颜色设置
}


int _tmain(int argc, _TCHAR* argv[])
{
	// 设置颜色为黑底白字
	SetColorFunc(7);
	char cAddress[50];		// 存放对方的IP信息
	int iModel = 0;			// 0-客户端  1-服务器

	// 输出控制UI
	printf("\n\n※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※\n");
	printf("※                                                                    ※\n");
	printf("※                    迷人聊天通信应用系统 V1.0                         ※\n");
	printf("※                                                                    ※\n");
	printf("※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※\n");
	printf("※                                                                    ※\n");
	printf("※                    通信应用系统使用说明书                          ※\n");
	printf("※                                                                    ※\n");
	printf("※        1.用户输入IP地址自动执行服务器或客户端功能                  ※\n");
	printf("※        2.服务器的机器IP地址，格式为:192.168.1.199                  ※\n");
	printf("※        3.本地作为服务器，输入127.0.0.1，若错误默认为服务器         ※\n");
	printf("※                                                                    ※\n");
	printf("※                                版权所有 违者必究@2019              ※\n");
	printf("※                                                                    ※\n");
	printf("※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※\n");

	printf("请输入对方的IP地址(需要对方机器处于服务器模式):\n");
	scanf_s("%s", cAddress, 50);
	printf("请输入你的昵称(昵称不能为空或者有空格，若程序暂时无反应，请检查IP地址是否正确):\n");
	scanf_s("%s", cName, 50);

	// 初始化WinSock 2.0版本
	WSADATA wsaData;
	int ret;
	if ((ret = WSAStartup(MAKEWORD(2, 2), &wsaData)) != 0)
	{
		printf("初始化Windows Socket错误.\n");
		system("pause");
		return 0;
	}

	// 创建套接字
	SOCKET acceptSocket;		// 声明socket变量 用来建立连接服务
	acceptSocket = socket(AF_INET, SOCK_STREAM, 0);		// 创建套接字socket
	if (acceptSocket == INVALID_SOCKET)
	{
		printf("建立Socket失败，请重新检查.\n");
		system("pause");
		return 0;
	}

	struct sockaddr_in server;		// 定义设置服务器地址和端口变量
	// 设置服务器地址信息
	server.sin_family = AF_INET;	// 设置格式为AF_INET  指的是协议簇，在使用socket编程中只能用AF_INET
	server.sin_port = htons(12266);	// 设置监听端口号为12266，设置需要连接的服务器监听端口（必须和服务器端口绑定的监听端口保持一致）
	server.sin_addr.s_addr = inet_addr(cAddress);		//设置要绑定的服务器的IP地址
	if (connect(acceptSocket, (SOCKADDR*)&server, sizeof(server)) == INVALID_SOCKET)  // 请求与服务器建立TCP连接
	{
		printf("请求建立连接失败，系统自动转换为【服务器模式】\n");
		iModel = 1;	// 无法连接对方ip，设置iModel参数，为后面自动进入【服务器模式】
	}




	return 0;
}


DWORD WINAPI Server_Send_ThreadFunc(LPVOID lp)		// 服务器发送消息线程函数
{
	SOCKET *s = (SOCKET*)lp;

	while (true)
	{
		int iRett;
		char cBuffer[1024];
		char cSendBuffer[256];

		memset(cBuffer, 0, sizeof(cBuffer));		// 将字符数组清零

		scanf_s("%s", cBuffer, 1024);				// 输入要发送数据信息
		iFlag = 1;

		// 判断用户输入是否为退出的指令:QUIT 或quit
		if (cBuffer[0] == 'q' || cBuffer[0] == 'Q' && cBuffer[1] == 'u' || cBuffer[1] == 'U' && cBuffer[2] == 'i' || cBuffer[2] == 'I' && cBuffer[3] == 't' || cBuffer[3] == 'T')
			break;
		memset(cSendBuffer, 0, sizeof(cSendBuffer));
		strcpy_s(cSendBuffer, cName);		// 先将名字复制到cSendBuffer
		strcat_s(cSendBuffer, " 说:");
		strcat_s(cSendBuffer, cBuffer);

		// 将数据信息发送出去
		iRett = ::send(*s, cSendBuffer, sizeof(cSendBuffer), 0);
		if (iRett == SOCKET_ERROR)
		{
			printf("\n你发送数据信息有错，请重新检测.\n");
			break;
		}
	}

	closesocket(*s);
	system("pause");

	return 0;
}

DWORD WINAPI Client_Receive_ThreadFunc(LPVOID lp)	// 客户端接收信息线程函数
{
	SOCKET *s = (SOCKET*)lp;
	while (true)
	{
		int iNrecv;
		char cBuffer[1024];
		iNrecv = ::recv(*s, cBuffer, 1024, 0);
		if (iNrecv > 0)
		{
			cBuffer[iNrecv] = '\0';
			SetColorFunc(14);
			printf("%s\n", cBuffer);
			SetColorFunc(8);
		}
		if (iNrecv == SOCKET_ERROR)
		{
			printf("与服务器连接已断开操作.\n");
			system("pause");
			return 0;
		}
	}
return 0;
}