从同一进程发送和接收UDP数据包

时间:2012-01-13 03:10:20

标签: c sockets

这是我到目前为止(剥离错误检查):

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 个答案:

答案 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);

答案 1 :(得分:2)

您在捕获操作系统分配的端口时忘记使用ntohs。这就是我最终得到的结果(我做了一些小事来使代码更加简洁):

#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;
}