Changeset 234 in flair-src


Ignore:
Timestamp:
Apr 10, 2018, 5:05:27 PM (4 years ago)
Author:
Sanahuja Guillaume
Message:

create file oscket only when necessary

Location:
trunk
Files:
18 edited

Legend:

Unmodified
Added
Removed
  • trunk/demos/CircleFollower/simulator/build_core2_64/bin/simulator_x4.sh

    r227 r234  
    88. $FLAIR_ROOT/flair-dev/scripts/distribution_specific_hack.sh
    99
    10 $EXEC -n x4_0 -t x4 -p 9000 -a 127.0.0.1 -x setup_x4.xml -o 10 -m $FLAIR_ROOT/flair-bin/models -s $FLAIR_ROOT/flair-bin/models/indoor_flight_arena.xml
     10$EXEC -n x4_0 -t x4 -p 9000 -a 127.0.0.1 -x setup_x4.xml -o 100 -m $FLAIR_ROOT/flair-bin/models -s $FLAIR_ROOT/flair-bin/models/indoor_flight_arena.xml
  • trunk/lib/FlairCore/src/FrameworkManager_impl.cpp

    r214 r234  
    193193      Printf("Error udt::close %s", UDT::getlasterror().getErrorMessage());
    194194
    195     status = UDT::close(file_sock);
    196     if (status != 0)
    197       Printf("Error udt::close %s", UDT::getlasterror().getErrorMessage());
    198 
    199195    SleepMS(200); // a revoir, sinon UDT::cleanup bloque en RT
    200196    UDT::cleanup();
     
    210206    return;
    211207  }
     208  this->address=address;
     209  this->port=port;
    212210                                                                                                                                                                                       
    213211        UDT::startup(1024*256,1024*256,1024*256);
     
    216214  // socket file_socket, doit être créé en premier, cf station sol
    217215  Printf("Connecting to %s:%i\n", address.c_str(), port);
    218   file_sock = GetSocket(address, port);
    219   com_sock = GetSocket(address, port);
     216  //file_sock = GetSocket(address, port);
     217  com_sock = GetSocket();
    220218
    221219  // receive buffer allocation
     
    226224
    227225  com = new ui_com(this, com_sock);
    228 
    229   // file managment
    230   bool blocking = true;
    231   UDT::setsockopt(file_sock, 0, UDT_SNDSYN, &blocking, sizeof(bool));
    232   UDT::setsockopt(file_sock, 0, UDT_RCVSYN, &blocking, sizeof(bool));
    233 
    234   int timeout = 100; // ms
    235   UDT::setsockopt(com_sock, 0, UDT_RCVTIMEO, &timeout, sizeof(int));
    236226
    237227  Start();
     
    292282  while (!ToBeStopped()) {
    293283    ssize_t bytesRead;
    294 
    295284    bytesRead = com->Receive(rcv_buf, rcv_buf_size);
    296 
     285    com->CheckConnection();
     286//printf("%i\n",bytesRead);
    297287    if (bytesRead == (ssize_t)rcv_buf_size)
    298288      Err("FrameworkManager max receive size, augmenter le buffer size!\n");
    299289
    300290    if (bytesRead > 0) {
    301       // printf("recu %ld, trame %x\n",bytesRead,(uint8_t)rcv_buf[0]);
    302       // rcv_buf[bytesRead-1]=0;//pour affichage
    303       // printf("%s\n",rcv_buf);
     291       //printf("recu %ld, trame %x %lld\n",bytesRead,(uint8_t)rcv_buf[0],GetTime()/(1000000));
     292       //rcv_buf[bytesRead-1]=0;//pour affichage
     293       //printf("%s\n",rcv_buf);
    304294
    305295      switch ((uint8_t)rcv_buf[0]) {
     
    376366}
    377367
    378 void FrameworkManager_impl::SendFile(string path, string name) {
     368void FrameworkManager_impl::SendFile(UDTSOCKET socket,string path, string name) {
    379369  char *buf, *more_buf;
    380370  int size;
     
    411401  Printf("sending %s, size: %i\n", filename.c_str(), size);
    412402  // send file information
    413   UDT::sendmsg(file_sock, buf, sizeof(uint8_t) + sizeof(int) + name.size(), -1,true);
     403  UDT::sendmsg(socket, buf, sizeof(uint8_t) + sizeof(int) + name.size(), -1,true);
    414404
    415405  more_buf = (char *)realloc((void *)buf, size);
     
    424414  pbuf->sgetn(buf, size);
    425415  // send the file
    426   nb_write = UDT::sendmsg(file_sock, buf, size, -1, true);
     416  nb_write = UDT::sendmsg(socket, buf, size, -1, true);
    427417
    428418  if (nb_write < 0) {
     
    436426}
    437427
    438 void FrameworkManager_impl::FinishSending() {
     428void FrameworkManager_impl::FinishSending(UDTSOCKET socket) {
    439429  char rm_cmd[256];
    440430
    441431  // send orignal xml
    442   SendFile(log_path, "setup.xml");
     432  SendFile(socket,log_path, "setup.xml");
    443433  sprintf(rm_cmd, "rm %s/setup.xml", log_path.c_str());
    444434  system(rm_cmd);
     
    447437  for (size_t i = 0; i < xml_changes.size(); i++) {
    448438    // Printf("%s\n",xml_changes.at(i).c_str());
    449     SendFile(log_path, xml_changes.at(i).c_str());
     439    SendFile(socket,log_path, xml_changes.at(i).c_str());
    450440    sprintf(rm_cmd, "rm %s/%s", log_path.c_str(), xml_changes.at(i).c_str());
    451441    system(rm_cmd);
     
    454444
    455445  // end notify
    456   char buf = END;
    457   int nb_write = UDT::sendmsg(file_sock, &buf, 1, -1, true);
     446  char buf = END_SENDING_FILES;
     447  int nb_write = UDT::sendmsg(socket, &buf, 1, -1, true);
    458448
    459449  if (nb_write < 0) {
     
    464454
    465455  //wait end ACK
    466   int nb_read = UDT::recvmsg(file_sock,&buf,1);
     456  int nb_read = UDT::recvmsg(socket,&buf,1);
    467457  if(nb_read<0) {
    468458    Err("UDT::recvmsg error (%s)\n",UDT::getlasterror().getErrorMessage());
     
    472462}
    473463
    474 UDTSOCKET FrameworkManager_impl::GetSocket(string address, uint16_t port) {
     464UDTSOCKET FrameworkManager_impl::GetSocket(void) {
    475465  while (1) {
    476466    UDTSOCKET new_fd;
     
    854844              caller->gcs_watchdog->Join();
    855845          }
     846          //create a socket for files
     847          UDTSOCKET file_sock = caller->GetSocket();
     848          char data=START_SENDING_FILES;
     849          ssize_t nb_write = UDT::sendmsg(file_sock, &data, 1, -1, true);
     850          if (nb_write < 0) {
     851            caller->Err("UDT::sendmsg error (%s)\n", UDT::getlasterror().getErrorMessage());
     852            if (UDT::getlasterror().getErrorCode() == CUDTException::ECONNLOST ||
     853                UDT::getlasterror().getErrorCode() == CUDTException::EINVSOCK) {
     854            }
     855          } else if (nb_write != 1) {
     856            caller->Err("%s, code %i (%ld/%ld)\n", UDT::getlasterror().getErrorMessage(),
     857                UDT::getlasterror().getErrorCode(), nb_write, 1);
     858          }
    856859          for (size_t i = 0; i < logs.size(); i++) {
    857860            if (logs.at(i).dbtFile != NULL) {
     
    859862
    860863              string filename = caller->FileName(logs.at(i).device) + ".dbt";
    861               caller->SendFile(caller->log_path, filename);
     864              caller->SendFile(file_sock,caller->log_path, filename);
    862865
    863866              fstream txt_file;
     
    870873              txt_file.close();
    871874
    872               caller->SendFile(caller->log_path, filename);
     875              caller->SendFile(file_sock,caller->log_path, filename);
    873876            }
    874877          }
    875878          // a revoir celui ci est le xml enregistré et pas forcement l'actuel
    876879          // if(caller->xml_file!="") caller->SendFile(caller->xml_file);
    877           caller->FinishSending();
     880          caller->FinishSending(file_sock);
     881         
     882          int status = UDT::close(file_sock);
     883          if (status != 0)
     884            caller->Err("Error udt::close %s", UDT::getlasterror().getErrorMessage());
    878885
    879886          logs.clear();
  • trunk/lib/FlairCore/src/ui_com.cpp

    r213 r234  
    223223}
    224224
     225void ui_com::CheckConnection(void) {
     226  int32_t val,len;
     227 
     228  int result=UDT::getsockopt(socket_fd,0,UDT_STATE,&val,&len);
     229   if (result < 0) {
     230    Printf("UDT::getsockopt error (%s)\n", UDT::getlasterror().getErrorMessage());
     231   }
     232 // printf("opt: %i %i %i\n",result,val,len);
     233}
     234
    225235void ui_com::Run(void) {
    226236  // check endianness
  • trunk/lib/FlairCore/src/unexported/FrameworkManager_impl.h

    r213 r234  
    8282private:
    8383  flair::core::FrameworkManager *self;
    84   UDTSOCKET file_sock, com_sock;
    85   UDTSOCKET GetSocket(std::string address, uint16_t port);
     84  UDTSOCKET com_sock;
     85  UDTSOCKET GetSocket(void);
    8686  void Run();
    87   void SendFile(std::string path, std::string name);
    88   void FinishSending(void);
     87  void SendFile(UDTSOCKET socket,std::string path, std::string name);
     88  void FinishSending(UDTSOCKET socket);
    8989  std::string FileName(flair::core::IODevice *device);
    9090  void SaveXmlChange(char *buf);
     
    9292  size_t rcv_buf_size;
    9393  char *rcv_buf;
     94  std::string address;
     95  uint16_t port;
    9496#ifdef __XENO__
    9597  int CreatePipe(RT_PIPE *fd, std::string name);
  • trunk/lib/FlairCore/src/unexported/communication.h

    r15 r234  
    66#define COMMUNICATION_H
    77
    8 // messages file socket
    9 #define FILE_INFO_LITTLE_ENDIAN 0x01
    10 #define FILE_INFO_BIG_ENDIAN 0x02
    11 #define END 0x03
    12 
     8// messages data socket
    139#define XML_HEADER 0x3c
    1410#define ZLIB_HEADER 0x78
    1511#define WATCHDOG_HEADER 0x01
     12#define DATAS_LITTLE_ENDIAN 0x02
     13#define DATAS_BIG_ENDIAN 0x03
    1614
    17 #define DATAS_LITTLE_ENDIAN 0xfd
    18 #define DATAS_BIG_ENDIAN 0xfe
     15#define START_SENDING_FILES 0x10
     16#define FILE_INFO_LITTLE_ENDIAN 0x11
     17#define FILE_INFO_BIG_ENDIAN 0x12
     18#define END_SENDING_FILES 0x13
    1919
    2020#endif // COMMUNICATION_H
  • trunk/lib/FlairCore/src/unexported/ui_com.h

    r15 r234  
    4848  void UnBlock(void);
    4949  bool ConnectionLost(void);
    50 
     50  void CheckConnection(void);
    5151private:
    5252  ssize_t send_size;
  • trunk/tools/Dbt2csv/CMakeLists.txt

    r107 r234  
    44
    55if("${CMAKE_SYSTEM_PROCESSOR}" MATCHES "i686" OR "${CMAKE_SYSTEM_PROCESSOR}" MATCHES "i586" OR "${CMAKE_SYSTEM_PROCESSOR}" MATCHES "x86_64")
     6warn("${PROJECT_NAME} is broken since changes in FlairGCS/src/file_ui.cpp")
     7return()
    68
    79SET(PROJECT_SOURCE_FILES
  • trunk/tools/FlairGCS/src/ConnectionLayout.cpp

    r222 r234  
    88#include <qendian.h>
    99#include "communication.h"
    10 #include <zlib.h>
    11 #include <assert.h>
    12 
    13 #define COMPRESS_CHUNK 1024
    14 #define RX_TIME 5000
    1510
    1611ConnectionLayout::ConnectionLayout(UdtSocket *socket, QString name)
    1712    : Layout(NULL, name, "root") {
    18   isRemoteNameDefined = false;
    1913  this->socket = socket;
    20 
    21   total_received=0;
    22   receive_timer = new QTimer(this);
    23   connect(receive_timer, SIGNAL(timeout()), this, SLOT(computeRxRate()));
    24   receive_timer->start(RX_TIME);
     14  this->name = name;
    2515}
    2616
    2717ConnectionLayout::~ConnectionLayout() {
    28   receive_timer->stop();
    2918}
    3019
    31 void ConnectionLayout::computeRxRate(void) {
    32   float receive_rate=((float)total_received/(RX_TIME/1000))/1000;//in Ko/s
    33   total_received=0;
    34   computedRxRate(receive_rate);
     20QString ConnectionLayout::getUDTStats() {
     21  return socket->getUDTStats();
    3522}
    3623
    3724void ConnectionLayout::receive(char *buf, int size) {
    38   total_received+=size-1;
    3925  // printf("trame %x\n",buf[0]);
    4026  // for(int i=0; i<size;i++) printf("%x ",buf[i]);
    4127  // printf("\n");
    42   switch (buf[0]) {
    43   case ZLIB_HEADER: {
    44     ssize_t out_size;
    45     char *uncompressbuf;
    46     uncompressBuffer(buf, size, &uncompressbuf, &out_size);
    47     handleUncompressedFrame(uncompressbuf, out_size);
    48     free(uncompressbuf);
    49     break;
    50   }
    51   default:
    52     handleUncompressedFrame(buf, size);
    53   }
    54   free(buf);
    55 }
    56 
    57 void ConnectionLayout::XmlToSend(QDomDocument doc) {
    58   // printf("xml to send\n%s\n",doc.toString().toLocal8Bit().constData());
    59   // xml to send a mettre dans le manager
    60   QMetaObject::invokeMethod(
    61       socket, "write", Qt::BlockingQueuedConnection,
    62       Q_ARG(const char *, doc.toString().toLocal8Bit().constData()),
    63       Q_ARG(qint64, doc.toString().toLocal8Bit().length()));
    64 }
    65 
    66 void ConnectionLayout::LoadXml(QDomDocument to_parse) {
    67   if (!isRemoteNameDefined) {
    68     printf("load xml: name not defined!\n");
    69     return;
    70   }
    71 
    72   QDomElement tmp = to_parse.firstChildElement("root");
    73   while (tmp.attribute("name") != remoteName && !tmp.isNull())
    74     tmp = to_parse.nextSiblingElement("root");
    75 
    76   if (!tmp.isNull()) {
    77     XmlWidget::LoadXml(tmp);
    78   } else {
    79     printf("%s not found in xml file \n", remoteName.toLocal8Bit().constData());
    80   }
    81 }
    82 
    83 void ConnectionLayout::handleUncompressedFrame(char *buf, ssize_t size) {
    8428  switch ((unsigned char)buf[0]) {
    8529  case XML_HEADER: {
     
    9236    if (!doc.setContent(xml)) {
    9337      printf("prob setContent fichier\n");
    94     }
    95 
    96     if (!isRemoteNameDefined) {
    97       isRemoteNameDefined = true;
    98       remoteName = doc.firstChildElement("root").attribute("name");
    99       setRemoteName(remoteName);
    100       SetAttribute("name", remoteName);
    10138    }
    10239
     
    12764}
    12865
     66void ConnectionLayout::XmlToSend(QDomDocument doc) {
     67  // printf("xml to send\n%s\n",doc.toString().toLocal8Bit().constData());
     68 
     69  // xml to send a mettre dans le manager
     70  socket->write(doc.toString().toLocal8Bit().constData(),doc.toString().toLocal8Bit().length());/*
     71  QMetaObject::invokeMethod(
     72      socket, "write", Qt::BlockingQueuedConnection,
     73      Q_ARG(const char *, doc.toString().toLocal8Bit().constData()),
     74      Q_ARG(qint64, doc.toString().toLocal8Bit().length()));*/
     75}
     76
     77void ConnectionLayout::LoadXml(QDomDocument to_parse) {
     78  QDomElement tmp = to_parse.firstChildElement("root");
     79  while (tmp.attribute("name") != name && !tmp.isNull())
     80    tmp = to_parse.nextSiblingElement("root");
     81
     82  if (!tmp.isNull()) {
     83    XmlWidget::LoadXml(tmp);
     84  } else {
     85    printf("%s not found in xml file \n", name.toLocal8Bit().constData());
     86  }
     87}
     88
    12989void ConnectionLayout::removeDataRemote(DataRemote *data) {
    13090  dataremotes.removeOne(data);
     
    13595}
    13696
    137 QString ConnectionLayout::getRemoteName() { return remoteName; }
     97QString ConnectionLayout::getName() { return name; }
    13898
    13999void ConnectionLayout::drawDatas(char *buf, int buf_size, uint16_t period,
     
    144104}
    145105
    146 int ConnectionLayout::uncompressBuffer(char *in, ssize_t in_size, char **out,
    147                                        ssize_t *out_size) {
    148   int ret;
    149   unsigned have;
    150   z_stream strm;
     106QString ConnectionLayout::getDocRootName(char* buf, int size) {
     107  QString xml;
     108  QDomDocument doc;
     109  xml = QString((char *)buf);
     110  xml.resize(size);
    151111
    152   // allocate inflate state
    153   strm.zalloc = Z_NULL;
    154   strm.zfree = Z_NULL;
    155   strm.opaque = Z_NULL;
    156   strm.avail_in = 0;
    157   strm.next_in = Z_NULL;
    158   ret = inflateInit(&strm);
    159   if (ret != Z_OK)
    160     return ret;
    161 
    162   *out = (char *)malloc(COMPRESS_CHUNK);
    163   if (!(*out))
    164     return Z_BUF_ERROR;
    165 
    166   strm.avail_in = in_size;
    167   strm.next_in = (unsigned char *)in;
    168   strm.avail_out = COMPRESS_CHUNK;
    169   strm.next_out = (unsigned char *)*out;
    170 
    171   ret = inflate(&strm, Z_NO_FLUSH);
    172   assert(ret != Z_STREAM_ERROR); // state not clobbered
    173   switch (ret) {
    174   case Z_NEED_DICT:
    175     ret = Z_DATA_ERROR; // and fall through
    176   case Z_DATA_ERROR:
    177   case Z_MEM_ERROR:
    178     (void)inflateEnd(&strm);
    179     return ret;
     112  if (!doc.setContent(xml)) {
     113    printf("prob setContent fichier\n");
    180114  }
    181   have = COMPRESS_CHUNK - strm.avail_out;
    182   *out_size = have;
    183 
    184   // printf("%i -> %i\n",in_size,have);
    185   // printf("%s\n",*out);
    186   // clean up and return
    187   (void)inflateEnd(&strm);
    188   return ret == Z_STREAM_END ? Z_OK : Z_DATA_ERROR;
     115 
     116  return doc.firstChildElement("root").attribute("name");
    189117}
  • trunk/tools/FlairGCS/src/ConnectionLayout.h

    r30 r234  
    66#define CONNECTIONLAYOUT_H
    77
    8 #include <QTimer>
    98#include "Layout.h"
    109
     
    2221  void removeDataRemote(DataRemote *data);
    2322  void LoadXml(QDomDocument to_parse);
    24   QString getRemoteName();
     23  QString getName();
     24  QString getUDTStats();
     25  static QString getDocRootName(char* buf, int size);
    2526
    2627private:
    27   static int uncompressBuffer(char *in, ssize_t in_size, char **out,
    28                               ssize_t *out_size);
    29   void handleUncompressedFrame(char *buf, ssize_t size);
    3028  void drawDatas(char *buf, int buf_size, uint16_t period,
    3129                 bool big_endian = false);
    32   bool isRemoteNameDefined;
    33   QString remoteName;
     30  QString name;
    3431  UdtSocket *socket;
    3532  QList<DataRemote *> dataremotes;
    36   unsigned int total_received;
    37   QTimer *receive_timer;
    38 
     33 
    3934private slots:
    4035  void receive(char *buf, int size);
    41   void computeRxRate(void);
    4236
    4337signals:
    44   void setRemoteName(QString name);
    45   void computedRxRate(float rxRate);
     38  void UDTStats(QString stats); 
    4639};
    4740
  • trunk/tools/FlairGCS/src/DataRemote.cpp

    r222 r234  
    5757  if (action == setRefreshRate) {
    5858    bool ok;
    59     // uint16_t time = QInputDialog::getInt(this, QString("Set refresh rate
    60     // (%1)").arg(title().text()),tr("Value (ms):"),
    61     // (uint16_t)(qRound(refresh_rate*1000)), 1, 65535, 10, &ok);
     59 
    6260    uint16_t time = QInputDialog::getInt(
    6361        parent, "Set refresh rate ", "Value (ms):",
     
    8381  SetAttribute("period", period);
    8482  SetAttribute("enabled", auto_refresh);
    85 
    8683  connectionLayout()->XmlToSend(XmlDoc());
    87 
    8884  RemoveAttribute("period");
    8985  RemoveAttribute("enabled");
  • trunk/tools/FlairGCS/src/Manager.cpp

    r30 r234  
    77#include "ConnectionLayout.h"
    88#include "communication.h"
     9#include "file_ui.h"
    910#include <QDate>
    1011#include <QFileDialog>
     
    1718#include <QVBoxLayout>
    1819#include <QModelIndex>
    19 //#include <qmetatype.h>
     20#include <QStatusBar>
    2021#include <qendian.h>
    2122#include <iostream>
     
    3637  qRegisterMetaType<QModelIndex>("QModelIndex"); // pour le file ui??
    3738  this->name = name;
    38 
     39//fprintf(stderr,"Manager %x\n",thread());
    3940  setWindowTitle(name);
    40 //statusBar()->showMessage(tr("Ready to serve"));
     41
    4142  // manager layout
    4243  managerLayout = new QVBoxLayout;
     
    7980  bool blocking = false;
    8081  UDT::setsockopt(serv, 0, UDT_RCVSYN, &blocking, sizeof(bool));
    81 
     82//UDT::setsockopt(serv, 0, UDT_CC, new CCCFactory<CUDPBlast>, sizeof(CCCFactory<CUDPBlast>));
    8283  sockaddr_in my_addr;
    8384  my_addr.sin_family = AF_INET;
     
    9697  QTimer *timer = new QTimer(this);
    9798  connect(timer, SIGNAL(timeout()), this, SLOT(acceptConnections()));
    98   timer->start(20);
     99  timer->start(500);
    99100}
    100101
    101102Manager::~Manager() {
    102   emit killUdtSockets();
     103  printf("%i\n",udtSockets.count());
     104  for (int i = 0; i < udtSockets.count(); i++) {
     105    udtSockets.at(i)->kill();
     106    udtSockets.at(i)->thread()->wait();
     107  }
    103108
    104109  // delete main_layout;
    105110  UDT::cleanup();
     111  printf("ok\n");
    106112}
    107113
    108114void Manager::acceptConnections(void) {
    109   static UDTSOCKET first_socket = 0;
    110115  UDTSOCKET socket;
    111116
     
    113118  int namelen = sizeof(their_addr);
    114119
    115   if (UDT::INVALID_SOCK ==
    116       (socket = UDT::accept(serv, (sockaddr *)&their_addr, &namelen))) {
     120  if (UDT::INVALID_SOCK ==(socket = UDT::accept(serv, (sockaddr *)&their_addr, &namelen))) {
    117121    if (UDT::getlasterror().getErrorCode() != 6002)
    118       printf("accept: %s, code %i\n", UDT::getlasterror().getErrorMessage(),
    119              UDT::getlasterror().getErrorCode());
     122      printf("accept error: %s, code %i\n", UDT::getlasterror().getErrorMessage(),UDT::getlasterror().getErrorCode());
    120123    return;
    121124  } else {
    122     printf("connected to %s:%i\n", inet_ntoa(their_addr.sin_addr),
    123            their_addr.sin_port);
    124 
    125     if (!first_socket) {
    126       first_socket = socket;
    127       return;
    128     } else {
    129       QThread *thread = new QThread(this);
    130       UdtSocket *new_udt = new UdtSocket(first_socket, socket, name);
    131       new_udt->moveToThread(thread);
    132 
    133       newConnection(new_udt);
    134 
    135       connect(this, SIGNAL(killUdtSockets()), thread, SLOT(quit()));
    136       connect(this, SIGNAL(killUdtSockets()), new_udt, SLOT(kill()),
    137               Qt::BlockingQueuedConnection);
    138 
    139       connect(thread, SIGNAL(started()), new_udt, SLOT(handleConnections()));
    140 
    141       thread->start();
    142       first_socket = 0;
    143     }
    144   }
    145 }
    146 
    147 void Manager::newConnection(UdtSocket *socket) {
    148 
    149   // no tabs to 2 tabs
    150   if (connectionsLayout.count() == 1) {
    151     tabBar->addTab(hiddenTabName);
    152     currentTab = 0;
    153     connectionsWidget.at(0)->show();
    154   }
    155 
    156   // layout utilisateur
    157   ConnectionLayout *newLayout = new ConnectionLayout(socket, "interface");
     125    QString name=QString("%1:%2").arg(inet_ntoa(their_addr.sin_addr)).arg(their_addr.sin_port);
     126    printf("connected to %s\n",name.toLocal8Bit().constData());
     127   
     128    QThread *thread = new QThread(this);
     129    UdtSocket *udtSocket = new UdtSocket(socket,name);
     130    udtSocket->moveToThread(thread);
     131    udtSockets.append(udtSocket);
     132
     133    connect(udtSocket, SIGNAL(newFileUI(UDTSOCKET)), this, SLOT(newFileUI(UDTSOCKET)));
     134    connect(udtSocket, SIGNAL(newConnectionLayout(QString)), this, SLOT(newConnectionLayout(QString)),Qt::BlockingQueuedConnection);
     135    connect(thread, SIGNAL(started()), udtSocket, SLOT(receiveData()));
     136
     137    thread->start(); 
     138  }
     139}
     140
     141void Manager::newConnectionLayout(QString name) {
     142  UdtSocket* udtSocket=(UdtSocket *)sender();
     143 
     144  ConnectionLayout *newLayout = new ConnectionLayout(udtSocket, name);
    158145  connectionsLayout.append(newLayout);
    159   connect(newLayout, SIGNAL(setRemoteName(QString)), this,
    160           SLOT(tabName(QString)));
    161   connect(newLayout, SIGNAL(computedRxRate(float)), this,
    162           SLOT(printRxRates(float)));
    163   connect(socket, SIGNAL(dataReady(char *, int)), newLayout,
    164           SLOT(receive(char *, int)));
    165   connect(newLayout, SIGNAL(destroyed(QObject *)), this,
    166           SLOT(layoutDestroyed(QObject *)));
    167   connect(socket, SIGNAL(destroyed()), newLayout, SLOT(deleteLater()));
    168 
    169   // widget
     146  connect(udtSocket, SIGNAL(UDTStats(QString)), newLayout, SIGNAL(UDTStats(QString)));//connection in 2 steps to get udtsocket as sender
     147  connect(newLayout, SIGNAL(UDTStats(QString)), this, SLOT(printUDTStats(QString)));
     148  connect(udtSocket, SIGNAL(dataReady(char *, int)), newLayout,SLOT(receive(char *, int)),Qt::BlockingQueuedConnection);
     149  connect(newLayout, SIGNAL(destroyed(QObject *)), this, SLOT(layoutDestroyed(QObject *)));
     150  connect(udtSocket, SIGNAL(destroyed()), newLayout, SLOT(deleteLater()));
     151 
     152   // widget
    170153  QWidget *newWidget = new QWidget();
    171154  connectionsWidget.append(newWidget);
     
    173156  managerLayout->insertWidget(1, newWidget);
    174157  newWidget->hide();
    175 
     158 
     159  //tab: avoid having only 1 tab (0, 2 or more)
    176160  if (connectionsLayout.count() == 1) { // first connection
    177     newWidget->show();
    178   } else { // add a tab for the new connection
    179     tabBar->addTab("unknown");
    180   }
    181 }
    182 
    183 void Manager::printRxRates(float rxRate) {
     161    connectionsWidget.at(0)->show();
     162    hiddenTabName = name;
     163  }
     164  if (connectionsLayout.count() == 2) {
     165    tabBar->addTab(hiddenTabName);
     166    currentTab = 0;
     167  }
     168  if (connectionsLayout.count() > 1) {
     169    tabBar->addTab(name);
     170  }
     171}
     172
     173void Manager::layoutDestroyed(QObject *obj) {
     174  int index = connectionsLayout.indexOf((ConnectionLayout *)obj);
     175
     176  //tab: avoid having only 1 tab (0, 2 or more)
     177  if (tabBar->count() > 1) {
     178    tabBar->removeTab(index);
     179  }
     180
     181  delete connectionsWidget.at(index);
     182  connectionsWidget.removeAt(index);
     183  connectionsLayout.removeOne((ConnectionLayout *)obj);
     184
     185  if (connectionsLayout.count() == 1) {
     186    hiddenTabName = tabBar->tabText(0);
     187    tabBar->removeTab(0);
     188  }
     189
     190  if (connectionsLayout.count() == 0) {
     191    status->showMessage("");
     192  }
     193}
     194
     195void Manager::newFileUI(UDTSOCKET socket) {
     196  //remove udtsocket as it will be automatically destroyed
     197  udtSockets.removeOne((UdtSocket *)sender());
     198 
     199  QThread *thread = new QThread(this);
     200  file_ui* fileUi = new file_ui(socket,name);
     201  fileUi->moveToThread(thread);
     202  connect(thread, SIGNAL(started()), fileUi, SLOT(receive()));
     203  connect(fileUi, SIGNAL(finished()), this, SLOT(deleteFileUI()));
     204  thread->start();
     205}
     206
     207void Manager::deleteFileUI(void) {
     208  sender()->thread()->quit();
     209  delete sender();
     210}
     211
     212void Manager::printUDTStats(QString stats) {
    184213  int index = connectionsLayout.indexOf((ConnectionLayout *)sender());
    185 
     214  if(index==-1) return;
     215 
    186216  if (tabBar->count() == 0) {
    187     status->showMessage(tr("rx rate %1 kB/s").arg(rxRate,0,'f',3));
     217    status->showMessage(stats);
    188218  } else if (index==tabBar->currentIndex()) {
    189     status->showMessage(tr("%1 rx rate %2 kB/s").arg(tabBar->tabText(index)).arg(rxRate,0,'f',3));
     219    status->showMessage(QString("%1: %2").arg(tabBar->tabText(index)).arg(stats));
    190220  }
    191221}
     
    200230    connectionsWidget.at(0)->show();
    201231  }
    202 }
    203 
    204 void Manager::tabName(QString name) {
    205   int index = connectionsLayout.indexOf((ConnectionLayout *)sender());
     232 
    206233  if (tabBar->count() == 0) {
    207     hiddenTabName = name;
     234    status->showMessage(connectionsLayout.at(0)->getUDTStats());
    208235  } else {
    209     tabBar->setTabText(index, name);
    210   }
    211 }
    212 
    213 void Manager::layoutDestroyed(QObject *obj) {
    214   int index = connectionsLayout.indexOf((ConnectionLayout *)obj);
    215 
    216   if (tabBar->count() > 1) {
    217     tabBar->removeTab(index);
    218   }
    219 
    220   delete connectionsWidget.at(index);
    221   connectionsWidget.removeAt(index);
    222   connectionsLayout.removeOne((ConnectionLayout *)obj);
    223 
    224   if (connectionsLayout.count() == 1) {
    225     hiddenTabName = tabBar->tabText(0);
    226     tabBar->removeTab(0);
    227   }
    228 
    229   if (connectionsLayout.count() == 0) {
    230     status->showMessage("");
     236    status->showMessage(QString("%1: %2").arg(tabBar->tabText(index)).arg(connectionsLayout.at(index)->getUDTStats()));
    231237  }
    232238}
     
    240246      QFile *file;
    241247      file = new QFile(dir_name + "/" +
    242                        connectionsLayout.at(i)->getRemoteName() + ".xml");
     248                       connectionsLayout.at(i)->getName() + ".xml");
    243249      if (!file->open(QIODevice::ReadOnly | QIODevice::Text)) {
    244250        QMessageBox::warning(this, "Warning",
    245251                             "Enable to load " +
    246                                  connectionsLayout.at(i)->getRemoteName() +
     252                                 connectionsLayout.at(i)->getName() +
    247253                                 ".xml");
    248254        continue;
     
    256262        QMessageBox::critical(
    257263            this, "Error",
    258             "unable to read " + connectionsLayout.at(i)->getRemoteName() +
     264            "unable to read " + connectionsLayout.at(i)->getName() +
    259265                ".xml" + " (" + errorMsg + " at " + QString::number(errorLine) +
    260266                "," + QString::number(errorColumn) + ")");
     
    312318    connectionsLayout.at(i)->GetFullXml((QDomElement *)xml);
    313319
    314     QFile fichier(dir_name + "/" + connectionsLayout.at(i)->getRemoteName() +
     320    QFile fichier(dir_name + "/" + connectionsLayout.at(i)->getName() +
    315321                  ".xml");
    316322    QString write_doc = (xml->ownerDocument()).toString();
     
    322328    }
    323329    QTextStream stream(&fichier);
    324     stream << write_doc; // On utilise l'opérateur << pour écrire write_doc dans
    325                          // le document XML.
     330    stream << write_doc;
    326331    fichier.close();
    327332
  • trunk/tools/FlairGCS/src/Manager.h

    r30 r234  
    99#include <udt.h>
    1010#include <qdom.h>
    11 #include <QStatusBar>
    1211
    1312class UdtSocket;
     
    1716class QPushButton;
    1817class QGridLayout;
     18class QStatusBar;
    1919
    2020class Manager : public QWidget {
     
    3030  QList<ConnectionLayout *> connectionsLayout;
    3131  QList<QWidget *> connectionsWidget;
     32  QList<UdtSocket *> udtSockets;
    3233  QTabBar *tabBar;
    3334  QString name, hiddenTabName;
     
    4142  QStatusBar *status;
    4243
    43   void newConnection(UdtSocket *socket);
    44 
    4544private slots:
    4645  void acceptConnections(void);
     
    4948  void save(void);
    5049  void reset(void);
     50  void tabBarCurrentChanged(int index);
     51  void printUDTStats(QString stats);
     52  void newFileUI(UDTSOCKET socket);
     53  void deleteFileUI(void);
     54  void newConnectionLayout(QString name);
    5155  void layoutDestroyed(QObject *obj);
    52   void tabBarCurrentChanged(int index);
    53   void tabName(QString name);
    54   void printRxRates(float rxRate);
    55 
    56 signals:
    57   void killUdtSockets(void);
    5856
    5957protected:
  • trunk/tools/FlairGCS/src/UdtSocket.cpp

    r30 r234  
    44// %flair:license}
    55#include "UdtSocket.h"
     6#include "ConnectionLayout.h"
    67#include <stdio.h>
    78#include <stdlib.h>
     
    1011#include <QDir>
    1112#include <QDate>
    12 #include "file_ui.h"
    1313#include "communication.h"
     14#include <zlib.h>
     15#include <assert.h>
     16#include <QThread>
     17#define COMPRESS_CHUNK 1024*100
    1418
    1519#ifndef WIN32
     
    2024#endif
    2125
     26#define COMPUTE_UDT_STATS_TIMER 2000
     27#define HEARTBEAT_TIMER 200
     28
    2229using namespace std;
    2330
    24 UdtSocket::UdtSocket(UDTSOCKET file_socket, UDTSOCKET com_socket, QString name)
    25     : QObject() {
    26   this->file_socket = file_socket;
    27   this->com_socket = com_socket;
    28   this->name = name;
     31UdtSocket::UdtSocket(UDTSOCKET socket,QString name): QObject() {
     32  this->socket = socket;
     33  this->name=name;
    2934  stop = false;
    30   file_dialog = new file_ui();
    31 
    32   bool blocking = true;
    33   UDT::setsockopt(file_socket, 0, UDT_RCVSYN, &blocking, sizeof(bool));
    34 
     35  destroySocket=true;
     36  socketType=unknown;
     37  total_received=0;
     38 
     39  bool blocking = false;
     40  if (UDT::setsockopt(socket, 0, UDT_RCVSYN, &blocking, sizeof(bool)) != 0)
     41    printf("UDT::setsockopt error (UDT_RCVSYN)\n");
     42   
    3543  heartbeat_timer = new QTimer(this);
    3644  connect(heartbeat_timer, SIGNAL(timeout()), this, SLOT(heartbeat()));
    37   heartbeat_timer->start(200);
     45  heartbeat_timer->start(HEARTBEAT_TIMER);
     46 
     47  udtstats_timer = new QTimer(this);
     48  connect(udtstats_timer, SIGNAL(timeout()), this, SLOT(getUTDStats()));
     49  udtstats_timer->start(COMPUTE_UDT_STATS_TIMER);
    3850}
    3951
    4052UdtSocket::~UdtSocket() {
    4153  heartbeat_timer->stop();
    42 
    43   UDT::close(file_socket);
    44   UDT::close(com_socket);
    45 
    46   file_dialog->deleteLater();
     54  udtstats_timer->stop();
     55  if(destroySocket) UDT::close(socket);         
     56}
     57
     58void UdtSocket::setName(QString name) {
     59  printf("  %s is %s\n",this->name.toLocal8Bit().constData(),name.toLocal8Bit().constData());
     60  this->name=name;
    4761}
    4862
     
    5266void UdtSocket::heartbeat(void) {
    5367  char data = WATCHDOG_HEADER;
    54   write(&data, 1);
     68  quint64 sent=write(&data, 1,HEARTBEAT_TIMER,false);
     69  if (sent != 1 && UDT::getlasterror().getErrorCode() == 2001) {
     70    heartbeat_timer->stop();
     71  }
     72 
     73}
     74
     75void UdtSocket::getUTDStats(void) {
     76  float rxRate=((float)total_received/(COMPUTE_UDT_STATS_TIMER/1000))/1000;//in Ko/s
     77  total_received=0;
     78 
     79  UDT::TRACEINFO perf;
     80  if (UDT::ERROR == UDT::perfmon(socket, &perf))   {
     81    printf("perfmon: %s\n",UDT::getlasterror().getErrorMessage());
     82  }/* else {
     83    printf("%s socket stats:\n",name.toLocal8Bit().constData());
     84    printf("total number of sent packets, including retransmissions: %i\n",perf.pktSentTotal);
     85    printf("total number of received packets: %i\n",perf.pktRecvTotal);
     86    printf("total number of lost packets, measured in the sending side: %i\n",perf.pktSndLossTotal);
     87    printf("total number of lost packets, measured in the receiving side: %i\n",perf.pktRcvLossTotal);
     88    printf("total number of retransmitted packets, measured in the sending side: %i\n",perf.pktRetransTotal);
     89    printf("total number of sent ACK packets: %i\n",perf.pktSentACKTotal);
     90    printf("total number of received ACK packets: %i\n",perf.pktRecvACKTotal);
     91    printf("total number of sent NAK packets: %i\n",perf.pktSentNAKTotal);
     92    printf("total number of received NAK packets: %i\n",perf.pktRecvNAKTotal);
     93    printf("round trip time: %fms\n",perf.msRTT);
     94     
     95  }*/
     96  stats=QString("rx rate %1kB/s, round trip %2ms, lost packets %3").arg(rxRate,0,'f',3).arg(perf.msRTT,0,'f',3).arg(perf.pktSndLossTotal);
     97  UDTStats(stats);
     98 
     99}
     100
     101QString UdtSocket::getUDTStats() {
     102  return stats;
    55103}
    56104
    57105void UdtSocket::kill(void) {
    58   printf("disconnected\n");
    59106  stop = true;
    60107  deleteLater();
    61108}
    62109
    63 void UdtSocket::handleConnections(void) {
     110void UdtSocket::receiveData(void) {
     111  int buf_size;
     112  int opt_size;
     113  UDT::getsockopt(socket, 0, UDT_RCVBUF, &buf_size, &opt_size);
     114  char *buf = (char *)malloc(buf_size);
     115  if (!buf) {
     116    printf("error malloc UdtSocket::receiveData buffer\n");
     117    return;
     118  }
     119  char *uncompressbuf=(char *)malloc(COMPRESS_CHUNK);
     120  if (!uncompressbuf) {
     121    printf("error malloc UdtSocket::receiveData uncompress buffer\n");
     122    free(buf);
     123    return;
     124  }
     125  //fprintf(stderr,"receiveData %x\n",thread());
     126 
    64127  while (!stop) {
     128    QCoreApplication::processEvents();
     129   
     130    //we need to use epoll?
     131    //tests are showing that UDT::recvmsg is waiting for the entire timeout before returning
     132    //note that in flair::core the behaviour of UDT::recvmsg timeout is as expected
     133    //is it a difference between client and server??
     134    //tests also show that we need to recreate the eid every time, otherwise wait returns immediately
    65135    int eid = UDT::epoll_create();
    66136    if (eid < 0) {
    67       printf("epoll_create error: %s\n", UDT::getlasterror().getErrorMessage());
    68     }
    69 
    70     if (UDT::epoll_add_usock(eid, file_socket) < 0) {
     137      printf("%s: epoll_create error (%s)\n",name.toLocal8Bit().constData(),UDT::getlasterror().getErrorMessage());
     138    }
     139
     140    if (UDT::epoll_add_usock(eid, socket) < 0) {
    71141      if (UDT::getlasterror().getErrorCode() == 5004) {
    72         printf("epoll_add_usock error\n");
    73         break;
     142        printf("disconnected from %s\n",name.toLocal8Bit().constData());
     143        heartbeat_timer->stop();
     144        deleteLater();
     145        stop=true;;
    74146      } else {
    75         printf("epoll_add_usock error: %s\n",
    76                UDT::getlasterror().getErrorMessage());
     147        printf("%s: epoll_add_usock error (%s)\n",name.toLocal8Bit().constData(),UDT::getlasterror().getErrorMessage());
    77148      }
    78149    }
    79     if (UDT::epoll_add_usock(eid, com_socket) < 0) {
    80       if (UDT::getlasterror().getErrorCode() == 5004) {
    81         printf("epoll_add_usock error\n");
    82         break;
    83       } else {
    84         printf("epoll_add_usock error: %s\n",
    85                UDT::getlasterror().getErrorMessage());
    86       }
    87     }
    88 
    89     set<UDTSOCKET> readfds;
    90 
    91     int rv = UDT::epoll_wait(eid, &readfds, NULL, 10);
    92 
     150 
     151    int num = 1;
     152    UDTSOCKET readfds;
     153    int rv = UDT::epoll_wait2(eid, &readfds, &num,NULL, NULL,100);
     154 
    93155    if (rv == -1) {
    94156      if (UDT::getlasterror().getErrorCode() != 6003)
    95157        printf("prob %i\n", UDT::getlasterror().getErrorCode());
    96       // printf("wait\n");
    97     } else if (rv == 0) {
    98       printf("timeout\n"); // a timeout occured
     158    } else if(readfds==socket && num==1 && rv==1) {
     159       
     160      int size=UDT::recvmsg(socket, buf, buf_size);
     161      if (size > 0) {
     162        total_received+=size;
     163       
     164        switch ((unsigned char)buf[0]) {
     165          case ZLIB_HEADER: {
     166            ssize_t out_size;
     167            uncompressBuffer(buf, size, uncompressbuf, &out_size);
     168            if((unsigned char)uncompressbuf[0]==XML_HEADER && socketType==unknown) {
     169              socketType=gui;
     170              QString remoteName=ConnectionLayout::getDocRootName(uncompressbuf, out_size);
     171              setName(remoteName);
     172              emit newConnectionLayout(remoteName);//connection is Qt::BlockingQueuedConnection
     173            }
     174            emit dataReady(uncompressbuf, out_size);//connection is Qt::BlockingQueuedConnection, as we have only one buffer
     175            break;
     176          }
     177          case START_SENDING_FILES: {
     178            if((unsigned char)uncompressbuf[0]==XML_HEADER && socketType==unknown) {
     179              socketType=log;
     180            }
     181            setName("log files");
     182            heartbeat_timer->stop();
     183            emit newFileUI(socket);
     184            deleteLater();
     185            stop=true;
     186            destroySocket=false;
     187            break;
     188          }
     189          case XML_HEADER:
     190            if(socketType==unknown) {
     191              socketType=gui;
     192              QString remoteName=ConnectionLayout::getDocRootName(buf, size );
     193              setName(remoteName);
     194              emit newConnectionLayout(remoteName);
     195            }
     196          case DATAS_BIG_ENDIAN:
     197          case DATAS_LITTLE_ENDIAN:
     198            emit dataReady(buf, size );
     199            break;
     200        }
     201      } else {
     202        //if(UDT::getlasterror().getErrorCode()!=6002)
     203        printf("udt socket:%s\n",UDT::getlasterror().getErrorMessage());
     204        //UDT::close(socket);//si deconnecté
     205        //free(buf);
     206        //break;
     207      }
    99208    } else {
    100       /*
    101       if(UDT::getlasterror().getErrorCode()==2001) {
    102           UDT::epoll_release(eid);
    103       }*/
    104       for (set<UDTSOCKET>::iterator i = readfds.begin(); i != readfds.end();
    105            i++) {
    106         // printf("a\n");
    107         if (*i == file_socket)
    108           receiveFile();
    109         if (*i == com_socket)
    110           receiveData();
    111       }
    112     }
    113 
    114     UDT::epoll_remove_usock(eid, file_socket);
    115     UDT::epoll_remove_usock(eid, com_socket);
     209      printf("udt socket:%s\n",UDT::getlasterror().getErrorMessage());
     210    }
     211    UDT::epoll_remove_usock(eid, socket);
    116212    UDT::epoll_release(eid);
    117 
    118     QCoreApplication::processEvents();
    119   }
    120   kill();
    121 }
    122 
    123 void UdtSocket::receiveFile(void) {
    124   char *recv_buf;
    125   int bytesRead;
    126   static bool flag_new_seq = true;
    127   static QString folder_name;
    128 
    129   // receive file info
    130   recv_buf = (char *)malloc(1024);
    131   bytesRead = UDT::recvmsg(file_socket, recv_buf, 1024);
    132 
    133   if (bytesRead <= 0) {
    134     free(recv_buf);
    135     return;
    136   }
    137 
    138   int size;
    139   memcpy(&size, &recv_buf[1], sizeof(int));
    140   if (recv_buf[0] == FILE_INFO_BIG_ENDIAN)
    141     size = qFromBigEndian(size);
    142 
    143   // printf("file_ui recu %i %x\n",bytesRead,recv_buf[0]);
    144   if ((recv_buf[0] == FILE_INFO_LITTLE_ENDIAN ||
    145        recv_buf[0] == FILE_INFO_BIG_ENDIAN) &&
    146       size > 0) {
    147     if (flag_new_seq == true) {
    148       // create directory for storage
    149       QDateTime dateTime = QDateTime::currentDateTime();
    150       folder_name = dateTime.toString("yyyyMMdd_hhmm") + "_" + name;
    151       if (QDir().exists(folder_name) == true) {
    152         folder_name = dateTime.toString("yyyyMMdd_hhmm_ss") + "_" + name;
    153       }
    154       QDir().mkdir(folder_name);
    155 
    156       flag_new_seq = false;
    157       file_dialog->log("Creating directory " + folder_name);
    158     }
    159 
    160     QString file_name =
    161         QString::fromAscii((const char *)&recv_buf[5], bytesRead - 5);
    162     QString file_path = folder_name + "/" + file_name;
    163     file_dialog->log(
    164         QString("receiving %1 (%2 bytes)").arg(file_name).arg(size));
    165     QFile fichier(file_path);
    166 
    167     if (!fichier.open(QIODevice::WriteOnly)) {
    168       file_dialog->log("      could not write to file!");
    169     } else {
    170       // receive file
    171       recv_buf = (char *)realloc((void *)recv_buf, size);
    172       bytesRead = UDT::recvmsg(file_socket, recv_buf, size);
    173       if (bytesRead != size) {
    174         file_dialog->log(QString("      error receiving file! (%1/%2)")
    175                              .arg(bytesRead)
    176                              .arg(size));
    177         free(recv_buf);
    178         return;
    179       } else {
    180         file_dialog->log("      ok");
    181       }
    182 
    183       QDataStream stream(&fichier);
    184       stream.writeRawData(recv_buf, size);
    185       fichier.close();
    186 
    187       file_dialog->addFile(file_path);
    188     }
    189 
    190     free(recv_buf);
    191   } else if (recv_buf[0] == END) {
    192     file_dialog->endOfFiles();
    193     flag_new_seq = true;
    194     //end ack
    195     UDT::sendmsg(file_socket,&recv_buf[0],1);
    196   }
    197 }
    198 
    199 void UdtSocket::receiveData(void) {
    200   while (1) {
    201     int buf_size;
    202     int opt_size;
    203     UDT::getsockopt(com_socket, 0, UDT_RCVBUF, &buf_size, &opt_size);
    204 
    205     char *buf = (char *)malloc(buf_size);
    206     int size;
    207     size = UDT::recvmsg(com_socket, buf, buf_size);
    208     buf = (char *)realloc(buf, size + 1);
    209 
    210     if (size > 0) {
    211       buf[size] = 0;
    212       emit dataReady(buf, size + 1);
    213     } else {
    214       // if(UDT::getlasterror().getErrorCode()!=6002) printf("udt socket:
    215       // %s\n",UDT::getlasterror().getErrorMessage());
    216       free(buf);
    217       break;
    218     }
    219   }
    220 }
    221 
    222 void UdtSocket::write(const char *buf, qint64 size) {
    223   // printf("write\n%s\n",buf);
    224   qint64 sent = UDT::sendmsg(com_socket, buf, size, -1, true);
     213  }
     214  free(uncompressbuf);
     215  free(buf);
     216  thread()->quit();
     217}
     218
     219qint64 UdtSocket::write(const char *buf, qint64 size,int ttl,bool inOrder) {
     220  qint64 sent = UDT::sendmsg(socket, buf, size, ttl, inOrder);
    225221  if (sent != size) {
    226     printf("erreur envoi: %s\n", UDT::getlasterror().getErrorMessage());
     222    printf("%s, error writting to udt (%s)\n",name.toLocal8Bit().constData(), UDT::getlasterror().getErrorMessage());
    227223    if (UDT::getlasterror().getErrorCode() == 2001) {
    228224      stop = true;
    229     }
    230   }
    231 }
     225      heartbeat_timer->stop();
     226      deleteLater();
     227    }
     228  }
     229  return sent;
     230}
     231
     232int UdtSocket::uncompressBuffer(char *in, ssize_t in_size, char *out,
     233                                       ssize_t *out_size) {
     234  int ret;
     235  unsigned have;
     236  z_stream strm;
     237
     238  // allocate inflate state
     239  strm.zalloc = Z_NULL;
     240  strm.zfree = Z_NULL;
     241  strm.opaque = Z_NULL;
     242  strm.avail_in = 0;
     243  strm.next_in = Z_NULL;
     244  ret = inflateInit(&strm);
     245  if (ret != Z_OK)
     246    return ret;
     247
     248  strm.avail_in = in_size;
     249  strm.next_in = (unsigned char *)in;
     250  strm.avail_out = COMPRESS_CHUNK;
     251  strm.next_out = (unsigned char *)out;
     252
     253  ret = inflate(&strm, Z_NO_FLUSH);
     254  assert(ret != Z_STREAM_ERROR); // state not clobbered
     255  switch (ret) {
     256  case Z_NEED_DICT:
     257    ret = Z_DATA_ERROR; // and fall through
     258  case Z_DATA_ERROR:
     259  case Z_MEM_ERROR:
     260    (void)inflateEnd(&strm);
     261    return ret;
     262  }
     263  have = COMPRESS_CHUNK - strm.avail_out;
     264  *out_size = have;
     265
     266  // clean up and return
     267  (void)inflateEnd(&strm);
     268  return ret == Z_STREAM_END ? Z_OK : Z_DATA_ERROR;
     269}
  • trunk/tools/FlairGCS/src/UdtSocket.h

    r15 r234  
    1616
    1717public:
    18   UdtSocket(UDTSOCKET file_socket, UDTSOCKET com_socket, QString name);
     18  UdtSocket(UDTSOCKET socket,QString name);
    1919  ~UdtSocket();
     20  void setName(QString name);
     21  QString getUDTStats();
    2022
    2123private:
    22   UDTSOCKET file_socket, com_socket;
    23   file_ui *file_dialog;
    24   bool stop;
     24  UDTSOCKET socket;
     25  bool stop,destroySocket;
     26  QTimer *heartbeat_timer,*udtstats_timer;
    2527  QString name;
    26   QTimer *heartbeat_timer;
    27   void receiveData(void);
    28   void receiveFile(void);
     28  static int uncompressBuffer(char *in, ssize_t in_size, char *out,ssize_t *out_size);
     29  enum SocketType { unknown, gui, log };
     30  SocketType socketType;
     31  unsigned int total_received;
     32  QString stats;
    2933
    3034signals:
    3135  void dataReady(char *, int size);
     36  void newFileUI(UDTSOCKET socket);
     37  void newConnectionLayout(QString name);
     38  void UDTStats(QString stats);
    3239
    3340public slots:
    34   void handleConnections(void);
     41  void receiveData(void);
    3542  void kill(void);
    36   void write(const char *buf, qint64 size);
     43  qint64 write(const char *buf, qint64 size,int ttl=-1,bool inOrder=true);
     44 
    3745private slots:
    3846  void heartbeat(void);
     47  void getUTDStats(void);
    3948};
    4049
  • trunk/tools/FlairGCS/src/file_ui.cpp

    r214 r234  
    44// %flair:license}
    55#include "file_ui.h"
     6#include "communication.h"
     7
    68#include <stdio.h>
    7 
    89#include <cstring>
    910#include <cstdlib>
     
    2223#include <QStringList>
    2324#include <QFormLayout>
    24 
    25 #ifndef WIN32
    26 #include <arpa/inet.h>
    27 #else
    28 #include <winsock2.h>
    29 #include <ws2tcpip.h>
    30 #endif
    31 
     25#include <QThread>
     26
     27#include <unistd.h>
    3228using namespace std;
    3329
    34 file_ui::file_ui() {
     30file_ui::file_ui(UDTSOCKET socket,QString name): QObject() {
     31  this->socket=socket;
     32  this->name=name;
     33 
     34  bool blocking = true;
     35  if (UDT::setsockopt(socket, 0, UDT_SNDSYN, &blocking, sizeof(bool))!= 0) {
     36    printf("UDT::setsockopt error (UDT_SNDSYN) %s\n",UDT::getlasterror().getErrorMessage());
     37  }
     38   
     39  linger _linger;
     40  _linger.l_onoff=1;
     41  _linger.l_linger=180;
     42 
     43  if (UDT::setsockopt(socket, 0, UDT_LINGER, &_linger, sizeof(struct linger)) != 0)
     44    printf("UDT::setsockopt error (UDT_LINGER) %s\n",UDT::getlasterror().getErrorMessage());
     45   
    3546  dialog = new QDialog();
     47//fprintf(stderr,"creator file ui %x\n",thread());
    3648  dialog->setWindowTitle("log files");
    3749  QGridLayout *main_layout = new QGridLayout(dialog);
     
    4254  input_text->setText("add your log comment here");
    4355  input_cleared = false;
    44 
    4556  ok_button->setEnabled(false);
    4657
     
    5667  main_layout->addWidget(ok_button, 3, 0);
    5768
    58   connect(ok_button, SIGNAL(clicked()), this, SLOT(save()),
    59           Qt::QueuedConnection);
    60   connect(this, SIGNAL(showDialog()), dialog, SLOT(show()),
    61           Qt::QueuedConnection);
    62   connect(this, SIGNAL(appendToLog(QString)), log_text, SLOT(append(QString)),
    63           Qt::QueuedConnection);
    64   connect(input_text, SIGNAL(cursorPositionChanged()), this,
    65           SLOT(clearInputText()), Qt::QueuedConnection);
     69  //connect for multithreaded stuffs
     70  connect(ok_button, SIGNAL(clicked()), this, SLOT(save()),Qt::QueuedConnection);
     71  connect(input_text, SIGNAL(cursorPositionChanged()), this,SLOT(clearInputText()),Qt::DirectConnection);
     72  connect(this, SIGNAL(appendToLog(QString)), log_text, SLOT(append(QString)));
    6673
    6774  file_names = new QStringList();
    68 }
    69 
    70 file_ui::~file_ui() { delete dialog; }
    71 
    72 void file_ui::log(QString text) { appendToLog(text); }
     75 
     76  dialog->show();
     77}
     78
     79file_ui::~file_ui() {
     80  delete dialog;
     81}
     82
     83void file_ui::receive(void) {
     84  char *recv_buf;
     85  int bytesRead;
     86  bool flag_new_seq = true;
     87  QString folder_name;
     88//fprintf(stderr,"file_ui thread %x\n",thread());
     89  while(1) {
     90    // receive file info
     91    recv_buf = (char *)malloc(1024);
     92    bytesRead = UDT::recvmsg(socket, recv_buf, 1024);
     93    if (bytesRead <= 0) {
     94      free(recv_buf);
     95      break;
     96    }
     97
     98    int size;
     99    memcpy(&size, &recv_buf[1], sizeof(int));
     100    if (recv_buf[0] == FILE_INFO_BIG_ENDIAN)
     101      size = qFromBigEndian(size);
     102
     103    // printf("file_ui recu %i %x\n",bytesRead,recv_buf[0]);
     104    if ((recv_buf[0]==FILE_INFO_LITTLE_ENDIAN || recv_buf[0]==FILE_INFO_BIG_ENDIAN) && size>0) {
     105      if (flag_new_seq == true) {
     106        // create directory for storage
     107        QDateTime dateTime = QDateTime::currentDateTime();
     108        folder_name = dateTime.toString("yyyyMMdd_hhmm") + "_" + name;
     109        if (QDir().exists(folder_name) == true) {
     110          folder_name = dateTime.toString("yyyyMMdd_hhmm_ss") + "_" + name;
     111        }
     112        QDir().mkdir(folder_name);
     113
     114        flag_new_seq = false;
     115        appendToLog("Creating directory " + folder_name);
     116      }
     117
     118      QString file_name=QString::fromAscii((const char *)&recv_buf[5], bytesRead - 5);
     119      QString file_path=folder_name+"/"+file_name;
     120      appendToLog(QString("receiving %1 (%2 bytes)").arg(file_name).arg(size));
     121      QFile fichier(file_path);
     122
     123      if (!fichier.open(QIODevice::WriteOnly)) {
     124        appendToLog("      could not write to file!");
     125      } else {
     126        // receive file
     127        recv_buf = (char *)realloc((void *)recv_buf, size);
     128        bytesRead = UDT::recvmsg(socket, recv_buf, size);
     129        if (bytesRead != size) {
     130          appendToLog(QString("      error receiving file! (%1/%2)").arg(bytesRead).arg(size));
     131          free(recv_buf);
     132          break;
     133        } else {
     134          appendToLog("      ok");
     135        }
     136
     137        QDataStream stream(&fichier);
     138        stream.writeRawData(recv_buf, size);
     139        fichier.close();
     140
     141        addFile(file_path);
     142      }
     143
     144      free(recv_buf);
     145    } else if (recv_buf[0] == END_SENDING_FILES) {
     146      //end ack
     147      UDT::sendmsg(socket,&recv_buf[0],1);
     148      endOfFiles();
     149      UDT::close(socket);
     150      printf("disconnected from log files\n");
     151      break;
     152    }
     153  }
     154}
    73155
    74156void file_ui::addFile(QString file_path) {
     
    86168    dbt2csv(file_path.replace(QString(".txt"), QString(".dbt")));
    87169  }
    88 
    89   if (file_names->size() == 1) {
    90     input_cleared = false;
    91     showDialog();
    92   }
    93170}
    94171
     
    101178    if (info.size() > max_file_size) {
    102179      max_file_size = info.size();
    103       csv_combo->setCurrentIndex(i +
    104                                  1); // first item of combobox is already taken
     180      csv_combo->setCurrentIndex(i+1); // first item of combobox is already taken
    105181    }
    106182  }
     
    112188  QStringList data_type;
    113189
    114   QString filename =
    115       file_path.section('/', -1); // remove path for displaying on logs
     190  QString filename =file_path.section('/', -1); // remove path for displaying on logs
    116191  appendToLog(QString("converting %1 to csv").arg(filename));
    117192
     
    145220      break;
    146221    QString txt_line = txt_in.readLine();
    147     data_type.append(txt_line.section(
    148         "(",
    149         -1)); // on part de la fin pour trouver la premiere parenthese ouvrante
     222    data_type.append(txt_line.section("(",-1)); // on part de la fin pour trouver la premiere parenthese ouvrante
    150223    // printf("type %s\n",txt_line.section("(",-1).toLocal8Bit().constData());
    151224  }
     
    241314  }
    242315
    243   log_text->clear();
    244   input_cleared = true; // avoid clearing it with setText
    245   input_text->setText("add your log comment here");
    246   file_names->clear();
    247   csv_combo->clear();
    248   csv_combo->addItem(QString("(no base time)"));
    249 
    250   dialog->setVisible(false);
    251   ok_button->setEnabled(false);
    252316  emit finished();
    253317}
  • trunk/tools/FlairGCS/src/file_ui.h

    r15 r234  
    77
    88#include <QObject>
     9#include <udt.h>
    910
    1011class QStringList;
     
    1920
    2021public:
    21   file_ui();
     22  file_ui(UDTSOCKET socket,QString name);
    2223  ~file_ui();
    2324  void log(QString text);
     
    3637  void dbt2csv(QString file_path);
    3738  bool is_greater(qint64 ref_us, qint64 csv_us, int ref_ns, int csv_ns);
    38   void closeEvent(QCloseEvent *e);
    3939  bool input_cleared;
     40  UDTSOCKET socket;
     41  QString name;
    4042
     43public slots:
     44  void receive(void);
     45 
    4146private slots:
    4247  void save(void);
     
    4449
    4550signals:
    46   void showDialog(void);
     51  void finished(void);
    4752  void appendToLog(QString);
    48   void finished();
    4953};
    5054
  • trunk/tools/FlairGCS/src/main.cpp

    r222 r234  
    66#include <QCleanlooksStyle>
    77#include <QLocale>
     8#include <QTextCursor>
    89#include <qmetatype.h>
    910#include <tclap/CmdLine.h>
     
    7475
    7576  qRegisterMetaType<const char *>("const char*");
     77  qRegisterMetaType<UDTSOCKET>("UDTSOCKET");
     78  qRegisterMetaType<QTextCursor>("QTextCursor");
    7679  QLocale::setDefault(QLocale::C);
    7780  QApplication app(argc, argv);
  • trunk/tools/VrpnBridge/src/main.cpp

    r233 r234  
    5858
    5959  while (1) {
    60     usleep(10000);
     60    usleep(5000);
    6161    server_connection->mainloop();
    6262    client_connection->mainloop();
Note: See TracChangeset for help on using the changeset viewer.