2012-01-13 38 views
0

這裏接收UDP數據包是我迄今(剝離錯誤檢查):發送和從同一進程

struct sockaddr_in addr, ss, dest; 
int port, s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); 
fcntl(s, F_SETFL, fcntl(s, F_GETFL, 0) | O_NONBLOCK); 

memset((char*) &addr, 0, sizeof(addr)); 
addr.sin_family = AF_INET; 
inet_aton("127.0.0.1", &addr.sin_addr); 
bind(s, (struct sockaddr*) &addr, sizeof(addr)); 

unsigned int len = sizeof(ss); 
getsockname(s, (struct sockaddr*) &ss, &len); 
port = ss.sin_port; 

memset((char*) &dest, 0, sizeof(dest)); 
dest.sin_family = AF_INET; 
dest.sin_port = htons(port); 
inet_aton("127.0.0.1", &dest.sin_addr); 
sendto(s, "test", 5, 0, (struct sockaddr*) &dest, sizeof(dest)); 

char buf[5]; 
recv(s, buf, 5, 0); 

最後一句失敗的Resource temporarily unavailable消息(因爲O_NONBLOCK標誌)。

在代碼片段中,讓操作系統綁定一個隨機端口,然後用getsockname獲得它。如果我使用一個固定的端口,並刪除呼叫getsockname然後它的工作。 PS:我在一臺Linux機器上。

回答

2

你忘了使用ntohs當你拍攝的OS分配的端口。這就是我最後的結局(我做了一些小事情來使代碼更加簡潔):

#include <netinet/in.h> 
#include <stdio.h> 
#include <errno.h> 
#include <string.h> 

int main() 
{ 
    struct sockaddr_in addr = {}, ss, dest = {}; 
    int port, s = socket(AF_INET, SOCK_DGRAM | SOCK_NONBLOCK, 0); 

    addr.sin_family = AF_INET; 
    addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK); 
    bind(s, (struct sockaddr*) &addr, sizeof(addr)); 

    unsigned int len = sizeof(ss); 
    getsockname(s, (struct sockaddr*) &ss, &len); 
    port = ntohs(ss.sin_port); 

    dest.sin_family = AF_INET; 
    dest.sin_port = htons(port); 
    dest.sin_addr.s_addr = htonl(INADDR_LOOPBACK); 
    sendto(s, "test", 5, 0, (struct sockaddr*) &dest, sizeof(dest)); 

    char buf[5]; 
    int got = recv(s, buf, 5, 0); 

    printf("got: %d, errno: %s\n", got, strerror(errno)); 

    return 0; 
} 
4

port = ss.sin_port應該給網絡有序的端口號。當您分配端口dest.sin_port = htons(port)時,您正在將htons()應用於已處於網絡字節順序的短路。改爲使用dest.sin_port = port,一切都應該沒問題。

另外,如果你想從getsockname()結果獲得一臺主機有序的端口號,你應該使用ntohs()

getsockname(s, (struct sockaddr*) &ss, &len); 
port = ntohs(ss.sin_port); 
/*...*/ 
dest.sin_port = htons(port);