CCommunicat


TCPcommunicat.h

// TCPcommunicat.h: interface for the CTCPcommunicat class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_TCPCOMMUNICAT_H__F903C55C_4E75_4F30_BFC3_D44A969C6C2D__INCLUDED_)
#define AFX_TCPCOMMUNICAT_H__F903C55C_4E75_4F30_BFC3_D44A969C6C2D__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
#include 

class CTCPcommunicat  
{
public:
    CTCPcommunicat();
    virtual ~CTCPcommunicat();

    bool OpenConnect();
    void CloseConnect();
    int SocketRead(char* ciBuffer,long nLen);
    bool SocketSend(char* sSendMsg,int nLen);


    void SetIP(CString sIP){ m_sHostIP= sIP;};
    CString GetIP(){return m_sHostIP;};
    void    SetPort(int nPort){m_iPort = nPort;};
    int GetPort(){ return m_iPort;};
    bool SetParamnetValue(int nType,double dValue);

    bool    m_bConnectSuccess;
private:
    void ReportWinsockErr(LPSTR lpszErrorMsg);


private:
    SOCKET m_hSocket;
    unsigned short m_iPort;
    CString m_sHostIP;
    CString m_sHostName;
    SOCKADDR_IN m_sockAddr;

};



/****************************************************
*              UDP协议类
*
*  使用UDP实现广播,因此在该网络上每台监听的设备都能收到
*
*  UDP协议不仅负责 开始、停止命令的发送,还需要发送开始采集和停止采集的命令(采集需要同步采集)
*
******************************************************/

class CUDPcommunicat
{
public:
    bool GetConnectFlag();
    int Receive(char* ciBuffer,long nLen);
    CUDPcommunicat();
    virtual ~CUDPcommunicat();

    bool Open();
    bool Close();
    
    int Broadcast(char* ciBuffer,long nLen);
    void SetPort(int nPort){m_iPort = nPort;};
    int GetPort(){return m_iPort;}; 
    void SetServerIP(CString sIP){ m_sHostIP= sIP;};

private:
    SOCKET m_hSocket;
    unsigned short m_iPort;
    CString m_sHostIP;
    CString m_sHostName;
    SOCKADDR_IN m_sockAddr;
        
    bool m_bConnectFlag;//链接成功标志

    int  m_nType;//通讯类型  0点发送  1广播
};


#endif // !defined(AFX_TCPCOMMUNICAT_H__F903C55C_4E75_4F30_BFC3_D44A969C6C2D__INCLUDED_)


TCPcommunicat.CPP

// TCPcommunicat.cpp: implementation of the CTCPcommunicat class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "TCPcommunicat.h"
#include 


#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CTCPcommunicat::CTCPcommunicat()
{
    m_sHostIP = "169.254.184.192";

//  m_sHostIP = "169.254.209.199";
    m_iPort   = 9005;

    m_bConnectSuccess = false;
}

CTCPcommunicat::~CTCPcommunicat()
{
    CloseConnect();
}

void CTCPcommunicat::ReportWinsockErr(LPSTR lpszErrorMsg)
{
    char m_chMsgBuffer[100];
    wsprintf(m_chMsgBuffer,"\nWinsock error %d: %s\n\n", WSAGetLastError(), lpszErrorMsg);
    MessageBeep(MB_ICONSTOP);
    CString sContent(m_chMsgBuffer);
    AfxMessageBox(sContent, NULL, MB_OK | MB_ICONSTOP);
    return;
}


void CTCPcommunicat::CloseConnect()
{
    try{
        closesocket(m_hSocket); // Closes the socket
        WSACleanup(); //terminates use of the Ws2_32.dll
        m_bConnectSuccess = false;
    }
    catch(CException *e)
    {
        delete e;
        throw;
    }
}


bool CTCPcommunicat::OpenConnect()
{
    WSADATA wsaData;
    DWORD dwIPAddr;

    m_bConnectSuccess = false;
    if (WSAStartup(WINSOCK_VERSION, &wsaData))
    {
        AfxMessageBox("Could not load Windows Sockets DLL.",NULL,MB_OK);
        return false;
    }
    
    if((dwIPAddr=inet_addr(m_sHostIP))==INADDR_NONE)
    {
        AfxMessageBox("IPAddress is error!\nPlease input again!",NULL,MB_OK);
        return false;
    }
    else
    {
        m_hSocket=socket(AF_INET,SOCK_STREAM,IPPROTO_IP);
        m_sockAddr.sin_family=AF_INET;
        m_sockAddr.sin_port=htons(m_iPort);
        m_sockAddr.sin_addr.S_un.S_addr=dwIPAddr;
        int nConnect=connect(m_hSocket,(LPSOCKADDR)&m_sockAddr,sizeof(m_sockAddr));
        
        if(nConnect)
        {
            ReportWinsockErr("Connect is error!!");
            return false;
        }
        //else
        //  AfxMessageBox("Successfully connected Server!!",NULL,MB_OK);

        fd_set fds;
        FD_ZERO(&fds);
        timeval tv;
        tv.tv_sec = 5;
        tv.tv_usec = 0;

        // wait for permission to send
        FD_SET(m_hSocket, &fds);
        int i = select(32, NULL, &fds, NULL, &tv); // write
        if (i<=0)
        {
            printf("select - error %d\n",WSAGetLastError());
            return false;
        }
    }
    m_bConnectSuccess = true;
    return true;
}

int CTCPcommunicat::SocketRead(char* ciBuffer,long nLen)
{
    if(nLen < 0) return -1;
    
    fd_set fds;
    FD_ZERO(&fds);
    timeval tv;
    tv.tv_sec = 5;
    tv.tv_usec = 0;
    FD_SET(m_hSocket, &fds);
    int i = select(32, &fds, NULL, NULL, &tv); 
    if (i<=0)
    {
        printf("no TCP response received\n");
        return -1;
    }

    int nResult = recv(m_hSocket, ciBuffer, nLen, 0); // returns the number of bytes received
    return nResult;
}

bool CTCPcommunicat::SocketSend(char* sSendMsg,int nLen)
{
    if(nLen < 1) return true;

    int nCharSend=send(m_hSocket,sSendMsg,nLen,0);//0,MSG_OOB
    if(nCharSend==SOCKET_ERROR)
    {
        //AfxMessageBox("Error occurred during send!!",NULL,MB_OK);
        //如果发生错误,清空缓冲区,重新链接
        return false;
    }
    return true;
}

bool CTCPcommunicat::SetParamnetValue(int nType,double dValue)
{
    char readbuffer[100];
    readbuffer[0] = 40;
    readbuffer[1] = nType;

    CString sData;
    sData.Format("%.2f",dValue);
    char  *cAmp;
    cAmp = sData.GetBuffer(sData.GetLength());
    int i;
    for(i=0;i

你可能感兴趣的:(CCommunicat)