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 <WINSOCK.H>

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 <iostream>


#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<sData.GetLength();i++)
        readbuffer[2+i] = cAmp[i];
    readbuffer[2+i] = '\n';
    
    SocketSend(readbuffer,100);

    return true;
}




//////////////////////////////////////////////////////////////////////
// Construction/Destruction   
// UDP链接协议 cpp
//////////////////////////////////////////////////////////////////////

CUDPcommunicat::CUDPcommunicat()
{
//  SOCKET m_hSocket;
//  unsigned short m_iPort;
//  CString m_sHostIP;
//  CString m_sHostName;
//  SOCKADDR_IN m_sockAddr; 

    m_sHostIP = "169.254.184.192";
    m_iPort   = 9009;//端口号
    m_nType = 0;//通讯类型  0点发送  1广播
    m_bConnectFlag = false;
}

CUDPcommunicat::~CUDPcommunicat()
{
    Close();
}

bool CUDPcommunicat::Open()
{
    WSADATA wsaData;
//  DWORD dwIPAddr;

    m_bConnectFlag = false;
    if (WSAStartup(WINSOCK_VERSION, &wsaData))
    {
        AfxMessageBox("Could not load Windows Sockets DLL.",NULL,MB_OK);
        return false;
    }
    
    //m_hSocket=socket(AF_INET,SOCK_STREAM,IPPROTO_IP);
    m_hSocket=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP);//数据报的形式
    int nBroatcast = 1;//设置为广播
    setsockopt(m_hSocket,SOL_SOCKET,SO_BROADCAST,(CHAR*)&nBroatcast,sizeof(nBroatcast) );

    m_sockAddr.sin_family=AF_INET;
    m_sockAddr.sin_port=htons(m_iPort);
    m_sockAddr.sin_addr.S_un.S_addr=htonl(INADDR_ANY);//网段内全部发送
    
    //绑定地址
    if(bind( m_hSocket,(SOCKADDR *)&m_sockAddr,sizeof(m_sockAddr))!=0)
    {
        printf("Can't bind socket to local port!Program stop.\n");//初始化失败返回-1
        return false;
    }
            
    m_bConnectFlag = true;
    return true;
}

bool CUDPcommunicat::Close()
{
    try{
        closesocket(m_hSocket); // Closes the socket
        WSACleanup(); //terminates use of the Ws2_32.dll
        m_bConnectFlag = false;
    }
    catch(CException *e)
    {
        delete e;
        return false;
    }
    return true;
}


int CUDPcommunicat::Broadcast(char* ciBuffer,long nLen)
{
    SOCKADDR_IN  addSend;//广播发送地址
    
    addSend.sin_family = AF_INET;
    addSend.sin_addr.s_addr = htonl(INADDR_BROADCAST );
    addSend.sin_port = htons(m_iPort);//发送用的端口，可以根据需要更改
    //nSize = sizeof ( SOCKADDR_IN );}

    return sendto(m_hSocket,ciBuffer,nLen,0,(SOCKADDR*)&addSend,sizeof(SOCKADDR_IN) );
}

int CUDPcommunicat::Receive(char* ciBuffer,long nLen)
{
    SOCKADDR_IN saClient;//收到的客户端地址
    int nSize = sizeof(SOCKADDR_IN);
    int nbSize = 0;
    
    if( (nbSize=recvfrom (m_hSocket,ciBuffer,nLen,0,(SOCKADDR FAR *) &saClient,&nSize))==SOCKET_ERROR )
            return -1;

    return nbSize;
}

//获得链接状态
bool CUDPcommunicat::GetConnectFlag()
{
    return m_bConnectFlag;
}