飞鸽传书源码分析四-消息发送

转载请注明出处：http://blog.csdn.net/mxway/article/details/44569831
本篇文章是在飞鸽传书源码v2.06的基础上进行分析的
飞鸽传书是一款工作在局域网的软件，支持局域网里不同设备之间的消息发送及文件的传输（消息发送使用udp，文件传输使用tcp）。
发送消息及文件传输是在飞鸽传书的发送对话框中进行，而发送对话框的打开是通过双击拖盘（win7系统）到任务栏的图标。而拖盘到任务栏的这个图标就是飞鸽传书的主窗口，对应的源码就是Mainwin.cpp中的TMainWin类。下面是TMainWin类处理双击事件，鼠标双击事件为什么是由EventButton进行处理可以参见第二篇文章-消息机制

BOOL TMainWin::EventButton(UINT uMsg, int nHitTest, POINTS pos)
{
    switch (uMsg)
    {
         ...
    case WM_LBUTTONDBLCLK:
    case WM_NCLBUTTONDBLCLK:
        if (cfg->OneClickPopup == FALSE)
            SendDlgOpen();
        return  TRUE;
         ...
    }
}

BOOL TMainWin::SendDlgOpen(HWND hRecvWnd, MsgBuf *msg)
{
    TSendDlg *sendDlg;
    ...
    if ((sendDlg = new TSendDlg(msgMng, shareMng, &hosts, cfg, logmng, hRecvWnd, msg)) == NULL)
        return  FALSE;

    sendList.AddObj(sendDlg);
    sendDlg->Create(), sendDlg->Show();
    ...
}
BOOL TDlg::Create(HINSTANCE hInstance)
{
    TApp::AddWin(this);

    if ((hWnd = ::CreateDialog(hInstance ? hInstance : TApp::hI, resId ? (LPCSTR)resId : resName, parent ? parent->hWnd : NULL, (DLGPROC)TApp::WinProc)) == NULL)
        return  TApp::DelWin(this), FALSE;
    else
        return  TRUE;
}

发送消息的对话框类似如下

在发送对话框中在编辑框中输入要发送的内容，在用户列表中选择要发送的对象，点send按钮就可以将消息发送到对方。这里可以选择对发送的消息是否进行加密，为了简化问题这里只分析最简单的消息发送。

一、飞鸽传书消息发送格式
飞鸽传书版本：数据包唯一编号:用户名:机器名称:命令：真实消息
(1)飞鸽传书版本是一个宏定义

#define IPMSG_VERSION           0x0001

(2)数据包唯一编号由飞鸽传书程序自动生成的。
(3)用户名：如果用户没有使用飞鸽传书设置自己的名称，默认使用的是当前登录系统的用户名
(4)机器名称，设备的名称如PC设置的计算机名称
(5)命令，命令是一个32位的无符号数，其功能分为两部分，后8位用于必选的命令，表示当对方收到该命令后要什么事，如用户上线通知，用户退出通知，发送消息。前面24位用于可选命令，如对传送消息进行加密，获取加密的密钥等。
(6)真实消息，这个是真正要发给对方的信息，包括要发送对方的消息及要传送给对方的文件信息。
生成上述格式消息的代码如下

ULONG MsgMng::MakeMsg(char *buf, int _packetNo, ULONG command, const char *msg, const char *exMsg, int *packet_len)
{
    int     len, ex_len = exMsg ? strlen(exMsg) + 1 : 0, max_len = MAX_UDPBUF;

    if (packet_len == NULL)
        packet_len = &len;
        //版本:数据包编号:用户名:设备名称:命令:
    *packet_len = wsprintf(buf, "%d:%ld:%s:%s:%ld:", IPMSG_VERSION, _packetNo, local.userName, local.hostName, command);

    if (ex_len + *packet_len + 1 >= MAX_UDPBUF)
        ex_len = 0;
    max_len -= ex_len;

    if (msg != NULL)    //LocalNewLineToUnix把\r\n转换为\n
        *packet_len += LocalNewLineToUnix(msg, buf + *packet_len, max_len - *packet_len);

    (*packet_len)++;

    if (ex_len)
    {
        //如果有附加消息（如文件发送，同时发送附加的消息）
        memcpy(buf + *packet_len, exMsg, ex_len);
        *packet_len += ex_len;
    }

    return  _packetNo;
}

二、消息发送
单击send按钮的实现代码如下：

BOOL TSendDlg::EvCommand(WORD wNotifyCode, WORD wID, LPARAM hWndCtl)
{
    switch (wID)
    {
    case IDOK:
        ...
        SendMsg();
        ...
    }
}

BOOL TSendDlg::SendMsg(void)
{
    command = IPMSG_SENDMSG|IPMSG_SENDCHECKOPT;

    //获取选中数
    if ((sendEntryNum = (int)SendDlgItemMessage(HOST_LIST, LVM_GETSELECTEDCOUNT, 0, 0)) <= 0 || (sendEntry = new SendEntry [sendEntryNum]) == NULL)
        return  FALSE;

    //获取要发送的消息数据
    GetDlgItemText(SEND_EDIT, msg.msgBuf, MAX_UDPBUF);

    int storeCnt = 0, status = 0, cnt;
    int localStatus = sendEntryNum <= cfg->EncryptNum && (cfg->pubKey.Key() || cfg->smallPubKey.Key()) ? IPMSG_ENCRYPTOPT : 0;

    //获取选中的host信息，host信息是由TMainWin传过来的
    for (cnt=0; cnt < memberCnt && storeCnt < sendEntryNum; cnt++)
    {
        if ((SendDlgItemMessage(HOST_LIST, LVM_GETITEMSTATE, cnt, LVIS_SELECTED) & LVIS_SELECTED) == 0)
            continue;
        char        hostStr[MAX_LISTBUF];
        Host        *host = hostArray[cnt];
        SendEntry   *entry = &sendEntry[storeCnt++];

        //发送的消息要进行加密
        status |= host->hostStatus & IPMSG_ENCRYPTOPT;
        MakeListString(cfg, host, hostStr);
        logmng->WriteSendHead(hostStr);

        entry->SetHost(host);
        entry->SetStatus((localStatus & host->hostStatus) ? host->pubKey.Key() == NULL ? ST_GETCRYPT : ST_MAKECRYPTMSG : ST_MAKEMSG);
        entry->SetCommand(command | (entry->Status() == ST_MAKEMSG ? 0 : IPMSG_ENCRYPTOPT));
    }

    //发送的消息太长进行截断
    msg.msgBuf[MAX_CRYPTLEN] = 0;   
    if (status &= localStatus)  
        command |= IPMSG_ENCRYPTOPT;

    logmng->WriteSendMsg(msg.msgBuf, command, shareInfo);

    if (shareInfo && shareInfo->fileCnt)        // ...\0no:fname:size:mtime:
    {
        //如果选中的文件或文件夹进行传输，生成要传输的文件信息
        char    buf[MAX_UDPBUF / 2];
        EncodeShareMsg(shareInfo, buf, sizeof(buf));
        shareStr = new char [strlen(buf) + 1];
        strcpy(shareStr, buf);
        shareMng->AddHostShare(shareInfo, sendEntry, sendEntryNum);
    }
    //真正进行发送消息的函数
    SendMsgSub();
    return  TRUE;
}

BOOL TSendDlg::SendMsgSub(void)
{
    BOOL    makeNomalMsg = TRUE;    


    for (int cnt=0; cnt < sendEntryNum; cnt++)
    {
        //如果需要从要发送的客户那获取加密的密钥，先获取密钥
        if (sendEntry[cnt].Status() == ST_GETCRYPT) {
            char    spec_str[MAX_BUF];
            int     spec = IPMSG_RSA_512 | IPMSG_RC2_40;

            if (cfg->pubKey.Key())
                spec |= IPMSG_RSA_1024 | IPMSG_BLOWFISH_128;
            wsprintf(spec_str, "%x", spec);
            msgMng->Send(&sendEntry[cnt].Host()->hostSub, IPMSG_GETPUBKEY, spec_str);
        }
        //对要发送的数据进行加密
        if (sendEntry[cnt].Status() == ST_MAKECRYPTMSG) {
            MakeEncryptPacket(sendEntry + cnt);     // ST_MAKECRYPTMSG -> ST_SENDMSG
        }

        if (sendEntry[cnt].Status() == ST_MAKEMSG) {
            sendEntry[cnt].SetStatus(ST_SENDMSG);
            if (makeNomalMsg)
                msgMng->MakeMsg(msgBuf, packetNo, command & ~IPMSG_ENCRYPTOPT, msg.msgBuf, shareStr, &packetLen), makeNomalMsg = FALSE;
        }
        //在MakeEncryptPacket对消息加密完成后设置host状态为ST_SENDMSG
        if (sendEntry[cnt].Status() == ST_SENDMSG) {
            const char  *str = sendEntry[cnt].Msg() ? sendEntry[cnt].Msg() : msgBuf;
            int     len = sendEntry[cnt].Msg() ? sendEntry[cnt].MsgLen() : packetLen;
            //向选中的机器发送数据
            msgMng->UdpSend(sendEntry[cnt].Host()->hostSub.addr, sendEntry[cnt].Host()->hostSub.portNo, str, len);
        }
    }

    return  TRUE;
}

下面是使用wireshark抓取的使用飞鸽传书发送消息的一个数据包。