【网络编程02】简单聊天程序

　　有了上次socket的基础知识，网络编程01  可以写一个简单的聊天程序,随便复习一下基础知识。

1.线程

　　我们要实现可以同时发送和接收信息，就需要多线程了，一个用于发，一个用于收。

　　这次用CreateThread来创建线程。函数原型

　　HANDLE WINAPI CreateThread

　　( 　 _In_opt_             LPSECURITY_ATTRIBUTES       　　　lpThreadAttributes,

　　　　_In_                    SIZE_T          　　　　　　　　　　　　dwStackSize,

　　　　_In_          　　   LPTHREAD_START_ROUTINE        　　lpStartAddress,

　　　　_In_opt_ 　　    LPVOID               　　 　　　　　　　　  lpParameter,

　　　　_In_　　　　      DWORD                     　　 　　　　　　 dwCreationFlags,

　　　　_Out_opt_ 　　 LPDWORD         　　　　　　　　　　    lpThreadId

　　);

　　函数看着复杂其实用起来很简单。只需要关心传入什么函数，函数参数是什么。

　　1.第一个参数表示线程内核对象的安全属性，一般传入NULL表示使用默认设置

　　2.第二个参数表示线程栈空间大小。传入0表示使用默认大小

　　3.第三个参数表示新线程所执行的线程函数地址（函数的名字），多个线程可以使用同一个函数地址

　　4.第四个参数是传给线程函数的参数。

　　5.第五个参数指定什么时候调用线程，为0表示线程创建之后立即就可以进行调用。

　　6.第六个参数返回线程的ID号，传入NULL表示不需要返回该线程ID号

2.服务器端代码

　　我们把发送信息和接收信息的功能封装成两个函数作为线程使用。

　　

#include<stdio.h>
#include<string.h>
#include<WinSock2.h>
#pragma comment (lib,"ws2_32.lib")
SOCKET newSocket;//全局变量，用来通信的套接字

void recvFun();//接收信息线程
void sendFun();//发送信息线程
void createConnect();//创建通信套接字

int main()
{
    HANDLE h1, h2;//线程句柄
    createConnect();
    printf("连接成功...
");
    //创建线程后立即运行
    h1 = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)sendFun, NULL, 0, NULL);
    h2 = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)recvFun, NULL, 0, NULL);
    WaitForSingleObject(h1, INFINITE);//会阻塞，直到线程结束
    WaitForSingleObject(h2, INFINITE);//会阻塞，直到线程结束
    closesocket(newSocket);//关闭套接字
    return 0;
}
void sendFun()
{    
    char buf[128];
    while (1)
    {    
        printf("你要发送的信息:");
        scanf("%s", buf);
        send(newSocket, buf, strlen(buf) + 1, 0);//发送信息
    }
}

void recvFun()
{
    char buf[128];
    while (1)
    {
        recv(newSocket, buf, 128, 0);//接收信息
        printf("%32s%s
","接收到的信息: ", buf);
    }
    
}

void createConnect()
{
    SOCKET serverSocket;//监视的套接字
    //SOCKET newSocket;//用来通信的套接字
    SOCKADDR_IN newAddr;//保存客户端的socket地址信息
    SOCKADDR_IN addr;//地址结构体，包括ip port(端口)
    WSADATA data;
    WORD version;//socket版本
    int info;
                 /*
                 在使用socket之前要进行版本的设定和初始化
                 看不懂不用管
                 */
    version = MAKEWORD(2, 2);//设定版本
    info = WSAStartup(version, &data);
    /*应用程序或DLL只能在一次成功的WSAStartup()调用之后
    才能调用进一步的Windows Sockets API函数。
    根据版本初始化 windows socket,返回0表示成功
    */
    if (info != 0)
    {
        printf("初始化失败
");
        return;
    }
    if (LOBYTE(data.wVersion) != 2 || HIBYTE(data.wVersion) != 2)
    {
        printf("加载失败
");
        WSACleanup();
        return;
    }
    //创建套接字，使用TCP协议
    serverSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    addr.sin_addr.S_un.S_addr = htonl(ADDR_ANY);//表示任何的ip过来连接都接受
    addr.sin_family = AF_INET;//使用ipv4的地址
    addr.sin_port = htons(6666);//设定应用占用的端口
    bind(serverSocket, (SOCKADDR*)&addr, sizeof(SOCKADDR));//将套接字与端口6666，接收的ip绑定
    listen(serverSocket, 3);//开始监听，是否有客服端请求连接,最大连接数为3
    printf("开始监听，等待连接..........
");
    int len = sizeof(SOCKADDR);
    newSocket = accept(serverSocket, (SOCKADDR*)&newAddr, &len);
}

3.客户端代码

　　客户端和服务器端代码差不多，也是两个线程用于接收和发送信息。

#include<stdio.h>
#include<string.h>
#include<WinSock2.h>
#pragma comment(lib,"ws2_32.lib")
SOCKET clientSocket;//全局变量，用于通信的socket
void createConnect();//创建套接字
void sendFun();//发送信息线程
void recvFun();//接收信息线程
int main()
{
    HANDLE h1, h2;//线程句柄，其实就是一串数字用来标识线程对象
    createConnect();//创建套接字
    //创建线程后立即执行
    h1 = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)sendFun, NULL, 0, NULL);
    h2 = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)recvFun, NULL, 0, NULL);
    WaitForSingleObject(h1, INFINITE);//会阻塞，直到线程运行结束
    WaitForSingleObject(h2, INFINITE);
    closesocket(clientSocket);//关闭套接字
    return 0;
}
void createConnect()
{
    SOCKADDR_IN addr;
    int info;
    WSADATA data;
    WORD version;
    //设定版本，与初始化
    version = MAKEWORD(2, 2);
    info = WSAStartup(version, &data);
    if (info != 0)
    {
        printf("初始化失败
");
        return;
    }
    if (LOBYTE(data.wVersion) != 2 || HIBYTE(data.wVersion) != 2)
    {
        printf("加载失败
");
        WSACleanup();
        return;
    }
    clientSocket = socket(AF_INET, SOCK_STREAM, 0);//创建套接字
                                                   //要连接的服务器的ip,因为现在服务器端就是本机，所以写本机ip
                                                   //127.0.0.1一个特殊的IP地址，表示是本机的IP地址
    addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    //端口要与服务器相同，不然找不到
    addr.sin_port = htons(6666);
    //用IPV4地址
    addr.sin_family = AF_INET;
    //主动连接服务器
    connect(clientSocket, (SOCKADDR*)&addr, sizeof(SOCKADDR));
}

void sendFun()
{    
    char buf[128]="你是谁";
    while (1)
    {    
        printf("你要发送的信息: ");
        scanf("%s", buf);
        //发送数据
        send(clientSocket, buf, strlen(buf) + 1, 0);
    }    
}


void recvFun()
{    
    char buf[128];
    //接收服务发送的数据
    while (1)
    {
        recv(clientSocket, buf, 128, 0);//接收数据
        printf("%32s%s
","接收的信息为: ", buf);
    }
}

运行结果,可以实现聊天的功能，虽然界面不行。

 学了socket编程还可以实现木马之类的东西。原理很简单，让目标用户在后台悄悄运行你的程序，在另一端你可以发送各种命令操作程序从而实现操作用户的电脑，窃取信息。比如你发送个强制关机的命令的 str="shutdown -s -f"，而目标用户的程序接收到了这个命令并且调用了 system(str)。目标用户的电脑马上就会关机。