Changeset 234 in flair-src for trunk/lib/FlairCore/src/FrameworkManager_impl.cpp

Timestamp:

Apr 10, 2018, 5:05:27 PM (7 years ago)

Author:

Sanahuja Guillaume

Message:

create file oscket only when necessary

File:

• trunk/lib/FlairCore/src/FrameworkManager_impl.cpp (modified) (16 diffs)

trunk/lib/FlairCore/src/FrameworkManager_impl.cpp

@@ -193 +193 @@
       Printf("Error udt::close %s", UDT::getlasterror().getErrorMessage());
 
-    status = UDT::close(file_sock);
-    if (status != 0)
-      Printf("Error udt::close %s", UDT::getlasterror().getErrorMessage());
-
     SleepMS(200); // a revoir, sinon UDT::cleanup bloque en RT
     UDT::cleanup();
@@ -210 +206 @@
     return;
   }
+  this->address=address;
+  this->port=port;
                                                                                                                                                                                         
         UDT::startup(1024*256,1024*256,1024*256);
@@ -216 +214 @@
   // socket file_socket, doit être créé en premier, cf station sol
   Printf("Connecting to %s:%i\n", address.c_str(), port);
-  file_sock = GetSocket(address, port);
-  com_sock = GetSocket(address, port);
+  //file_sock = GetSocket(address, port);
+  com_sock = GetSocket();
 
   // receive buffer allocation
@@ -226 +224 @@
 
   com = new ui_com(this, com_sock);
-
-  // file managment
-  bool blocking = true;
-  UDT::setsockopt(file_sock, 0, UDT_SNDSYN, &blocking, sizeof(bool));
-  UDT::setsockopt(file_sock, 0, UDT_RCVSYN, &blocking, sizeof(bool));
-
-  int timeout = 100; // ms
-  UDT::setsockopt(com_sock, 0, UDT_RCVTIMEO, &timeout, sizeof(int));
 
   Start();
@@ -292 +282 @@
   while (!ToBeStopped()) {
     ssize_t bytesRead;
-
     bytesRead = com->Receive(rcv_buf, rcv_buf_size);
-
+    com->CheckConnection();
+//printf("%i\n",bytesRead);
     if (bytesRead == (ssize_t)rcv_buf_size)
       Err("FrameworkManager max receive size, augmenter le buffer size!\n");
 
     if (bytesRead > 0) {
-      // printf("recu %ld, trame %x\n",bytesRead,(uint8_t)rcv_buf[0]);
-      // rcv_buf[bytesRead-1]=0;//pour affichage
-      // printf("%s\n",rcv_buf);
+       //printf("recu %ld, trame %x %lld\n",bytesRead,(uint8_t)rcv_buf[0],GetTime()/(1000000));
+       //rcv_buf[bytesRead-1]=0;//pour affichage
+       //printf("%s\n",rcv_buf);
 
       switch ((uint8_t)rcv_buf[0]) {
@@ -376 +366 @@
 }
 
-void FrameworkManager_impl::SendFile(string path, string name) {
+void FrameworkManager_impl::SendFile(UDTSOCKET socket,string path, string name) {
   char *buf, *more_buf;
   int size;
@@ -411 +401 @@
   Printf("sending %s, size: %i\n", filename.c_str(), size);
   // send file information
-  UDT::sendmsg(file_sock, buf, sizeof(uint8_t) + sizeof(int) + name.size(), -1,true);
+  UDT::sendmsg(socket, buf, sizeof(uint8_t) + sizeof(int) + name.size(), -1,true);
 
   more_buf = (char *)realloc((void *)buf, size);
@@ -424 +414 @@
   pbuf->sgetn(buf, size);
   // send the file
-  nb_write = UDT::sendmsg(file_sock, buf, size, -1, true);
+  nb_write = UDT::sendmsg(socket, buf, size, -1, true);
 
   if (nb_write < 0) {
@@ -436 +426 @@
 }
 
-void FrameworkManager_impl::FinishSending() {
+void FrameworkManager_impl::FinishSending(UDTSOCKET socket) {
   char rm_cmd[256];
 
   // send orignal xml
-  SendFile(log_path, "setup.xml");
+  SendFile(socket,log_path, "setup.xml");
   sprintf(rm_cmd, "rm %s/setup.xml", log_path.c_str());
   system(rm_cmd);
@@ -447 +437 @@
   for (size_t i = 0; i < xml_changes.size(); i++) {
     // Printf("%s\n",xml_changes.at(i).c_str());
-    SendFile(log_path, xml_changes.at(i).c_str());
+    SendFile(socket,log_path, xml_changes.at(i).c_str());
     sprintf(rm_cmd, "rm %s/%s", log_path.c_str(), xml_changes.at(i).c_str());
     system(rm_cmd);
@@ -454 +444 @@
 
   // end notify
-  char buf = END;
-  int nb_write = UDT::sendmsg(file_sock, &buf, 1, -1, true);
+  char buf = END_SENDING_FILES;
+  int nb_write = UDT::sendmsg(socket, &buf, 1, -1, true);
 
   if (nb_write < 0) {
@@ -464 +454 @@
 
   //wait end ACK
-  int nb_read = UDT::recvmsg(file_sock,&buf,1);
+  int nb_read = UDT::recvmsg(socket,&buf,1);
   if(nb_read<0) {
     Err("UDT::recvmsg error (%s)\n",UDT::getlasterror().getErrorMessage());
@@ -472 +462 @@
 }
 
-UDTSOCKET FrameworkManager_impl::GetSocket(string address, uint16_t port) {
+UDTSOCKET FrameworkManager_impl::GetSocket(void) {
   while (1) {
     UDTSOCKET new_fd;
@@ -854 +844 @@
               caller->gcs_watchdog->Join();
           }
+          //create a socket for files
+          UDTSOCKET file_sock = caller->GetSocket();
+          char data=START_SENDING_FILES;
+          ssize_t nb_write = UDT::sendmsg(file_sock, &data, 1, -1, true);
+          if (nb_write < 0) {
+            caller->Err("UDT::sendmsg error (%s)\n", UDT::getlasterror().getErrorMessage());
+            if (UDT::getlasterror().getErrorCode() == CUDTException::ECONNLOST ||
+                UDT::getlasterror().getErrorCode() == CUDTException::EINVSOCK) {
+            }
+          } else if (nb_write != 1) {
+            caller->Err("%s, code %i (%ld/%ld)\n", UDT::getlasterror().getErrorMessage(),
+                UDT::getlasterror().getErrorCode(), nb_write, 1);
+          }
           for (size_t i = 0; i < logs.size(); i++) {
             if (logs.at(i).dbtFile != NULL) {
@@ -859 +862 @@
 
               string filename = caller->FileName(logs.at(i).device) + ".dbt";
-              caller->SendFile(caller->log_path, filename);
+              caller->SendFile(file_sock,caller->log_path, filename);
 
               fstream txt_file;
@@ -870 +873 @@
               txt_file.close();
 
-              caller->SendFile(caller->log_path, filename);
+              caller->SendFile(file_sock,caller->log_path, filename);
             }
           }
           // a revoir celui ci est le xml enregistré et pas forcement l'actuel
           // if(caller->xml_file!="") caller->SendFile(caller->xml_file);
-          caller->FinishSending();
+          caller->FinishSending(file_sock);
+          
+          int status = UDT::close(file_sock);
+          if (status != 0)
+            caller->Err("Error udt::close %s", UDT::getlasterror().getErrorMessage());
 
           logs.clear();