QT 下 QUdpSocket 加入指定源組播(SSM)的方法

QT下的類 QUdpSocket 僅提供了加入組播的功能 joinMulticastGroup，卻沒有提供加入SSM的功能。
分析了一下QUdpSocket的源碼，發現joinMulticastGroup僅是socket函數 setsocketoption的一個簡單封裝，隱藏了不同系統下實現的不同。於是參考了QT的源碼，自己寫了一個加入SSM的功能，代碼見下面。目前只支持window下運行。linux下的實現基本類似，區別僅是包含的頭文件不同 ，有時間再補上linux下的實現。

 

#if defined(Q_OS_WIN32)

// 注意不可以在<winsock2.h>之前包含"windows.h"，否則會導入"winsock.h"，里面宏的定義不一致
//#include "windows.h"
#include <winsock2.h>
#include <ws2tcpip.h>

// 添加指定源組播地址
int AddSourceMembership(QUdpSocket& socket, string groupIP, string localIP, vector<string>& groupSourceList)
{
    // 將字符串形式的IP轉換為整數
    auto ConvertIP = [](string& addr){ return htonl(QHostAddress(QString::fromStdString(addr)).toIPv4Address()); };
    ip_mreq_source mcast;
    mcast.imr_interface.S_un.S_addr = ConvertIP(localIP);
    mcast.imr_multiaddr.S_un.S_addr = ConvertIP(groupIP);

    int ret;
    for(string source : groupSourceList)
    {   // 多個組播源依次加入
        if(source.empty()) continue;
        mcast.imr_sourceaddr.S_un.S_addr = ConvertIP(source);
        ret = setsockopt(socket.socketDescriptor(), IPPROTO_IP, IP_ADD_SOURCE_MEMBERSHIP, (char*)&mcast, sizeof(mcast));
    }
    return ret;
}

#elif  defined(Q_OS_LINUX)

#endif