tcp客户端

// vc_tcpclientDlg.cpp : implementation file
//

#include "stdafx.h"
#include "vc_tcpclient.h"
#include "vc_tcpclientDlg.h"

#pragma comment(lib,"ws2_32.lib")
//#include "winsock2 .h"

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



//0,非阻塞，1，阻塞
#define            ZS                        1
///printf控制台输出使能
#define            LZ_PRINTF_ENABLE        0
///TCP DEBUG  为1，目标IP为局域网IP
#define            LZ_LOCAL_TCP_DEBUG        0
////定时根据域名解析出对应的实时IP地址
#define            LZ_DNS_TIMER_ENABLE        1

#if LZ_PRINTF_ENABLE>0
#pragma comment(linker, "/subsystem:\"CONSOLE\" /entry:\"WinMainCRTStartup\"")
#endif
/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
    CAboutDlg();

// Dialog Data
    //{{AFX_DATA(CAboutDlg)
    enum { IDD = IDD_ABOUTBOX };
    //}}AFX_DATA

    // ClassWizard generated virtual function overrides
    //{{AFX_VIRTUAL(CAboutDlg)
    protected:
    virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
    //}}AFX_VIRTUAL

// Implementation
protected:
    //{{AFX_MSG(CAboutDlg)
    //}}AFX_MSG
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
    //{{AFX_DATA_INIT(CAboutDlg)
    //}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialog::DoDataExchange(pDX);
    //{{AFX_DATA_MAP(CAboutDlg)
    //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
    //{{AFX_MSG_MAP(CAboutDlg)
        // No message handlers
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CVc_tcpclientDlg dialog

CVc_tcpclientDlg::CVc_tcpclientDlg(CWnd* pParent /*=NULL*/)
    : CDialog(CVc_tcpclientDlg::IDD, pParent)
{
    //{{AFX_DATA_INIT(CVc_tcpclientDlg)
    m_strTcpTx = _T("");
    //}}AFX_DATA_INIT
    // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CVc_tcpclientDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialog::DoDataExchange(pDX);
    //{{AFX_DATA_MAP(CVc_tcpclientDlg)
    DDX_Text(pDX, IDC_EDIT_TX, m_strTcpTx);
    //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CVc_tcpclientDlg, CDialog)
    //{{AFX_MSG_MAP(CVc_tcpclientDlg)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_WM_TIMER()
    ON_BN_CLICKED(IDC_BUTTON_CONNECT, OnButtonConnect)
    ON_BN_CLICKED(IDC_BUTTON_SEND, OnButtonSend)
    ON_BN_CLICKED(IDC_BUTTON4, OnButton4)
    ON_BN_CLICKED(IDC_BUTTON_LED1, OnButtonLed1)
    ON_BN_CLICKED(IDC_BUTTON_LED2, OnButtonLed2)
    ON_BN_CLICKED(IDC_BUTTON1, OnButton1)
    ON_MESSAGE(WM_RECEIVE,OnMsgReceive)
    ON_MESSAGE(WM_TRANS,OnTransmit)
    ON_BN_CLICKED(IDC_BUTTON_AUTO_CONNECT, OnButtonAutoConnect)
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CVc_tcpclientDlg message handlers

BOOL CVc_tcpclientDlg::OnInitDialog()
{
    CDialog::OnInitDialog();

    // Add "About..." menu item to system menu.

    // IDM_ABOUTBOX must be in the system command range.
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

    CMenu* pSysMenu = GetSystemMenu(FALSE);
    if (pSysMenu != NULL)
    {
        CString strAboutMenu;
        strAboutMenu.LoadString(IDS_ABOUTBOX);
        if (!strAboutMenu.IsEmpty())
        {
            pSysMenu->AppendMenu(MF_SEPARATOR);
            pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
        }
    }

    // Set the icon for this dialog. The framework does this automatically
    // when the application's main window is not a dialog
    SetIcon(m_hIcon, TRUE);            // Set big icon
    SetIcon(m_hIcon, FALSE);        // Set small icon

    m_bConnected=FALSE;
    char szHostName[]="number007cool.gicp.net";
    InitSocket();
    GetRemoteIP(szHostName);
    InitTimer();
    // TODO: Add extra initialization here

    return TRUE; // return TRUE unless you set the focus to a control
}

void CVc_tcpclientDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
    if ((nID & 0xFFF0) == IDM_ABOUTBOX)
    {
        CAboutDlg dlgAbout;
        dlgAbout.DoModal();
    }
    else
    {
        CDialog::OnSysCommand(nID, lParam);
    }
}

// If you add a minimize button to your dialog, you will need the code below
// to draw the icon. For MFC applications using the document/view model,
// this is automatically done for you by the framework.

void CVc_tcpclientDlg::OnPaint()
{
    if (IsIconic())
    {
        CPaintDC dc(this); // device context for painting

        SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

        // Center icon in client rectangle
        int cxIcon = GetSystemMetrics(SM_CXICON);
        int cyIcon = GetSystemMetrics(SM_CYICON);
        CRect rect;
        GetClientRect(&rect);
        int x = (rect.Width() - cxIcon + 1) / 2;
        int y = (rect.Height() - cyIcon + 1) / 2;

        // Draw the icon
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CDialog::OnPaint();
    }
}

// The system calls this to obtain the cursor to display while the user drags
// the minimized window.
HCURSOR CVc_tcpclientDlg::OnQueryDragIcon()
{
    return (HCURSOR) m_hIcon;
}

int CVc_tcpclientDlg::InitSocket()
{
    WSAData wsadata;
    int wsaret;
    wsaret=WSAStartup(0x0202,&wsadata);
    if (wsaret!=0)
    {
        AfxMessageBox("wsa  start failed");
        return 1;
    }

    unsigned long iMode=1;
    ioctlsocket(m_socket,FIONBIO,&iMode);//非阻塞模式

//    sockaddr_in localaddr;
//    localaddr.sin_port=htons(51000);
//    bind(m_socket,(sockaddr *)&localaddr,sizeof(localaddr));
    return 0;

}

void CVc_tcpclientDlg::OnButtonConnect()
{
    // TODO: Add your control notification handler code here

    GetDlgItem(IDC_BUTTON_CONNECT)->EnableWindow(FALSE);
    GetDlgItem(IDC_STATIC_CONNECT)->SetWindowText("");
    Sleep(500);
    GetDlgItem(IDC_BUTTON_CONNECT)->EnableWindow(TRUE);


    if (m_bConnected==FALSE)
    {
#if ZS==0
        unsigned long iMode=1;
#else
        unsigned long iMode=0;
#endif
        m_socket=::socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
        if(m_socket==INVALID_SOCKET)
        {
            AfxMessageBox("failed socket");
        }
#if LZ_LOCAL_TCP_DEBUG>0
        m_strRemoteIP="192.168.7.111";//test use
#endif
        m_serveraddr.sin_family=AF_INET;
        m_serveraddr.sin_port=htons(51000);
        m_serveraddr.sin_addr.S_un.S_addr=inet_addr(m_strRemoteIP);

        ioctlsocket(m_socket,FIONBIO,(u_long FAR *)&iMode);

        struct fd_set mask;
        FD_ZERO(&mask);
        FD_SET(m_socket,&mask);
        struct timeval timeout;
        timeout.tv_sec=2;
        timeout.tv_usec=0;


#if ZS==0
     if (select(0,&mask,NULL,NULL,&timeout)==1)
     {
#endif
            int bconnect=connect(m_socket,(sockaddr *)&m_serveraddr,sizeof(m_serveraddr));
                //    printf("connect :  %d\r\n",bconnect);
            if ( bconnect==-1 )
            {
                GetDlgItem(IDC_STATIC_CONNECT)->SetWindowText("连接失败");
                m_bConnected=FALSE;
            }
            else
            {
                GetDlgItem(IDC_STATIC_CONNECT)->SetWindowText("连接成功");
                m_bConnected=TRUE;
                GetDlgItem(IDC_BUTTON_CONNECT)->SetWindowText("断开");
                DWORD dwid;

            m_hThread=CreateThread(NULL,1024,(LPTHREAD_START_ROUTINE)ThreadRecvProc,this,0,&dwid);
            }

#if ZS==0
     }
#endif

    }
    else if (m_bConnected==TRUE)
    {
        GetDlgItem(IDC_BUTTON_CONNECT)->SetWindowText("连接");
        closesocket(m_socket);
        m_bConnected=FALSE;
        return;
    }

}
int CVc_tcpclientDlg::GetRemoteIP(char *szName)
{
    HOSTENT *host_entry;
    u_char byhostip[4];
    try
    {
        host_entry=gethostbyname(szName);
    }
    catch (...)
    {
        return 1;
    }

    if (host_entry!=0)
    {
        CString strTemp;
        m_strRemoteIP="";

        for (int i=0;i<4;i++)
        {
            strTemp.Format("%u",(u_char)host_entry->h_addr_list[0][i]);
            m_strRemoteIP+=strTemp;
            if (i!=3)
            {
                m_strRemoteIP+=".";
            }
        }
    ///    m_strRemoteIP="112.10.198.139";
        GetDlgItem(IDC_STATIC_REMOTEIP)->SetWindowText(m_strRemoteIP);
    }

    return 0;
}
////定时器处理函数 /////////////////////////////
void CVc_tcpclientDlg::OnTimer(UINT nIDEvent)
{
    // TODO: Add your message handler code here and/or call default
    LONG ret;
    char szname[]="number007cool.gicp.net";
    switch( nIDEvent)
    {
        case 1:
            try
            {
#if LZ_DNS_TIMER_ENABLE>0
                GetRemoteIP(szname);
#endif
            }
            catch (CException* e)
            {

            }
            break;
        case 2://定时发送读取温度请求

            ret=send(m_socket,"C",1,0);
            printf("send ret is %d\r\n",ret);
            if (ret<0)//根据发送函数返回值判断连接断开
            {
                OnButtonConnect();
            }
            SendMessage(WM_TRANS,0,ret);

            break;
        case 6://定时判断连接是否断开
            if (m_bConnected==FALSE)
            {
                OnButtonConnect();
            }
            else
            {
                OnButton1();//get temp
            }
            break;
        default:
            break;
    }
    CDialog::OnTimer(nIDEvent);
}

int CVc_tcpclientDlg::InitTimer()
{
    SetTimer(1,1000,NULL);


    return 0;
}



void CVc_tcpclientDlg::OnButtonSend()
{
    // TODO: Add your control notification handler code here
    UpdateData(TRUE);
    int ret=send(m_socket,m_strTcpTx,m_strTcpTx.GetLength(),0);
    printf("sned:  %d\r\n",ret);

    ::SendMessage(AfxGetMainWnd()->m_hWnd,WM_TRANS,ret,0);
    if (ret<0)
    {
        m_bConnected=FALSE;
    }

}
////////////////接收线程
int CVc_tcpclientDlg::ThreadRecvProc(LPVOID lparam)
{
    BYTE byrecv[2024];
    int ret;
    static BYTE byflag=0;
    CVc_tcpclientDlg *dlg=(CVc_tcpclientDlg *)lparam;
    CString strtemp;
    CString strrx;
    while (TRUE)
    {
    //    if(!dlg->m_bConnected)
            //break;

        ret=recv(dlg->m_socket,(char *)byrecv,1024,0);
        if (ret>0)
        {

            //for (int i=0;i<ret;i++)
            {
                strtemp.Format("%s",byrecv);
                strrx+=strtemp+"\r\n";
            }
            dlg->GetDlgItem(IDC_EDIT_RX)->SetWindowText(strrx);
            strrx="";
            ::SendMessage(AfxGetMainWnd()->m_hWnd,WM_RECEIVE,0,0);
        }
        Sleep(1);
    }

    return 0;
}
//// 接收灯   显示     消息响应函数
void CVc_tcpclientDlg::OnMsgReceive(WPARAM wParam, LPARAM lParam)
{
    HWND hwnd=::GetDlgItem(m_hWnd,IDC_GREEN);
    if (::IsWindowVisible(hwnd))
    {
        ::ShowWindow(hwnd,SW_HIDE);
    }
    else
    {
        ::ShowWindow(hwnd,SW_SHOW);
    }
}
//// 发送灯   显示      消息响应函数
void CVc_tcpclientDlg::OnTransmit(WPARAM wparam, LPARAM lparam)
{
    if (!m_bConnected)
    {
        return;
    }
    HWND hwnd=::GetDlgItem(m_hWnd,IDC_RED);
    if (lparam>0)//如果发送成功就闪灯
    {
        if (::IsWindowVisible(hwnd))
        {
            ::ShowWindow(hwnd,SW_HIDE);
        }
        else
        {
            ::ShowWindow(hwnd,SW_SHOW);
        }
    }

}

void CVc_tcpclientDlg::OnButton2()
{
    // TODO: Add your control notification handler code here
    static int flag=0;
    if (flag==0)
    {
        flag=1;
        int ret=send(m_socket,"1",1,0);
        SendMessage(WM_TRANS,ret,0);
        GetDlgItem(IDC_BUTTON_LED1)->SetWindowText("灭灯1");
    }
    else
    {
        flag=0;
        int ret=send(m_socket,"2",1,0);
        SendMessage(WM_TRANS,ret,0);
        GetDlgItem(IDC_BUTTON_LED1)->SetWindowText("亮灯1");
    }


}

void CVc_tcpclientDlg::OnButton3()
{
    // TODO: Add your control notification handler code here
    int ret=send(m_socket,"2",1,0);
    SendMessage(WM_TRANS,ret,0);
}

void CVc_tcpclientDlg::OnButton4()
{
    // TODO: Add your control notification handler code here
    int ret=send(m_socket,"0",1,0);
    SendMessage(WM_TRANS,ret,0);

}

void CVc_tcpclientDlg::OnButtonLed1()
{
    // TODO: Add your control notification handler code here
        static int flag=0;
    if (flag==0)
    {
        flag=1;
        int ret=send(m_socket,"1",1,0);
        SendMessage(WM_TRANS,ret,0);
        GetDlgItem(IDC_BUTTON_LED1)->SetWindowText("灭灯1");
    }
    else
    {
        flag=0;
        int ret=send(m_socket,"2",1,0);
        SendMessage(WM_TRANS,ret,0);
        GetDlgItem(IDC_BUTTON_LED1)->SetWindowText("亮灯1");
    }
}

void CVc_tcpclientDlg::OnButtonLed2()
{
    // TODO: Add your control notification handler code here
    static int flag=0;
    if (flag==0)
    {
        flag=1;
        int ret=send(m_socket,"3",1,0);
        SendMessage(WM_TRANS,ret,0);
        GetDlgItem(IDC_BUTTON_LED2)->SetWindowText("灭灯2");
    }
    else
    {
        flag=0;
        int ret=send(m_socket,"4",1,0);
        SendMessage(WM_TRANS,ret,0);
        GetDlgItem(IDC_BUTTON_LED2)->SetWindowText("亮灯2");
    }
}

void CVc_tcpclientDlg::OnButton1()
{
    // TODO: Add your control notification handler code here
    static bool btimer2open=FALSE;
    if (m_bConnected)
    {
        if (btimer2open==FALSE)
        {
            SetTimer(2,11,NULL);
            btimer2open=TRUE;
        }

    }
    else
    {
        KillTimer(2);
        btimer2open=FALSE;
    }
}

void CVc_tcpclientDlg::OnButtonAutoConnect()
{
    // TODO: Add your control notification handler code here
    SetTimer(6,10000,NULL);
}