Bug(1)

程序要求：内网之间客户端截屏(.bmp)并传送给服务端。

server:

#include <winsock2.h>    // 为了使用Winsock API函数
#include <stdio.h>
#include <windows.h>
 #include<fstream>
using namespace std;
 
// 告诉连接器与WS2_32库连接
#pragma comment(lib,"WS2_32.lib")
 
DWORD rem=,num=;
void stranl(char *_str)
{
    char _rem[];
    char _num[];
    int i,j=;
    for(i=;_str[i]!='#';i++)
    {
        _num[i]=_str[i];
    }
    _num[i]='\0';
    i++;
    for(;_str[i]!='\0';i++)
    {
        _rem[j]=_str[i];
        j++;
    }
    _rem[j]='\0';
    num=atoi(_num);
    rem=atoi(_rem);
 
}
int main(int argc, char* argv[])
{
    // 初始化WS2_32.dll
    WSADATA wsaData;
    WORD sockVersion = MAKEWORD(, );
    ::WSAStartup(sockVersion, &wsaData);
 
    // 创建套节字
    SOCKET s = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if(s == INVALID_SOCKET)
    {
        printf("Failed socket() \n");
        ::WSACleanup();
        return ;
    }
 
    // 填充sockaddr_in结构
    sockaddr_in sin;
    sin.sin_family = AF_INET;
    sin.sin_port = htons();
    sin.sin_addr.S_un.S_addr = INADDR_ANY;
 
    // 绑定这个套节字到一个本地地址
    if(::bind(s, (LPSOCKADDR)&sin, sizeof(sin)) == SOCKET_ERROR)
    {
        printf("Failed bind() \n");
        ::WSACleanup();
        return ;
    }
 
    // 进入监听模式
    if(::listen(s, ) == SOCKET_ERROR)
    {
        printf("Failed listen()");
        ::WSACleanup();
        return ;
    }
 
    // 循环接受客户的连接请求
    sockaddr_in remoteAddr;
    int nAddrLen = sizeof(remoteAddr);
    SOCKET client;
    char szCmd[];
    char szNumRem[];
    int i;
    char block[];
    int BlkSize=,allNum=;
    char *allData=NULL;
    DWORD BmpSize=;
    int nSize=;
    while(TRUE)
    {
        // 接受一个新连接
        client = ::accept(s, (SOCKADDR*)&remoteAddr, &nAddrLen);
        if(client == INVALID_SOCKET)
        {
            printf("Failed accept()");
            continue;
        }
 
        printf("接受到一个连接：%s \r\n", inet_ntoa(remoteAddr.sin_addr));
        while()
        {
            printf("请输入命令：\n");
            scanf("%s",szCmd);
            //分析命令
            if(!=strcmp(szCmd,"q"))  //如果不是退出命令，则：
            {
                if(==strcmp(szCmd,"scrshot")) // 如果是截屏命令，则：
                {
                    // 向客户端发送命令，让其做好准备
                    ::send(client, szCmd, strlen(szCmd), );
                    nSize = ::recv(client, szNumRem, , );
                    if(nSize > )
                    {
                         szNumRem[nSize]='\0';
                        //解析字符串
                         printf("字符串:%s,大小:%d\n",szNumRem,nSize);
                         stranl(szNumRem);
                         printf("次数:%d,余数:%d\n",num,rem);
                         BmpSize=num*+rem;
                         allData = new char[BmpSize];
                    }
                    printf("输入开始命令：");
                    scanf("%s",szCmd);
                    ::send(client, szCmd, strlen(szCmd), );
                    for(i=;i<num;i++)
                    {
                        BlkSize = ::recv(client, block, , );
                        //if(BlkSize==1024)
                        if()
                        {
                            memcpy(allData+*i,block,);
                        }
                        else
                        {
                            printf("我擦！\n");
                            system("pause");
                        }
                    }
                    BlkSize = ::recv(client, block, , );
                    if(BlkSize==rem)
                    {
                        memcpy(allData+*i,block,rem);
                    }
                    ofstream of("ClicentScreenShot.bmp",ios_base::binary);
                    of.write(allData,BmpSize);
                    printf("已完全接收图片！\n");
                }
            }
            else
                break;
        }
        // 关闭同客户端的连接
        ::closesocket(client);
     }
 
    // 关闭监听套节字
    ::closesocket(s);
    // 释放WS2_32库
    ::WSACleanup();    
 
    return ;
}

client：

#include <winsock2.h>    // 为了使用Winsock API函数
#include <stdio.h>
#include <windows.h>
#include<fstream>
using namespace std; 
 
// 告诉连接器与WS2_32库连接
#pragma comment(lib,"WS2_32.lib")
 
char *allData=NULL;
DWORD BmpSize=;
int main(int argc, char* argv[])
{
    // 初始化WS2_32.dll
    WSADATA wsaData;
    WORD sockVersion = MAKEWORD(, );
    ::WSAStartup(sockVersion, &wsaData);
 
    // 创建套节字
    SOCKET s = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if(s == INVALID_SOCKET)
    {
        printf("Failed socket() \n");
        ::WSACleanup();
        return ;
    }
 
    // 也可以在这里调用bind函数绑定一个本地地址
    // 否则系统将会自动安排
 
    // 填写远程地址信息
    sockaddr_in servAddr;
    servAddr.sin_family = AF_INET;
    servAddr.sin_port = htons();
    // 注意，这里要填写服务器程序（10ServerDemo程序）所在机器的IP地址
    // 如果你的计算机没有联网，直接使用127.0.0.1即可
    servAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
 
    if(::connect(s, (sockaddr*)&servAddr, sizeof(servAddr)) == -)
    {
        printf("Failed connect() \n");
        ::WSACleanup();
        return ;
    }
 
    // 接收命令
    char szCmdBuff[];
    char str[];
    char tmp[];
    int i,j;
    char block[];
    int nRecv = ::recv(s, szCmdBuff, , );
    if(nRecv > )
    {
        szCmdBuff[nRecv] = '\0';
        printf("接收到命令：%s\n", szCmdBuff);
    }
    //命令处理
     if(==strcmp(szCmdBuff,"scrshot"))
     {
         //截屏，并计算图像大小
         _ScreenSnap();
         //每次发送1024字节，计算发送次数及剩余字节数
         //格式： 次数#余数
         DWORD rem=BmpSize%;
         DWORD num=BmpSize/;
         itoa(num,str,);
         strcat(str,"#");
         itoa(rem,tmp,);
         strcat(str,tmp);
         printf("字符串：%s\n",str);
         ::send(s, str, strlen(str), );
         ::recv(s, szCmdBuff, , );
         if(==strcmp(szCmdBuff,"start"))
         {
            for(i=;i<num;i++)
            {
                for(j=;j<;j++)
                    block[j]=allData[i*+j];
                ::send(s,block,,);
 
             }
            for(j=;j<rem;j++)
                    block[j]=allData[num*+j];
             ::send(s,block,num,);
         }
     }
    // 关闭套节字
    ::closesocket(s);
    // 释放WS2_32库
    ::WSACleanup();
    return ;
}

结果：