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