网络编程：select模型的简单实例

#include <stdio.h>
#include <string>
#include <stdlib.h>
#include "QtCore/QCoreApplication"
#include "Qt/qstring.h"
#include "Qt/qstringlist.h"
#include <cstdlib>
#define FD_SETSIZE 1024
#pragma comment(lib, "WS2_32")
#include <Windows.h>
using namespace std;  
#pragma comment(linker, "/subsystem:\"CONSOLE\" /entry:\"WinMainCRTStartup\"")

int main(int argc, char** argv)  
{      
    QCoreApplication app(argc, argv);
    string confName = ".\\conf.ini";
    WORD wVersionRequested;  
    WSADATA wsaData;       
    wVersionRequested = MAKEWORD(1, 1);  
    int err = WSAStartup( wVersionRequested, &wsaData );  
    if ( err != 0 ) {  
        return -1;          
    }  
    if ( LOBYTE( wsaData.wVersion ) != 1 || HIBYTE( wsaData.wVersion ) != 1 ) {          
        WSACleanup( );  
        return -1;   
    }  
    SOCKET sockSrv = socket(AF_INET, SOCK_STREAM, 0);  
    if (sockSrv == INVALID_SOCKET)
    {
        printf("socket() called failed\n");
        return 0;
    }
    SOCKADDR_IN addrSrv;  
    addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
    addrSrv.sin_family = AF_INET;  
    int socket_port = GetPrivateProfileIntA("Main", "Socket_port", -1, confName.c_str());
    addrSrv.sin_port = htons(socket_port);  
    if (bind(sockSrv, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR)) == SOCKET_ERROR)
    {
        printf("bind() called failed\n");
        return 0;
    }
    if (listen(sockSrv, 10) == SOCKET_ERROR)
    {
        printf("listen() called failed\n");
        return 0;
    }             
    int len = sizeof(SOCKADDR);    
    printf("等待客户端连接\n");
    ULONG NonBlock = 1;   
    if(ioctlsocket(sockSrv, FIONBIO, &NonBlock) == SOCKET_ERROR)   
    {   
        printf("ioctlsocket() failed with error %d\n", WSAGetLastError());
        return -1;   
    }     
    struct timeval timeout={2,0};   
    fd_set fdSocket;
    fd_set fdRead;
    FD_ZERO(&fdSocket);
    FD_SET(sockSrv, &fdSocket);

    while (1)
    {    
        FD_ZERO(&fdRead);
        fdRead = fdSocket;
        int nRet = select(0,&fdRead,(fd_set*) 0,(fd_set*) 0,&timeout);         
        if(nRet>0)
        {
            for(int i=0; i<(int)fdSocket.fd_count; i++)
            {
                if(FD_ISSET(fdSocket.fd_array[i], &fdRead))
                {
                    if(fdSocket.fd_array[i] == sockSrv)         
                    {
                        if(fdSocket.fd_count < FD_SETSIZE)
                        {  
                            sockaddr_in addrRemote;
                            int nAddrLen = sizeof(addrRemote);
                            SOCKET sockConn = accept(sockSrv, (SOCKADDR*)&addrRemote, &nAddrLen);
                            if (INVALID_SOCKET == sockConn)
                            {
                                printf("accept() called failed\n");
                                continue;
                            }
                            FD_SET(sockConn, &fdSocket);
                            printf("接收到连接：%s\n", ::inet_ntoa(addrRemote.sin_addr));
                        }
                        else
                        {
                            printf(" Too much connections! \n");
                            continue;
                        }
                    }
                    else
                    {    
                        char recvBuf[8];                       
                        int ret = recv(fdSocket.fd_array[i], recvBuf, 8, 0);  
                        if (ret <= 0)
                        {     
                             printf("ret=%d,客户端断开\n", ret);
                             closesocket(fdSocket.fd_array[i]);
                             FD_CLR(fdSocket.fd_array[i], &fdSocket);                           
                        }        
                        else
                        {    
                            char temp[3];
                            memcpy(temp, recvBuf+2, 2);
                            temp[2] = 0;  
                            QString taskID;
                            if (strncmp(temp, "01", 2) == 0)//创建任务
                            {
                                int length;
                                char contentLen[4];
                                memcpy(contentLen, recvBuf+4, 4);
                                sscanf(contentLen, "%d", &length);
                                char* str = new char[length+1];
                                int size = 0;
                                int ret;
                                while(1)
                                {
                                    ret = recv(fdSocket.fd_array[i], str+size, length-size, 0);
                                    if (ret > 0)
                                    {
                                        size += ret;
                                        if (size == length)
                                        {
                                            break;
                                        }
                                    }                              
                                }
                                str[length] = 0;
                                printf("服务器收到创建消息：%s\n", str);
                                Sleep(1000);//模拟解析处理
                                QString content = str;
                                taskID = content.split("|").at(0);
                                //响应创建
                                int port = GetPrivateProfileIntA("Main", "port", -1, confName.c_str());
                                QString portTo = QString::number(port);
                                QString res = "010000060|";
                                res += portTo;
                                char data[14];
                                memcpy(data, res.toStdString().c_str(), 14);
                                int r = send(fdSocket.fd_array[i], data, 14, 0);    
                                printf("服务器响应创建消息：%s\n", res.toStdString().c_str());
                                Sleep(1000);//模拟视频分析
                                //发送停止消息
                                int stopLen = taskID.size();
                                char idLen[4];
                                sprintf(idLen, "%04d", stopLen);
                                res = "0102";   
                                res += idLen;
                                res += taskID;
                                send(fdSocket.fd_array[i], res.toStdString().c_str(), res.size(), 0);
                                printf("服务器发送停止消息：%s\n", res.toStdString().c_str());
                                delete[] str;
                                str = NULL;
                            
                           
                        }    
                        else if (strncmp(temp, "02", 2) == 0)//停止任务
                        {
                            int length;
                            char contentLen[4];
                            memcpy(contentLen, recvBuf+4, 4);
                            sscanf(contentLen, "%d", &length);
                            char* str = new char[length+1];
                            int size = 0;
                            int ret;
                            while(1)
                            {
                                ret = recv(fdSocket.fd_array[i], str+size, length-size, 0);
                                if (ret > 0)
                                {
                                    size += ret;
                                    if (size == length)
                                    {
                                        break;
                                    }
                                }                              
                            }
                            str[length] = 0;
                            printf("服务器收到停止消息：%s\n", str);
                            Sleep(1000);//模拟停止
                            //响应停止
                            send(fdSocket.fd_array[i], "010000010", 9, 0);
                            printf("服务器响应停止消息：%s\n", "010000010");
                            delete[] str;
                            str = NULL;
                        }
                               
                    }
                }
            }  

        }        
      }
        else if(nRet == SOCKET_ERROR)
        {
            printf("select出错\n");
            break;
        }
        else
        {
            continue;
        }
    }
    printf("程序结束\n");
    closesocket(sockSrv);
    WSACleanup();  
    return 0;  
}