VC6聊天室程序设计

2017 年 9 月 6 日 星期三
/ , ,
14

VC6聊天室程序设计

了解了C语言多线程的实例和简单的TCP通信，来编写一个简单的聊天室

客户端程序

客户端需要两个线程，主线程接受用户输入并发送到服务器；另一个线程监听服务器发来的消息，显示在屏幕上

#include "winsock2.h"
#include "stdio.h"

#define SERVER_IP   "10.80.167.248"
#define SERVER_PORP 8884

//blog:zfblog.xyz
//author:Frey

DWORD WINAPI ThreadFun(LPVOID pM)
{
    SOCKET sockClient=*(SOCKET *)pM;
    char recvInfo[100];
    while(1)
    {
        if(recv(sockClient,recvInfo,100,0)>0)
        {
            printf("%s",recvInfo);
        }
    }
    return 0;
}

void main()
{
    //加载套接字库
    WORD wVersionRequested;
    WSADATA wsaData;
    int err;
    wVersionRequested = MAKEWORD( 1, 1 ); //版本好为1.1
    err = WSAStartup( wVersionRequested, &wsaData );
    if ( err != 0 ) {
        return;
    }
    if ( LOBYTE( wsaData.wVersion ) != 1 ||
    HIBYTE( wsaData.wVersion ) != 1 ) {
        WSACleanup( );
        return;
    }
    SOCKET sockClient=socket(AF_INET,SOCK_STREAM,0); //SOCK_STREAM参数设置为TCP连接
    SOCKADDR_IN addrServer; //服务器地址结构
    addrServer.sin_addr.S_un.S_addr=inet_addr(SERVER_IP); //服务器地址
    addrServer.sin_port=htons(SERVER_PORP); //服务器端口号
    addrServer.sin_family=AF_INET;
    //与服务器端建立连接，进行通信
    char name[100];
    printf("请输入姓名:");
    scanf("%s",name);
    int connReult=connect(sockClient,(SOCKADDR*)&addrServer,sizeof(SOCKADDR));
    if(connReult!=WSAEADDRNOTAVAIL) //访问成功
    {
        CreateThread(NULL, 0, ThreadFun, (void *)&sockClient, 0, NULL);
        printf("连接成功\n");
        //成功建立连接后向服务器端发送数据，结果将显示在服务器端上
        char sendInfo[100];
        sprintf(sendInfo,name);
        send(sockClient,sendInfo,strlen(sendInfo)+1,0);
        //接收来自服务器端发送来的信息
        //char recvInfo[100];
        //recv(sockClient,recvInfo,100,0);
        //printf("%s\n",recvInfo);
        while(1)
        {
            scanf("%s",sendInfo);
            send(sockClient,sendInfo,strlen(sendInfo)+1,0);
        }
    }
    else
    {
        int errCode=WSAGetLastError();
        printf("the errcode is:%d\n",errCode);
    }
    closesocket(sockClient);
    WSACleanup();
}

服务器程序

用一个链表来存储用户的套接字和姓名； 主线程监听用户的连接，每当有新用户连接，将其信息加入链表，并传入第二个线程； 第二个线程接受客户端发来的信息，并将信息传入第三个线程； 第三个线程从链表中读取用户套接字，将接到的信息转发给所有在线用户。

#include "winsock2.h"
#include "stdio.h"
#include <windows.h>

//blog:vhcffh.com
//author:Frey

#define SERVER_PORP 8884

struct client_info{
    SOCKET sockConn;
    char name[100];
    client_info *next;
};//存放每一个用户的信息

client_info * C_info_head;


DWORD WINAPI Threadmes(LPVOID pM)
{
    char sendInfo[100];
    sprintf(sendInfo,(char *)pM);
    for(int i=0;i<=c_info_num;i++){
        SOCKET sockConn=C_info[i].sockConn;
        //if(C_info[i].num!=-1){
        send(sockConn,sendInfo,strlen(sendInfo)+1,0);
        //}
    }
    return 0;
}

//将某条消息群发给所有客户端
void sendmessage(char *message)
{
    CreateThread(NULL, 0, Threadmes, (void *)message, 0, NULL);
}
//接收每个用户的信息
DWORD WINAPI ThreadFun(LPVOID pM)
{
    client_info c_info = *(client_info *)pM;
    SOCKET sockConn=c_info.sockConn;

    char sendInfo[100];
    //inet_ntoa将结构转换为十进制的IP地址字符串
    //sprintf(sendInfo,"welcome %s to this Server!",inet_ntoa(addrClient.sin_addr));
    //成功建立连接后向客户端发送数据，结果将显示在客户端上
    //send(sockConn,sendInfo,strlen(sendInfo)+1,0);
    //从客户端接收数据，结果显示在服务器上
    char recvInfo[100];
    recv(sockConn,recvInfo,100,0);
    sprintf(c_info.name,recvInfo);

    printf("欢迎%s进入聊天室\n",recvInfo);
    sprintf(sendInfo,"欢迎%s进入聊天室\n",recvInfo);
    sendmessage(sendInfo);


    while(1)
    {
        if(recv(sockConn,recvInfo,100,0)<0)
            break;
        printf("[%s]:%s\n",c_info.name,recvInfo);
        sprintf(sendInfo,"[%s]:%s\n",c_info.name,recvInfo);
        sendmessage(sendInfo);
    }
    //将本次建立连接中得到套接字关闭
    closesocket(sockConn);
    return 0;
}


void main()
{
    //加载套接字（winsock）库，加载这段代码拷贝于MSDN中WSAStartup的介绍
    WORD wVersionRequested;
    WSADATA wsaData;
    int err;
    wVersionRequested = MAKEWORD( 1, 1 ); //版本号为1.1
    err = WSAStartup( wVersionRequested, &wsaData );
    if ( err != 0 ) {
        return;
    }
    if ( LOBYTE( wsaData.wVersion ) != 1 ||
    HIBYTE( wsaData.wVersion ) != 1 ) {
        WSACleanup( );
        return;
    }
    //消息存储及发送


    //----------------------------------------------------
    //创建套接字
    SOCKET sockServer=socket(AF_INET,SOCK_STREAM,0); //SOCK_STREAM参数设置为TCP连接
    SOCKADDR_IN addrServer; //设置服务器端套接字的相关属性
    addrServer.sin_addr.S_un.S_addr=htonl(INADDR_ANY); //设置IP
    addrServer.sin_family=AF_INET;
    addrServer.sin_port=htons(SERVER_PORP); //设置端口号
    //将套接字绑定到本地地址和指定端口上
    bind(sockServer,(SOCKADDR*)&addrServer,sizeof(SOCKADDR));
    //将套接字设置为监听模式，并将最大请求连接数设置成5，超过此数的请求全部作废
    listen(sockServer,5);
    SOCKADDR_IN addrClient; //用来接收客户端的设置，包括IP和端口
    int len=sizeof(SOCKADDR);

    while(1) //不断监听
    {
        //得到创建连接后的一个新的套接字，用来和客户端进行沟通，原套接字继续监听客户的连接请求
        SOCKET sockConn=accept(sockServer,(SOCKADDR*)&addrClient,&len);
        if(sockConn!=INVALID_SOCKET) //创建成功
        {
            c_info_num++;
            C_info[c_info_num].num=c_info_num;
            C_info[c_info_num].sockConn=sockConn;
            CreateThread(NULL, 0, ThreadFun, (void *)&C_info[c_info_num], 0, NULL);
        }
        else
        {
            int errCode=WSAGetLastError();
            printf("the errcode is:%d\n",errCode);
        }
    }
    //如果本程序不是死循环，那么在此处还应添加以下代码：
    closesocket(sockServer); //对一直处于监听状态的套接字进行关闭
    WSACleanup(); //终止对winsocket库的使用
}

未完成的问题：用户离线时出现bug；全部代码在https://github.com/summerIwinter/Chatroom

评论已关闭