#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;
}
C语言socket 案例
最新推荐文章于 2021-05-19 09:23:11 发布