C套接字重定向

Question

我有两个套接字文件描述符，a和b，它们由socket函数返回。问题是:如何才能将从 a 读取的任何内容写入到 b，但是，从 b 读取的任何内容都将被写入写入a。这种情况有点像代理。您能给我一些想法吗，谢谢!

  int fd_a, fd_b;

  void fd_init()
  {
    fd_a = socket(AF_INET, SOCK_STREAM, 0);
    fd_b = socket(AF_INET, SOCK_STREAM, 0); 
  }

  void* work_a(void* arg)
  {
    // read something from fd_a then write immediately to fd_b
  }

  void* work_b(void* arg)
  {
    // read something from fd_b then write immediately to fd_a
  }

  int main(int argc, char* argv[])
  {
    // ...
    fd_init();
    pthread_create(pthread_a, 0, work_a, NULL);
    pthread_create(phtread_b, 0, work_b, NULL);
    pthread_join(pthread_a);
    pthread_join(pthread_b);
    // ...
    return EXIT_SUCCESS;
  }

注意:不能在 work_a() 函数或 work_b() 函数中使用缓冲区数组

Answer 1

我不知道有什么有效的方法可以在没有缓冲区的情况下执行此操作，因此这里是一个不使用单独线程且仅使用一个缓冲区的示例。您可以在多个线程中使用下面的 doRW 函数，但每个线程只能使用一个缓冲区。 Recv 和 send 似乎是线程安全的方法，如“Are parallel calls to send/recv on the same socket valid?”中所述。

int fd_a, fd_b;

void fd_init()
{
    fd_a = socket(AF_INET, SOCK_STREAM, 0);
    fd_b = socket(AF_INET, SOCK_STREAM, 0); 
}
bool doRW(int a, int b, char* buf, int size){
    int len = recv(a, buf, size, 0);
    if(len == 0) return false; //disconnected
    else if(len > 0) send(b, buf, len, 0);
    return true;
}
int main(int argc, char* argv[])
{
    // ...
    fd_init();
    fcntl(fd_a, F_SETFL, O_NONBLOCK);
    fcntl(fd_b, F_SETFL, O_NONBLOCK);
    char* buf = malloc(2048);
    while(doRW(fd_a, fd_b, buf, 2048) && doRW(fd_b, fd_a, buf, 2048)) ;
    free(buf);
    return EXIT_SUCCESS;
}