0

我正在创建一个套接字程序来将数据从一台 pc 传输到另一台,但是当我将一些二进制数据发送到另一端进行处理时出现问题。在这种情况下,我需要一个线程在数据套接字发送数据时监听消息套接字。所以我发现问题不在于套接字,如果我尝试将数据写入屏幕(这次没有套接字),就会出现问题。所以我尝试使用 fflush(stdout) 刷新数据但没有运气。代码以这种方式工作。

Initialize the 2 sockets.
Initialize 2 threads.
  One to get the data back through the data socket.
  The other send the data.    
And while sending all the data one pthread_join(m_thread2) in the main function, because the data can take 1 second to be processed or one hour so i keep the program alive this way.

我创建了一个较小的版本,使用两个线程来读取并发送到屏幕,并且在主线程中。

代码:

#include <iostream>
#include <fstream>
#include <string.h>

using namespace std;

const int RCVBUFSIZE=2000;
char echoString[RCVBUFSIZE];
int recvMsgSize;
static void * _sendExec(void *instance);
static void * _recvExec(void *instance);
int main(){
  pthread_t m_thread, m_thread2;
  int merror, merror2;
  merror=pthread_create(&m_thread, NULL, _sendExec, NULL);
  merror2=pthread_create(&m_thread2, NULL, _recvExec, NULL);
  pthread_join(m_thread2, NULL);
}
static void * _sendExec(void *instance){
  int size;
  for(;;){
    while((size=read(fileno(stdin), echoString, RCVBUFSIZE))>0){
       write(fileno(stdout), echoString, size);
    }
    fflush(stdin);
    fflush(stdout);
    pthread_exit(0);
  }
}
static void * _recvExec(void *instance){
  while(1){
     //recvMsgSize=msgTmp->recv(buffer, RCVBUFSIZE)
     write(fileno(stdout), "", 0);
     sleep(1);
  }
}

如果您尝试cat file.tar.gz | ./a.out | tar -zvt,您会看到并非所有数据都显示在屏幕上,如果我放在主屏幕上,删除 pthread_join 就可以了,问题是我需要返回数据,这可能需要一些时间。就像我做一个cat file.tar.gz | ssh root@server "tar -zvt". 问题是,我可以在使用recvExec接收所有数据后杀死sendExec,但它只是在更改代码并删除套接字部分后将stdin刷新给我,只是为了说明问题

感谢人们

4

1 回答 1

1

在您的示例中,tar正在等待更多输入,因为您从不提供文件结束指示。试试这个:

static void * _sendExec(void *instance){
  int size;
  for(;;){
    while((size=read(fileno(stdin), echoString, RCVBUFSIZE))>0){
       write(fileno(stdout), echoString, size);
    }
    fflush(stdin);
    fflush(stdout);
    fclose(stdout); // THIS IS THE LINE THAT FIXES THE SAMPLE PROGRAM
    pthread_exit(0);
  }
}

在添加fclose修复您的示例程序时,我不一定会在您的主程序中推荐它。您的示例仍然有一个无限循环(在 中_recvExec)并且永​​远不会终止。

于 2012-05-04T14:19:57.057 回答