SokcetClient VC++6.0

// SokcetClient.cpp: implementation of the SokcetClient class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "SokcetClient.h"
#include <iostream>
#include <Ws2tcpip.h>
#include "StringHelper.h"
#include "LoginResponse.h"
#include "HeartBeatRequest.h"
#include <string.h>
#include <stdio.h>
#include <Windows.h>
#include <map>
#include <cctype>
#include <algorithm>
#include "MyDefine.h"
#include "MD5.h"
#include <functional>
#include "StringHelper.h"
#include "JsonHelper.h"
#include <process.h>
#include <list>

using namespace std;

static bool IsCloseSocket = false;
static SOCKET m_nLocalSocket;
static HANDLE  reciveHandel;
static HANDLE  heartbeatHandel;
static string IP;
static int Port;
static string ClientNo;
static string ClientName;
static string Key;
static string ParkingNo;
static string ParkingName;
static IPaySDKAPI* paySDKAPI;
static int nRet = SOCKET_ERROR;

typedef list<string> ListString;
static ListString listJson;

SokcetClient::SokcetClient()
{
    m_nLocalSocket = -;
    WSADATA wsaData;
    if (WSAStartup(MAKEWORD(, ), &wsaData) != ) {
        std::cout << "Socket版本加载失败" << std::endl;
    }
}

SokcetClient::~SokcetClient()
{
    closeSocket();
}

void SokcetClient::closeSocket()
{
    IsCloseSocket = true;
    if (m_nLocalSocket != -){
        closesocket(m_nLocalSocket);    //关闭socket连接
    }
    m_nLocalSocket = -;
    CloseHandle(reciveHandel);
    CloseHandle(heartbeatHandel);
    WSACleanup();    //终止ws2_32.lib的使用

}

DWORD _stdcall SokcetClient::Myrecv(LPVOID lpParameter)
{

    int rs = -;
    int resultRecv = -;
    fd_set rfds;
    char m_message[];
    int size = sizeof(m_message);

    while(nRet == SOCKET_ERROR)
    {
        sockaddr_in m_nServeraddr;
        memset(&m_nServeraddr, , sizeof(m_nServeraddr));
        m_nServeraddr.sin_family = AF_INET;
        m_nServeraddr.sin_port = htons(Port);
        m_nServeraddr.sin_addr.s_addr = inet_addr(IP.c_str());
        nRet = connect(m_nLocalSocket, (sockaddr*)&m_nServeraddr, sizeof(m_nServeraddr));
        if (nRet == SOCKET_ERROR)
        {
            Sleep();
        }
    }
    while (false == IsCloseSocket)
    {

        memset(m_message, '\0', size);
        FD_ZERO(&rfds);
        FD_SET(m_nLocalSocket, &rfds);
        struct timeval tv;
        tv.tv_sec = ;
        tv.tv_usec = ;
        rs = select(m_nLocalSocket, &rfds, NULL, NULL, &tv);
        if (rs > )
        {
            resultRecv = recv(m_nLocalSocket, m_message, size, );

            if (resultRecv > )
            {
                //LoginResponse loginResponse;

                //输出消息
                paySDKAPI->receive(m_message);

                memset(m_message, '\0', size);

            }
            else
            {
                //这几种错误码，认为连接是正常的，继续接收
                if ((resultRecv < ) && (errno == EAGAIN || errno == EINTR))
                {
                    continue;//继续接收数据
                }
                else
                {
                    m_nLocalSocket = -;
                    CloseHandle(reciveHandel);
                    CloseHandle(heartbeatHandel);

                    m_nLocalSocket = socket(AF_INET, SOCK_STREAM , IPPROTO_TCP);
                    if (m_nLocalSocket != INVALID_SOCKET)
                    {
                        int nRet = SOCKET_ERROR;
                    }
                    if (m_nLocalSocket != -)
                    {
                        sockaddr_in m_nServeraddr;
                        memset(&m_nServeraddr, , sizeof(m_nServeraddr));
                        m_nServeraddr.sin_family = AF_INET;
                        m_nServeraddr.sin_port = htons(Port);
                        m_nServeraddr.sin_addr.s_addr = inet_addr(IP.c_str());
                        nRet = connect(m_nLocalSocket, (sockaddr*)&m_nServeraddr, sizeof(m_nServeraddr));
                    }
                    if (nRet == SOCKET_ERROR)
                    {
                        Sleep();
                    }
                }
            }
        }
        else if ( == rs )
        {
            if(listJson.size() >  && m_nLocalSocket != - && nRet != SOCKET_ERROR)
            {
                string data = listJson.front();
                int size = strlen(data.c_str());
                listJson.pop_front();
                send(m_nLocalSocket, data.c_str(), size, );
            }
        }

    }
    return ;
}    

DWORD _stdcall SokcetClient::MyHeartBeat(LPVOID lpParameter)
{
    while (nRet == SOCKET_ERROR)
    {
        Sleep();
    }
    while (false == IsCloseSocket)
    {
        Sleep(*);
        if (m_nLocalSocket != -)
        {
            outputMessage(StringHelper::GBKToUTF8("发送心跳>>>>>>").c_str());
            HeartBeatRequest heartBeatRequest;
            heartBeatRequest.bizCode = "B1005";
            heartBeatRequest.clientName = StringHelper::GBKToUTF8("中心服务器");
            heartBeatRequest.clientNo = "";
            heartBeatRequest.parkingNo = "";
            heartBeatRequest.reqNo = "";
            heartBeatRequest.sign = "";

            map<std::string, std::string> myMap;
            myMap.insert(map<string,string>::value_type("bizCode",heartBeatRequest.bizCode));
            myMap.insert(map<string,string>::value_type("clientName",heartBeatRequest.clientName));
            myMap.insert(map<string,string>::value_type("clientNo",heartBeatRequest.clientNo));
            myMap.insert(map<string,string>::value_type("parkingNo",heartBeatRequest.parkingNo));
            myMap.insert(map<string,string>::value_type("reqNo",heartBeatRequest.reqNo));
            myMap.insert(map<string,string>::value_type("sign",heartBeatRequest.sign));

            map<string, string>::iterator iter;
            string str = "";
            string key = "";
            for (iter = myMap.begin(); iter != myMap.end(); ++iter) {
                if (!(*iter).second.empty() && (*iter).second.length() > ) {
                    key = (*iter).first;
                    transform(key.begin(), key.end(), key.begin(), tolower);
                    str = str + key + "=" + (*iter).second + "&";
                }
            }

            str = str + "key=909276b944077b0b2123ac89f69d1016";

            MD5 md5;
            md5.update(str);
            heartBeatRequest.sign = md5.toString();
            transform(heartBeatRequest.sign.begin(), heartBeatRequest.sign.end(), heartBeatRequest.sign.begin(), toupper);
            if(myMap.count("sign") != ){
                myMap.erase("sign");
            }
            myMap.insert(map<string,string>::value_type("sign",heartBeatRequest.sign));

            string json2 = JsonHelper::map2jsonstr(myMap);
            json2= json2+"\n";
            std::cout << json2 << endl;
            if (m_nLocalSocket != -)
            {
                if(listJson.size() < )
                {
                    listJson.insert(listJson.begin(),json2);
                }
            }

        }
    }
    return ;
}

bool SokcetClient::ConnectSocket(IPaySDKAPI* iPaySDKAPI,string ip, int prot,string clientNo,string clientName,string parkingNo, string key)
{
    paySDKAPI = iPaySDKAPI;
    IsCloseSocket = false;
    IP = ip;
    Port = prot;
    if (m_nLocalSocket == -)
    {    

        m_nLocalSocket = socket(AF_INET, SOCK_STREAM , IPPROTO_TCP);
        if (m_nLocalSocket != INVALID_SOCKET)
        {
            outputMessage(StringHelper::GBKToUTF8("客服端socket创建成功").c_str());

        }

    }

    reciveHandel = CreateThread(NULL,,Myrecv,NULL,,NULL);
    heartbeatHandel = CreateThread(NULL,,MyHeartBeat,NULL,,NULL);
    return true;

}

void SokcetClient::Mysend(const char* buffer)
{
    if(listJson.size() < )
    {
        listJson.insert(listJson.begin(),buffer);
    }
}

void SokcetClient::outputMessage(const char * outstr)
{
    std::cout << outstr << std::endl;
}