Та же программа API сокетов BSD работает в Python, но не в C ++ или Rust.

Question

Я пытался прослушивать многоадресные UDP-сообщения от моей умной лампы Yeelight. Через регулярный интервал лампочка сообщает о своем присутствии на IP 239.255.255.250 порт 1982. Кроме того, при отправке специального c запроса ( на странице 5 - раздел 3.1 ) на тот же IP-адрес и порт лампочка ответит.

Используя сокеты, я пытаюсь устанавливает sh связь. На двух машинах (macOS и Linux) программа Python, показанная ниже, запущенная в REPL, работает, но код Rust и код C ++, показанные ниже (по сути, выполняющие то же самое), не работают. Кажется, что программа Rust дважды успешно получает 0 байт и затем зависает в ожидании. В программе на C ++ функция recv сразу зависает, не возвращаясь.

В чем может быть проблема? Почему только Python может успешно общаться?

import socket as s
import struct

sock = s.socket(s.AF_INET, s.SOCK_DGRAM, s.IPPROTO_UDP)

sock.setsockopt(s.IPPROTO_IP, s.IP_ADD_MEMBERSHIP, struct.pack("4sL", s.inet_aton("239.255.255.250"), s.INADDR_ANY))

sock.sendto("M-SEARCH * HTTP/1.1\r\nHOST: 239.255.255.250:1982\r\nMAN: \"ssdp:discover\"\r\nST: wifi_bulb\r\n".encode('UTF-8'), ("239.255.255.250", 1982))

sock.recv(4096) # Message received successfully right away from first call

#[macro_use] extern crate log;
use socket2::{Socket, Domain, Type, Protocol, SockAddr};
use std::net::{SocketAddrV4, Ipv4Addr};

fn main() {
    env_logger::init();

    let yeelight_ip = Ipv4Addr::new(239, 255,255, 250);
    let yeelight_port: u16 = 1982;

    let socket = Socket::new(
        Domain::ipv4(), 
        Type::dgram(), 
        Some(Protocol::udp())
    ).expect("Failed to create socket!");

    socket.join_multicast_v4(
        &yeelight_ip, 
        &Ipv4Addr::UNSPECIFIED
    ).expect("Unable to join multicast broadcast!");

    let msg = "M-SEARCH * HTTP/1.1\r\nHOST: 239.255.255.250:1982\r\nMAN: \"ssdp:discover\"\r\nST: wifi_bulb\r\n";
    match socket.send_to(
        msg.as_bytes(), 
        &SockAddr::from(
            SocketAddrV4::new(
                yeelight_ip, 
                yeelight_port
            )
        )
    ) {
        Ok(bytes_sent) => {
            // Some lines of debug printing
        },
        Err(_) => eprintln!("Error broadcasting request for identification!")

    loop {
        let mut buffer = Vec::with_capacity(1024 * 1024);
        let received_bytes = socket
            .recv(&mut buffer)
            .expect("Unable to receive message!");
        debug!("Received {} bytes", received_bytes);
    }
}

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <iostream>
#include <string>
#include <array>

const std::string msg = "M-SEARCH * HTTP/1.1\r\nHOST: 239.255.255.250:1982\r\nMAN: \"ssdp:discover\"\r\nST: wifi_bulb\r\n";

int main() {
    auto yeelight_ip = in_addr();
    inet_aton("239.255.255.250", &yeelight_ip);

    unsigned int yeelight_port = 1982;

    auto sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
    std::cerr << "Created socket!" << std::endl;

    auto mreq = ip_mreq();
    mreq.imr_interface.s_addr = INADDR_ANY;
    mreq.imr_multiaddr.s_addr = yeelight_ip.s_addr;

    if (setsockopt(sock, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq)) != 0) {
        std::cerr << "Error joining multicast!" << std::endl;
        exit(errno);
    }
    else {
        std::cerr << "Joined multicast!" << std::endl;
    }

    auto addr = sockaddr_in();
    addr.sin_port = yeelight_port;
    for (int i = 0; i < 8; i++) addr.sin_zero[i] = 0;
    addr.sin_family = AF_INET;
    addr.sin_addr = yeelight_ip;

    if (sendto(sock, msg.c_str(), msg.length(), 0, (sockaddr*)(&addr), sizeof(addr)) == -1) {
        std::cerr << "Error broadcasting request for identification!" << std::endl;
        std::cerr << errno << std::endl;
    }
    else {
        std::cerr << "Sent broadcast request for idetification!" << std::endl;
    }

    while (true) {
        std::array<char, 1024 * 1024> buffer;
        auto bytes_read = recv(sock, buffer.data(), buffer.size(), 0);
        if (bytes_read == -1) {
            std::cerr << "Unable to receive message!" << std::endl;
            std::cerr << errno << std::endl;
        }
        else {
            std::cerr << "Read " << bytes_read << " bytes..." << std::endl;
            std::cout << std::string(buffer.begin(), buffer.begin() + bytes_read) << std::endl;
        }
    }

    return 0;
}

Answer 1

Программа Python работает должным образом, поскольку Python прекрасно выполняет преобразования «за кулисами».

Программа C (++) не работает, поскольку порядок байтов порта был неправильным . Я забыл, что мне нужно использовать addr.sin_port = htons(yeelight_port);.

Что касается программы Rust, я забыл, что with_capacity выделил только внутренний буфер, но Vec все еще пуст. Решение, описанное здесь , - позвонить по номеру buffer.resize(buffer.capacity(), 0);.