source: flair-src/trunk/lib/FlairCore/src/ui_com.cpp@ 454

Last change on this file since 454 was 453, checked in by Sanahuja Guillaume, 3 years ago

add a security warning

File size: 21.3 KB
RevLine 
[2]1// %flair:license{
[15]2// This file is part of the Flair framework distributed under the
3// CECILL-C License, Version 1.0.
[2]4// %flair:license}
5// created: 2011/05/01
6// filename: ui_com.cpp
7//
8// author: Guillaume Sanahuja
9// Copyright Heudiasyc UMR UTC/CNRS 7253
10//
11// version: $Id: $
12//
13// purpose: classe permettant la lecture et l'ecriture sur socket UDT
14//
15//
16/*********************************************************************/
17
18#include "ui_com.h"
19#include <cstdlib>
20#include <string.h>
21#include <unistd.h>
22#include "Mutex.h"
23#include "SendData.h"
24#include "communication.h"
25#include "FrameworkManager.h"
26#include "zlib.h"
27#include <assert.h>
28#include "config.h"
29
30#ifdef __XENO__
31#include <pthread.h>
32#include <native/task.h>
33#include <native/task.h>
34#include <fcntl.h>
35#endif
36
37using std::string;
38using namespace flair::core;
39using namespace flair::gui;
40
[445]41ui_com::ui_com(const Object *parent, UDTSOCKET sock) : Thread(parent, "send", 2,16384*2) {
[15]42 // mutex
43 send_mutex = new Mutex(this, ObjectName());
[2]44
[15]45 socket_fd = sock;
46 connection_lost = false;
[2]47
48#ifdef __XENO__
[442]49 Err("multi buffering is not well implemented in RT\n");
50#endif
51
52#ifdef __XENO__
[15]53 int status;
54 string tmp_name;
[2]55
[15]56 is_running = true;
[2]57
[15]58 // pipe
59 tmp_name = getFrameworkManager()->ObjectName() + "-" + ObjectName() + "-pipe";
60 // xenomai limitation
61 if (tmp_name.size() > 31)
62 Err("rt_pipe_create error (%s is too long)\n", tmp_name.c_str());
[2]63#ifdef RT_PIPE_SIZE
[15]64 status = rt_pipe_create(&pipe, tmp_name.c_str(), P_MINOR_AUTO, RT_PIPE_SIZE);
[2]65#else
[15]66 status = rt_pipe_create(&pipe, tmp_name.c_str(), P_MINOR_AUTO, 0);
[2]67#endif
68
[15]69 if (status != 0) {
[133]70 char errorMsg[256];
71 Err("rt_pipe_create error (%s)\n", strerror_r(-status, errorMsg, sizeof(errorMsg)));
[15]72 // return -1;
73 }
[2]74
[15]75// start user side thread
[2]76#ifdef NRT_STACK_SIZE
[15]77 // Initialize thread creation attributes
78 pthread_attr_t attr;
79 if (pthread_attr_init(&attr) != 0) {
80 Err("pthread_attr_init error\n");
81 }
[2]82
[15]83 if (pthread_attr_setstacksize(&attr, NRT_STACK_SIZE) != 0) {
84 Err("pthread_attr_setstacksize error\n");
85 }
[2]86
[15]87 if (pthread_create(&thread, &attr, user_thread, (void *)this) < 0)
88#else // NRT_STACK_SIZE
89 if (pthread_create(&thread, NULL, user_thread, (void *)this) < 0)
90#endif // NRT_STACK_SIZE
91 {
92 Err("pthread_create error\n");
93 // return -1;
94 }
[2]95#ifdef NRT_STACK_SIZE
[15]96 if (pthread_attr_destroy(&attr) != 0) {
97 Err("pthread_attr_destroy error\n");
98 }
[2]99#endif
100
[15]101#endif //__XENO__
102 int timeout = 100;
103 if (UDT::setsockopt(socket_fd, 0, UDT_RCVTIMEO, &timeout, sizeof(int)) != 0)
104 Err("UDT::setsockopt error (UDT_RCVTIMEO)\n");
[247]105 /*
106 timeout=-1;
107 if (UDT::setsockopt(socket_fd, 0, UDT_SNDTIMEO, &timeout, sizeof(int)) != 0)
108 Err("UDT::setsockopt error (UDT_SNDTIMEO)\n");
109*/
[15]110 bool blocking = true;
111 if (UDT::setsockopt(socket_fd, 0, UDT_SNDSYN, &blocking, sizeof(bool)) != 0)
112 Err("UDT::setsockopt error (UDT_SNDSYN)\n");
[2]113
[15]114 if (UDT::setsockopt(socket_fd, 0, UDT_RCVSYN, &blocking, sizeof(bool)) != 0)
115 Err("UDT::setsockopt error (UDT_RCVSYN)\n");
116 //#endif //__XENO__
[2]117
[15]118 Start();
[2]119}
120
[15]121ui_com::~ui_com() {
[243]122//Printf("destruction ui_com\n");
[2]123
124#ifdef __XENO__
[15]125 is_running = false;
[2]126
[15]127 pthread_join(thread, NULL);
[2]128
[15]129 int status = rt_pipe_delete(&pipe);
[133]130 if (status != 0) {
131 char errorMsg[256];
132 Err("rt_pipe_delete error (%s)\n", strerror_r(-status, errorMsg, sizeof(errorMsg)));
133 }
[2]134#endif
135
[15]136 SafeStop();
[2]137
[15]138 if (IsSuspended() == true)
139 Resume();
[2]140
[15]141 Join();
[247]142
143 char buf=CLOSING_CONNECTION;
144 Send(&buf,1);
145
[445]146 for (size_t i = 0; i < pushedDatas.size(); i++) {
147 free(pushedDatas.at(i).buf);
148 }
[243]149 //Printf("destruction ui_com ok\n");
[2]150}
151
[439]152//send, public part; dispatch to SendNRT (nrt) or rt_pipe (rt)
153//rt_pipe is read by user thread which calls SendNRT
[252]154void ui_com::Send(char *buf, ssize_t size,int ttl) {
[15]155 if (connection_lost == true)
156 return;
[2]157
[15]158 char *tosend = buf;
159 if (buf[0] == XML_HEADER) {
160 // cut xml header
161 tosend = strstr(buf, "<root");
162 size -= tosend - buf;
163 }
[2]164
165#ifdef __XENO__
[439]166 //data buf
167 ssize_t nb_write = rt_pipe_write(&pipe, tosend, size, P_NORMAL);
[15]168 if (nb_write < 0) {
[133]169 char errorMsg[256];
170 Err("rt_pipe_write error (%s)\n", strerror_r(-nb_write, errorMsg, sizeof(errorMsg)));
[15]171 } else if (nb_write != size) {
172 Err("rt_pipe_write error %i/%i\n", nb_write, size);
173 }
[439]174 //ttl
175 nb_write = rt_pipe_write(&pipe, &ttl, sizeof(ttl), P_NORMAL);
176 if (nb_write < 0) {
177 char errorMsg[256];
178 Err("rt_pipe_write error (%s)\n", strerror_r(-nb_write, errorMsg, sizeof(errorMsg)));
179 } else if (nb_write != sizeof(ttl)) {
180 Err("rt_pipe_write error %i/%i\n", nb_write, size);
181 }
[2]182#else //__XENO__
[439]183 SendNRT(tosend,size,ttl);
184#endif //__XENO__
185}
[2]186
[439]187//send, private part; called to effectively send to udt
188//this is always in nrt
189void ui_com::SendNRT(char *buf, ssize_t size,int ttl) {
190 ssize_t nb_write;
191 bool sendItCompressed = false;
192
[2]193#ifdef COMPRESS_FRAMES
[15]194 if (buf[0] == XML_HEADER) {
195 sendItCompressed = true;
196 }
[439]197#endif // COMPRESS_FRAMES
[2]198
[15]199 if (sendItCompressed) {
200 char *out;
201 ssize_t out_size;
[439]202 if (compressBuffer(buf, size, &out, &out_size, 9) == Z_OK) {
[15]203 size = out_size;
[252]204 nb_write = UDT::sendmsg(socket_fd, out, size,ttl, true);
[15]205 free(out);
206 } else {
207 Warn("Compress error, sending it uncompressed\n");
208 sendItCompressed = false;
[2]209 }
[15]210 }
[2]211
[15]212 if (!sendItCompressed) {
[439]213 nb_write = UDT::sendmsg(socket_fd, buf, size, ttl, true);
[15]214 }
[2]215
[247]216 //Printf("write %i %i\n",nb_write,size);
[15]217 if (nb_write < 0) {
218 Err("UDT::sendmsg error (%s)\n", UDT::getlasterror().getErrorMessage());
219 if (UDT::getlasterror().getErrorCode() == CUDTException::ECONNLOST ||
220 UDT::getlasterror().getErrorCode() == CUDTException::EINVSOCK) {
221 connection_lost = true;
[2]222 }
[15]223 } else if (nb_write != size) {
224 Err("%s, code %i (%ld/%ld)\n", UDT::getlasterror().getErrorMessage(),
225 UDT::getlasterror().getErrorCode(), nb_write, size);
226 }
[2]227}
228
[15]229ssize_t ui_com::Receive(char *buf, ssize_t buf_size) {
230 ssize_t bytesRead = UDT::recvmsg(socket_fd, buf, buf_size);
[2]231
[15]232 if (bytesRead < 0) {
233 if (UDT::getlasterror().getErrorCode() == CUDTException::ECONNLOST) {
234 Err("UDT::recvmsg error (%s)\n", UDT::getlasterror().getErrorMessage());
235 connection_lost = true;
[2]236 }
[15]237 }
[2]238
[15]239 return bytesRead;
[2]240}
241
[234]242void ui_com::CheckConnection(void) {
243 int32_t val,len;
244
245 int result=UDT::getsockopt(socket_fd,0,UDT_STATE,&val,&len);
246 if (result < 0) {
247 Printf("UDT::getsockopt error (%s)\n", UDT::getlasterror().getErrorMessage());
248 }
249 // printf("opt: %i %i %i\n",result,val,len);
250}
251
[439]252//todo: le run (rt) push les données une par une
253//le nrt recupere et decide quand envoyer :
254// -quand toutes les données d'une meme periode sont recues
255// -quand on a atteind le nb buffering
256// -le nrt alloue dynamiquement le buffer d'envoi
257//-> enelver l'alloc du buffer au startup (UpdateDataToSendSize)
[15]258void ui_com::Run(void) {
[2]259#ifdef __XENO__
[15]260 WarnUponSwitches(true);
261 printf("\n"); // a revoir pourquoi??
262// sans ce printf, dans le simu_roll, le suspend ne fonctionne pas...
[2]263#endif
[15]264 // on attend d'avoir des choses à faire
265 Suspend();
[2]266
[15]267 while (!ToBeStopped()) {
268 size_t resume_id;
269 Time min = 0xffffffffffffffffULL;
[2]270
[15]271 // on recpuere l'id de la prochaine execution
272 send_mutex->GetMutex();
[441]273 resume_id = resumeTimes.size();
274 for (size_t i = 0; i < resumeTimes.size(); i++) {
275 if (resumeTimes.at(i) < min && datasToSend.at(i)->IsEnabled() == true) {
276 min = resumeTimes.at(i);
[15]277 resume_id = i;
278 }
279 }
[2]280
[15]281 // attente
[441]282 if (resume_id < resumeTimes.size()) {
283 Time time = resumeTimes.at(resume_id);
284 uint16_t resume_period = datasToSend.at(resume_id)->SendPeriod();
[15]285 send_mutex->ReleaseMutex();
286 // on dort jusqu'a la prochaine execution
287 SleepUntil(time);
[2]288
[15]289 // envoi des donnees
290 send_mutex->GetMutex();
[2]291
[445]292 //mulit buffering
[441]293 for (size_t i = 0; i < datasToSend.size(); i++) {
294 if (datasToSend.at(i)->SendPeriod() == resume_period && datasToSend.at(i)->IsEnabled() == true) {
295 PushDatasToSend(datasToSend.at(i));
[440]296 }
297 }
298
[15]299 // on planifie la prochaine execution
[441]300 for (size_t i = 0; i < datasToSend.size(); i++) {
301 if (datasToSend.at(i)->SendPeriod() == resume_period) {
302 resumeTimes.at(i) += datasToSend.at(i)->SendPeriod() * 1000000;
[2]303 }
[15]304 }
305 send_mutex->ReleaseMutex();
[157]306 //Printf("%i %lld\n",resume_period,GetTime()/1000000);
[15]307 } else {
308 send_mutex->ReleaseMutex();
309 // rien a faire, suspend
[157]310 //Printf("rien a faire suspend\n");
[15]311 Suspend();
[157]312 //Printf("wake\n");
[15]313 // on planifie la prochaine execution
314 Time time = GetTime();
315 send_mutex->GetMutex();
[441]316 for (size_t i = 0; i < datasToSend.size(); i++) {
317 resumeTimes.at(i) =
318 time + (Time)datasToSend.at(i)->SendPeriod() * 1000000;
[15]319 }
320 send_mutex->ReleaseMutex();
[2]321 }
[15]322 }
[2]323}
[440]324
325
326//called with send_mutex locked
327//in nrt
[441]328//TODO: RT part has to put datas in pipe, then user thread call this
329void ui_com::PushDatasToSend(const SendData *dataToSend) {
[440]330 PushedData_t* ptr=NULL;
331 //check if we already have one buffer for this couple period/nb_buffering
[441]332 for (size_t i = 0; i < pushedDatas.size(); i++) {
333 if (pushedDatas.at(i).period==dataToSend->SendPeriod() && pushedDatas.at(i).nb_buffering==dataToSend->NbBuffering()) {
334 ptr=&pushedDatas.at(i);
[440]335 }
336 }
337 if(ptr==NULL) {
[441]338 Warn("no match in buffers\n");
339 return;
[440]340 }
341
[442]342 if(ptr->actual_size==0) {
[445]343 if(ptr->nb_buffering>1) {
344 if (IsBigEndian()) {
345 ptr->buf[0] = MULTIPLE_DATA_BIG_ENDIAN;
346 } else {
347 ptr->buf[0] = MULTIPLE_DATA_LITTLE_ENDIAN;
348 }
349 ptr->actual_size+=sizeof(char);
350 memcpy(ptr->buf+ptr->actual_size,&(ptr->period),sizeof(uint16_t));
351 ptr->actual_size+=sizeof(uint16_t);
352 memcpy(ptr->buf+ptr->actual_size,&(ptr->nb_buffering),sizeof(uint16_t));
353 ptr->actual_size+=sizeof(uint16_t);
[442]354 } else {
[445]355 if (IsBigEndian()) {
356 ptr->buf[0] = DATA_BIG_ENDIAN;
357 } else {
358 ptr->buf[0] = DATA_LITTLE_ENDIAN;
359 }
360 ptr->actual_size+=sizeof(char);
361 memcpy(ptr->buf+ptr->actual_size,&(ptr->period),sizeof(uint16_t));
362 ptr->actual_size+=sizeof(uint16_t);
[442]363 }
364 }
365
[453]366 if(ptr->actual_size+dataToSend->SendSize()>ptr->final_size) {
367 Warn("out of buf %i/%i\n",ptr->actual_size+dataToSend->SendSize(),ptr->final_size);
368 return;
369 }
370
[441]371 dataToSend->CopyDatas(ptr->buf+ptr->actual_size);
372 ptr->actual_size+=dataToSend->SendSize();
[453]373
[444]374 //Printf("nb buffered %i/%i\n",ptr->actual_size,ptr->final_size);
375 //Printf("pushed size %i period %i nb buffering %i\n",dataToSend->SendSize(),dataToSend->SendPeriod(),dataToSend->NbBuffering());
[442]376 if(ptr->actual_size>=ptr->final_size) {//par securité on test aussi le cas ou supérieur
[444]377 //Printf("ready to send\n");
[443]378 SendNRT(ptr->buf, ptr->actual_size,ptr->period);
[441]379 //clean
380 ptr->actual_size=0;
381 }
[440]382}
383
384
[2]385#ifdef __XENO__
[15]386void *ui_com::user_thread(void *arg) {
387 int pipe_fd = -1;
388 string devname;
389 char *buf = NULL;
390 ui_com *caller = (ui_com *)arg;
391 int rv;
392 fd_set set;
393 struct timeval timeout;
[439]394 ssize_t nb_read;
395 int ttl;
[2]396
[15]397 buf = (char *)malloc(NRT_PIPE_SIZE);
398 if (buf == NULL) {
399 caller->Err("malloc error\n");
400 }
[2]401
[15]402 devname = NRT_PIPE_PATH + getFrameworkManager()->ObjectName() + "-" +
403 caller->ObjectName() + "-pipe";
404 while (pipe_fd < 0) {
405 pipe_fd = open(devname.c_str(), O_RDWR);
[133]406 if (pipe_fd < 0 && errno != ENOENT) {
407 char errorMsg[256];
408 caller->Err("open pipe_fd error (%s)\n", strerror_r(errno, errorMsg, sizeof(errorMsg)));
409 }
[15]410 usleep(1000);
411 }
[2]412
[15]413 while (1) {
414 FD_ZERO(&set); // clear the set
415 FD_SET(pipe_fd, &set); // add our file descriptor to the set
[2]416
[15]417 timeout.tv_sec = 0;
418 timeout.tv_usec = SELECT_TIMEOUT_MS * 1000;
[2]419
[15]420 rv = select(FD_SETSIZE, &set, NULL, NULL, &timeout);
[2]421
[15]422 if (rv == -1) {
[439]423 if (caller->is_running == false && UDT::getlasterror().getErrorCode() == CUDTException::ETIMEOUT)
[15]424 break; // timeout
425 if (UDT::getlasterror().getErrorCode() != CUDTException::ETIMEOUT)
426 caller->Err("epoll_wait, %s, code %i\n",
[439]427 UDT::getlasterror().getErrorMessage(), UDT::getlasterror().getErrorCode());
[15]428 } else if (rv == 0) {
429 // printf("timeout\n"); // a timeout occured
430 if (caller->is_running == false)
431 break;
[2]432
[15]433 } else {
[439]434 //read data buf and ttl
435 //RT part has made 2 rt_pipe_write; they should be available in nrt
436 //(select ensure at least one data is ready)
[15]437 nb_read = read(pipe_fd, buf, NRT_PIPE_SIZE);
438 buf[nb_read] = 0;
[439]439 read(pipe_fd, &ttl, sizeof(ttl));
[15]440// printf("envoi\n%s\n",buf);
[2]441
[439]442 caller->SendNRT(buf, nb_read,ttl);
[2]443 }
[15]444 }
[2]445
[15]446 close(pipe_fd);
447 if (buf != NULL)
448 free(buf);
449 pthread_exit(0);
[2]450}
451#endif
452
[15]453int ui_com::compressBuffer(char *in, ssize_t in_size, char **out,
454 ssize_t *out_size, int level) {
455 int ret, flush;
456 unsigned have;
457 z_stream strm;
[2]458
[15]459 /* allocate deflate state */
460 strm.zalloc = Z_NULL;
461 strm.zfree = Z_NULL;
462 strm.opaque = Z_NULL;
463 ret = deflateInit(&strm, level);
464 if (ret != Z_OK)
465 return ret;
[2]466
[15]467 *out = (char *)malloc(COMPRESS_CHUNK);
468 if (!(*out))
469 return Z_BUF_ERROR;
[2]470
[15]471 strm.next_in = (unsigned char *)in;
472 strm.avail_out = COMPRESS_CHUNK;
473 strm.next_out = (unsigned char *)*out;
474 strm.avail_in = in_size;
475 flush = Z_FINISH;
[2]476
[15]477 ret = deflate(&strm, flush); /* no bad return value */
478 if (ret == Z_STREAM_ERROR) {
479 free(*out);
480 return ret;
481 }
[2]482
[15]483 have = COMPRESS_CHUNK - strm.avail_out;
484 *out_size = have;
485 // printf("%i -> %i\n",in_size,have);
486 /* clean up and return */
487 (void)deflateEnd(&strm);
[2]488
[15]489 if (strm.avail_out != 0) {
490 return Z_OK;
491 } else {
492 return Z_STREAM_ERROR;
493 }
[2]494}
495
[15]496int ui_com::uncompressBuffer(unsigned char *in, ssize_t in_size,
497 unsigned char **out, ssize_t *out_size) {
498 int ret;
499 // unsigned have;
500 z_stream strm;
[2]501
[15]502 /* allocate inflate state */
503 strm.zalloc = Z_NULL;
504 strm.zfree = Z_NULL;
505 strm.opaque = Z_NULL;
506 strm.avail_in = 0;
507 strm.next_in = Z_NULL;
508 ret = inflateInit(&strm);
509 if (ret != Z_OK)
510 return ret;
[2]511
[15]512 *out = (unsigned char *)malloc(COMPRESS_CHUNK);
513 if (!(*out))
514 return Z_BUF_ERROR;
[2]515
[15]516 strm.avail_in = in_size;
517 strm.next_in = in;
518 strm.avail_out = COMPRESS_CHUNK;
519 strm.next_out = *out;
[2]520
[15]521 ret = inflate(&strm, Z_NO_FLUSH);
522 assert(ret != Z_STREAM_ERROR); /* state not clobbered */
523 switch (ret) {
524 case Z_NEED_DICT:
525 ret = Z_DATA_ERROR; /* and fall through */
526 case Z_DATA_ERROR:
527 case Z_MEM_ERROR:
[2]528 (void)inflateEnd(&strm);
[15]529 return ret;
530 }
531 // have = COMPRESS_CHUNK - strm.avail_out;
[2]532
[15]533 /* clean up and return */
534 (void)inflateEnd(&strm);
535 return ret == Z_STREAM_END ? Z_OK : Z_DATA_ERROR;
[2]536}
537
[15]538bool ui_com::ConnectionLost() { return connection_lost; }
[2]539
[15]540void ui_com::AddSendData(const SendData *obj) {
541 send_mutex->GetMutex();
[2]542
[441]543 resumeTimes.push_back(GetTime() + (Time)obj->SendPeriod() * 1000000);
[15]544
545 // on resume en meme temps tout ceux qui ont la meme periode
[441]546 for (size_t i = 0; i < datasToSend.size(); i++) {
547 if (datasToSend.at(i)->SendPeriod() == obj->SendPeriod()) {
548 resumeTimes.at(resumeTimes.size() - 1) = resumeTimes.at(i);
[15]549 break;
[2]550 }
[15]551 }
[2]552
[441]553 datasToSend.push_back(obj);
[2]554
[15]555 send_mutex->ReleaseMutex();
[2]556}
557
[441]558//TODO: check if it is RT compatible
[442]559//must be called with mutex locked
[15]560void ui_com::UpdateSendData(const SendData *obj) {
[452]561 // Printf("UpdateSendData %s\n",obj->ObjectName().c_str());
[15]562 // le mutex est deja pris par l'appellant
563 size_t id, i;
[2]564
[15]565 // on recupere l'id
[441]566 for (i = 0; i < datasToSend.size(); i++) {
567 if (datasToSend.at(i) == obj) {
[15]568 id = i;
569 break;
[2]570 }
[15]571 }
[2]572
[15]573 // on resume en meme temps tout ceux qui ont la meme periode
[441]574 for (i = 0; i < datasToSend.size(); i++) {
[15]575 if (i == id)
576 continue;
[441]577 if (datasToSend.at(i)->IsEnabled() == true && datasToSend.at(i)->SendPeriod() == obj->SendPeriod()) {
578 resumeTimes.at(id) = resumeTimes.at(i);
[15]579 break;
[2]580 }
[15]581 }
[2]582
[15]583 // si aucun match, on planifie l'execution
[441]584 if (i == datasToSend.size())
585 resumeTimes.at(id) = GetTime() + (Time)obj->SendPeriod() * 1000000;
[2]586
[441]587 if (IsSuspended() == true) Resume();
[2]588
[441]589//update all buffers if necessary, discard datas
590 std::vector<PushedData_t>::iterator pushedDatasIterator = pushedDatas.begin();
591 while(pushedDatasIterator != pushedDatas.end()) {
592 size_t bufSize=0;
593 for (size_t i = 0; i < datasToSend.size(); i++) {
594 if (datasToSend.at(i)->IsEnabled() && datasToSend.at(i)->SendPeriod() == (*pushedDatasIterator).period && datasToSend.at(i)->NbBuffering()==(*pushedDatasIterator).nb_buffering) {
595 bufSize+=datasToSend.at(i)->SendSize();
596 }
597 }
598 bufSize*=(*pushedDatasIterator).nb_buffering;
[445]599 if(bufSize!=0) {
600 bufSize+=sizeof(char)+sizeof(uint16_t);//header+period
601 if((*pushedDatasIterator).nb_buffering>1) bufSize+=sizeof(uint16_t);//+nb_buffering
602 }
[441]603 if(bufSize!=(*pushedDatasIterator).final_size && bufSize!=0) {
[444]604 //Printf("change buf size %i->%i\n",(*pushedDatasIterator).final_size,bufSize);
[441]605 (*pushedDatasIterator).actual_size=0;
606 (*pushedDatasIterator).final_size=bufSize;
607 free((*pushedDatasIterator).buf);
608 (*pushedDatasIterator).buf=(char*)malloc(bufSize);
609 }
610 if(bufSize==0) {
[444]611 //Printf("delete buf\n");
[441]612 free((*pushedDatasIterator).buf);
613 pushedDatasIterator = pushedDatas.erase(pushedDatasIterator);
614 } else {
615 ++pushedDatasIterator;
616 }
617 }
618
619 //check if we need a new buffer for this couple period/nb_buffering
620 if(obj->IsEnabled()) {
621 bool match=false;
622 for (size_t i = 0; i < pushedDatas.size(); i++) {
623 if (pushedDatas.at(i).period==obj->SendPeriod() && pushedDatas.at(i).nb_buffering==obj->NbBuffering()) {
[452]624 //Printf("match item %i, period %i nb buff %i, size %i\n",i,obj->SendPeriod(),obj->NbBuffering(),pushedDatas.at(i).final_size);
[441]625 match=true;
626 }
627 }
628 //create new one
629 if(!match) {
[444]630 //printf("no match\n");
[441]631 PushedData_t tmp;
632 tmp.period=obj->SendPeriod();
633 tmp.nb_buffering=obj->NbBuffering();
634 tmp.actual_size=0;
635 tmp.final_size=0;
636 for (size_t i = 0; i < datasToSend.size(); i++) {
[452]637 //Printf("data %s %i/%i %i %i\n",datasToSend.at(i)->ObjectName().c_str(),i,datasToSend.size(),datasToSend.at(i)->SendPeriod() ,datasToSend.at(i)->NbBuffering());
[441]638 if (datasToSend.at(i)->IsEnabled() && datasToSend.at(i)->SendPeriod() == obj->SendPeriod() && datasToSend.at(i)->NbBuffering()==obj->NbBuffering()) {
639 tmp.final_size+=datasToSend.at(i)->SendSize();
[444]640 //Printf("add %i\n",datasToSend.at(i)->SendSize());
[441]641 }
642 }
643 tmp.final_size*=obj->NbBuffering();
[445]644 tmp.final_size+=sizeof(char)+sizeof(uint16_t);//header+period
645 if(tmp.nb_buffering>1) tmp.final_size+=sizeof(uint16_t);//+nb_buffering
[444]646 //printf("final size %i\n",tmp.final_size);
[441]647 tmp.buf=(char*)malloc(tmp.final_size);
648 if(tmp.buf!=NULL) {
649 pushedDatas.push_back(tmp);
650 } else {
651 Warn("could not allocate buffer of size %i\n",tmp.final_size);
652 }
653 }
654 }
[444]655 //Printf("nb buf %i\n",pushedDatas.size());
[15]656 return;
[2]657}
658
[442]659void ui_com::UpdateDataToSendBufferSize(const SendData *obj) {
[444]660 if(!obj->IsEnabled()) return;
[452]661 //Printf("UpdateDataToSendBufferSize %s\n",obj->ObjectName().c_str());
[442]662 send_mutex->GetMutex();
663
664 PushedData_t* ptr=NULL;
665 //get buffer for this couple period/nb_buffering
666 for (size_t i = 0; i < pushedDatas.size(); i++) {
667 if (pushedDatas.at(i).period==obj->SendPeriod() && pushedDatas.at(i).nb_buffering==obj->NbBuffering()) {
668 ptr=&pushedDatas.at(i);
669 }
670 }
671 if(ptr==NULL) {
672 Err("no corresponding match for couple %i %i\n",obj->SendPeriod(),obj->NbBuffering());
673 return;
674 }
675
[452]676 ptr->final_size=0;
677 //Printf("actual %i\n",ptr->final_size);
678 for (size_t i = 0; i < datasToSend.size(); i++) {
679 //Printf("data %s %i/%i %i %i\n",datasToSend.at(i)->ObjectName().c_str(),i,datasToSend.size(),datasToSend.at(i)->SendPeriod() ,datasToSend.at(i)->NbBuffering());
680 if (datasToSend.at(i)->IsEnabled() && datasToSend.at(i)->SendPeriod() == obj->SendPeriod() && datasToSend.at(i)->NbBuffering()==obj->NbBuffering()) {
681 ptr->final_size+=datasToSend.at(i)->SendSize();
682 //Printf("add %i\n",datasToSend.at(i)->SendSize());
683 }
684 }
685 ptr->final_size*=obj->NbBuffering();
686 ptr->final_size+=sizeof(char)+sizeof(uint16_t);//header+period
[445]687 if(ptr->nb_buffering>1) ptr->final_size+=sizeof(uint16_t);//+nb_buffering
[452]688
[445]689 //Printf("final %i\n",ptr->final_size);
[442]690 free(ptr->buf);
691 ptr->buf=(char*)malloc(ptr->final_size);
692
693 send_mutex->ReleaseMutex();
694}
695
[15]696void ui_com::RemoveSendData(const SendData *obj) {
697 // printf("remove_data_to_send %i\n",data_to_send.size());
[2]698
[15]699 send_mutex->GetMutex();
700 // on recupere l'id
[441]701 for (size_t i = 0; i < datasToSend.size(); i++) {
702 if (datasToSend.at(i) == obj) {
703 datasToSend.erase(datasToSend.begin() + i);
704 resumeTimes.erase(resumeTimes.begin() + i);
[15]705 // printf("remove_data_to_send %i ok\n",data_to_send.size());
706 break;
[2]707 }
[15]708 }
709 send_mutex->ReleaseMutex();
[2]710
[15]711 return;
[2]712}
713
[15]714void ui_com::Block(void) { send_mutex->GetMutex(); }
[2]715
[15]716void ui_com::UnBlock(void) { send_mutex->ReleaseMutex(); }
Note: See TracBrowser for help on using the repository browser.