source: lib/trace_parallel.c @ 29bbef0

4.0.1-hotfixescachetimestampsdevelopdpdk-ndagetsilivelibtrace4ndag_formatpfringrc-4.0.1rc-4.0.2rc-4.0.3rc-4.0.4ringdecrementfixringperformanceringtimestampfixes
Last change on this file since 29bbef0 was 29bbef0, checked in by Richard Sanger <rsangerarj@…>, 8 years ago

My work from over summer, with a few things tidied up and updated to include recent commits/patches to bring this up to date. Still very much work in progress.

  • Property mode set to 100644
File size: 59.4 KB
Line 
1/*
2 * This file is part of libtrace
3 *
4 * Copyright (c) 2007,2008,2009,2010 The University of Waikato, Hamilton,
5 * New Zealand.
6 *
7 * All rights reserved.
8 *
9 * This code has been developed by the University of Waikato WAND
10 * research group. For further information please see http://www.wand.net.nz/
11 *
12 * libtrace is free software; you can redistribute it and/or modify
13 * it under the terms of the GNU General Public License as published by
14 * the Free Software Foundation; either version 2 of the License, or
15 * (at your option) any later version.
16 *
17 * libtrace is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20 * GNU General Public License for more details.
21 *
22 * You should have received a copy of the GNU General Public License
23 * along with libtrace; if not, write to the Free Software
24 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
25 *
26 * $Id$
27 *
28 */
29
30
31#define _GNU_SOURCE
32#include "common.h"
33#include "config.h"
34#include <assert.h>
35#include <errno.h>
36#include <fcntl.h>
37#include <stdio.h>
38#include <stdlib.h>
39#include <string.h>
40#include <sys/stat.h>
41#include <sys/types.h>
42#ifndef WIN32
43#include <sys/socket.h>
44#endif
45#include <stdarg.h>
46#include <sys/param.h>
47
48#ifdef HAVE_LIMITS_H
49#  include <limits.h>
50#endif
51
52#ifdef HAVE_SYS_LIMITS_H
53#  include <sys/limits.h>
54#endif
55
56#ifdef HAVE_NET_IF_ARP_H
57#  include <net/if_arp.h>
58#endif
59
60#ifdef HAVE_NET_IF_H
61#  include <net/if.h>
62#endif
63
64#ifdef HAVE_NETINET_IN_H
65#  include <netinet/in.h>
66#endif
67
68#ifdef HAVE_NET_ETHERNET_H
69#  include <net/ethernet.h>
70#endif
71
72#ifdef HAVE_NETINET_IF_ETHER_H
73#  include <netinet/if_ether.h>
74#endif
75
76#include <time.h>
77#ifdef WIN32
78#include <sys/timeb.h>
79#endif
80
81#include "libtrace.h"
82#include "libtrace_int.h"
83
84#ifdef HAVE_PCAP_BPF_H
85#  include <pcap-bpf.h>
86#else
87#  ifdef HAVE_NET_BPF_H
88#    include <net/bpf.h>
89#  endif
90#endif
91
92
93#include "libtrace_int.h"
94#include "format_helper.h"
95#include "rt_protocol.h"
96#include "hash_toeplitz.h"
97
98#include <pthread.h>
99#include <signal.h>
100
101
102struct multithreading_stats {
103        uint64_t full_queue_hits;
104        uint64_t wait_for_fill_complete_hits;
105} contention_stats[1024];
106
107
108/**
109 * @return True if the format supports parallel threads.
110 */
111static inline bool trace_supports_parallel(libtrace_t *trace)
112{
113        assert(trace);
114        assert(trace->format);
115        if (trace->format->pstart_input)
116                return true;
117        else
118                return false;
119        //return trace->format->pstart_input;
120}
121
122DLLEXPORT inline void print_contention_stats(libtrace_t *libtrace) {
123        int i;
124        struct multithreading_stats totals = {0};
125        for (i = 0; i < libtrace->mapper_thread_count ; i++) {
126                printf("\nStats for mapper thread#%d\n", i);
127                printf("\tfull_queue_hits: %"PRIu64"\n", contention_stats[i].full_queue_hits);
128                totals.full_queue_hits += contention_stats[i].full_queue_hits;
129                printf("\twait_for_fill_complete_hits: %"PRIu64"\n", contention_stats[i].wait_for_fill_complete_hits);
130                totals.wait_for_fill_complete_hits += contention_stats[i].wait_for_fill_complete_hits;
131        }
132        printf("\nTotals for mapper threads\n");
133        printf("\tfull_queue_hits: %"PRIu64"\n", totals.full_queue_hits);
134        printf("\twait_for_fill_complete_hits: %"PRIu64"\n", totals.wait_for_fill_complete_hits);
135
136        return;
137}
138
139inline void libtrace_zero_thread(libtrace_thread_t * t) {
140        t->trace = NULL;
141        t->ret = NULL;
142        t->type = THREAD_EMPTY;
143        libtrace_zero_ringbuffer(&t->rbuffer);
144        libtrace_zero_vector(&t->vector);
145        libtrace_zero_deque(&t->deque);
146        t->recorded_first = false;
147        t->map_num = -1;
148}
149
150// Ints are aligned int is atomic so safe to read and write at same time
151// However write must be locked, read doesn't (We never try read before written to table)
152libtrace_thread_t * get_thread_table(libtrace_t *libtrace) {
153        int i = 0;
154        pthread_t tid = pthread_self();
155
156        for (;i<libtrace->mapper_thread_count ;++i) {
157                if (pthread_equal(tid, libtrace->mapper_threads[i].tid))
158                        return &libtrace->mapper_threads[i];
159        }
160        return NULL;
161}
162
163int get_thread_table_num(libtrace_t *libtrace);
164DLLEXPORT int get_thread_table_num(libtrace_t *libtrace) {
165        int i = 0;
166        pthread_t tid = pthread_self();
167        for (;i<libtrace->mapper_thread_count; ++i) {
168                if (pthread_equal(tid, libtrace->mapper_threads[i].tid))
169                        return i;
170        }
171        return -1;
172}
173
174static libtrace_thread_t * get_thread_descriptor(libtrace_t *libtrace) {
175        libtrace_thread_t *ret;
176        if (!(ret = get_thread_table(libtrace))) {
177                pthread_t tid = pthread_self();
178                // Check if we are reducer or something else
179                if (pthread_equal(tid, libtrace->reducer_thread.tid))
180                        ret = &libtrace->reducer_thread;
181                else if (pthread_equal(tid, libtrace->hasher_thread.tid))
182                        ret = &libtrace->hasher_thread;
183                else
184                        ret = NULL;
185        }
186        return ret;
187}
188
189/**
190 * Holds threads in a paused state, until released by broadcasting
191 * the condition mutex.
192 */
193static void trace_thread_pause(libtrace_t *trace) {
194        printf("Pausing thread #%d\n", get_thread_table_num(trace));
195        assert(pthread_mutex_lock(&trace->libtrace_lock) == 0);
196        trace->perpkt_pausing++;
197        pthread_cond_broadcast(&trace->perpkt_cond);
198        while (!trace->started) {
199                assert(pthread_cond_wait(&trace->perpkt_cond, &trace->libtrace_lock) == 0);
200        }
201        trace->perpkt_pausing--;
202        pthread_cond_broadcast(&trace->perpkt_cond);
203        assert(pthread_mutex_unlock(&trace->libtrace_lock) == 0);
204        printf("Releasing thread #%d\n", get_thread_table_num(trace));
205}
206
207void* mapper_start(void *data) {
208        libtrace_t *trace = (libtrace_t *)data;
209        libtrace_thread_t * t;
210        libtrace_message_t message;
211        libtrace_packet_t *packet = NULL;
212
213        assert(pthread_mutex_lock(&trace->libtrace_lock) == 0);
214        t = get_thread_table(trace);
215        assert(t);
216        //printf("Yay Started Mapper thread #%d\n", (int) get_thread_table_num(trace));
217        assert(pthread_mutex_unlock(&trace->libtrace_lock) == 0);
218
219        /* ~~~~~~~~~~~ Setup complete now we loop ~~~~~~~~~~~~~~~ */
220        // Send a message to say we've started
221
222        message.code = MESSAGE_STARTED;
223        message.sender = t;
224        message.additional = NULL;
225
226        // Let the per_packet function know we have started
227        (*trace->per_pkt)(trace, NULL, &message, t);
228
229
230        for (;;) {
231                int psize;
232
233                if (libtrace_message_queue_try_get(&t->messages, &message) != LIBTRACE_MQ_FAILED) {
234                        switch (message.code) {
235                                case MESSAGE_PAUSE:
236                                        trace_thread_pause(trace);
237                                        break;
238                                case MESSAGE_STOP:
239                                        goto stop;
240                        }
241                        (*trace->per_pkt)(trace, NULL, &message, t);
242                        continue;
243                }
244
245                if (trace->mapper_thread_count == 1) {
246                        if (!packet) {
247                                if (!libtrace_ringbuffer_try_sread_bl(&trace->packet_freelist, (void **) &packet))
248                                        packet = trace_create_packet();
249                        }
250                        assert(packet);
251                        if ((psize = trace_read_packet(trace, packet)) <1) {
252                                break;
253                        }
254                } else {
255                        psize = trace_pread_packet(trace, &packet);
256                }
257
258                if (psize > 0) {
259                        packet = (*trace->per_pkt)(trace, packet, NULL, t);
260                        continue;
261                }
262
263                if (psize == -2)
264                        continue; // We have a message
265
266                if (psize < 1) { // consider sending a message
267                        break;
268                }
269
270        }
271
272
273stop:
274        /* ~~~~~~~~~~~~~~ Trace is finished do tear down ~~~~~~~~~~~~~~~~~~~~~ */
275        // Let the per_packet function know we have stopped
276        message.code = MESSAGE_STOPPED;
277        message.sender = message.additional = NULL;
278        (*trace->per_pkt)(trace, NULL, &message, t);
279
280        // And we're at the end free the memories
281        assert(pthread_mutex_lock(&trace->libtrace_lock) == 0);
282        t->state = THREAD_FINISHED;
283        assert(pthread_mutex_unlock(&trace->libtrace_lock) == 0);
284
285        // Notify only after we've defiantly set the state to finished
286        message.code = MESSAGE_MAPPER_ENDED;
287        message.additional = NULL;
288        trace_send_message_to_reducer(trace, &message);
289
290        pthread_exit(NULL);
291};
292
293/** True if trace has dedicated hasher thread otherwise false */
294inline int trace_has_dedicated_hasher(libtrace_t * libtrace);
295inline int trace_has_dedicated_hasher(libtrace_t * libtrace)
296{
297        return libtrace->hasher_thread.type == THREAD_HASHER;
298}
299
300/**
301 * The start point for our single threaded hasher thread, this will read
302 * and hash a packet from a data source and queue it against the correct
303 * core to process it.
304 */
305static void* hasher_start(void *data) {
306        libtrace_t *trace = (libtrace_t *)data;
307        libtrace_thread_t * t;
308        int i;
309        libtrace_packet_t * packet;
310
311        assert(trace_has_dedicated_hasher(trace));
312        /* Wait until all threads are started and objects are initialised (ring buffers) */
313        assert(pthread_mutex_lock(&trace->libtrace_lock) == 0);
314        t = &trace->hasher_thread;
315        assert(t->type == THREAD_HASHER && pthread_equal(pthread_self(), t->tid));
316        printf("Hasher Thread started\n");
317        assert(pthread_mutex_unlock(&trace->libtrace_lock) == 0);
318        int pkt_skipped = 0;
319        /* Read all packets in then hash and queue against the correct thread */
320        while (1) {
321                int thread;
322                if (!pkt_skipped && !libtrace_ringbuffer_try_sread_bl(&trace->packet_freelist, (void **) &packet))
323                        packet = trace_create_packet();
324                assert(packet);
325
326                if (libtrace_halt) // Signal to die has been sent - TODO
327                        break;
328
329                if ((packet->error = trace_read_packet(trace, packet)) <1 /*&& psize != LIBTRACE_MESSAGE_WAITING*/) {
330                        break; /* We are EOF or error'd either way we stop  */
331                }
332
333                /* We are guaranteed to have a hash function i.e. != NULL */
334                trace_packet_set_hash(packet, (*trace->hasher)(packet, trace->hasher_data));
335                thread = trace_packet_get_hash(packet) % trace->mapper_thread_count;
336                /* Blocking write to the correct queue - I'm the only writer */
337                if (trace->mapper_threads[thread].state != THREAD_FINISHED) {
338                        libtrace_ringbuffer_write(&trace->mapper_threads[thread].rbuffer, packet);
339                        pkt_skipped = 0;
340                } else {
341                        pkt_skipped = 1; // Reuse that packet no one read it
342                }
343        }
344
345        /* Broadcast our last failed read to all threads */
346        for (i = 0; i < trace->mapper_thread_count; i++) {
347                libtrace_packet_t * bcast;
348                printf("Broadcasting error/EOF now the trace is over\n");
349                if (i == trace->mapper_thread_count - 1) {
350                        bcast = packet;
351                } else {
352                        bcast = trace_create_packet();
353                        bcast->error = packet->error;
354                }
355                assert(pthread_mutex_lock(&trace->libtrace_lock) == 0);
356                if (trace->mapper_threads[i].state != THREAD_FINISHED) {
357                        assert(pthread_mutex_unlock(&trace->libtrace_lock) == 0);
358                        // Unlock early otherwise we could deadlock
359                        libtrace_ringbuffer_write(&trace->mapper_threads[i].rbuffer, NULL);
360                } else {
361                        assert(pthread_mutex_unlock(&trace->libtrace_lock) == 0);
362                }
363        }
364        // We dont need to free packet
365
366        // And we're at the end free the memories
367        t->state = THREAD_FINISHED;
368
369        // Notify only after we've defiantly set the state to finished
370        libtrace_message_t message;
371        message.code = MESSAGE_MAPPER_ENDED;
372        message.additional = NULL;
373        trace_send_message_to_reducer(trace, &message);
374
375        // TODO remove from TTABLE t sometime
376        pthread_exit(NULL);
377};
378
379/**
380 * Moves src into dest(Complete copy) and copies the memory buffer and
381 * its flags from dest into src ready for reuse without needing extra mallocs.
382 */
383static inline void swap_packets(libtrace_packet_t *dest, libtrace_packet_t *src) {
384        // Save the passed in buffer status
385        assert(dest->trace == NULL); // Must be a empty packet
386        void * temp_buf = dest->buffer;
387        buf_control_t temp_buf_control = dest->buf_control;
388        // Completely copy StoredPacket into packet
389        memcpy(dest, src, sizeof(libtrace_packet_t));
390        // Set the buffer settings on the returned packet
391        src->buffer = temp_buf;
392        src->buf_control = temp_buf_control;
393        src->trace = NULL;
394}
395
396/* Our simplest case when a thread becomes ready it can obtain a exclusive
397 * lock to read a packet from the underlying trace.
398 */
399inline static int trace_pread_packet_first_in_first_served(libtrace_t *libtrace, libtrace_packet_t **packet)
400{
401        // We need this to fill the 'first' packet table
402        libtrace_thread_t *t = get_thread_table(libtrace);
403        if (!*packet) {
404                if (!libtrace_ringbuffer_try_sread_bl(&libtrace->packet_freelist, (void **) packet))
405                        *packet = trace_create_packet();
406        }
407        assert(*packet);
408        assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
409        /* Read a packet */
410        (*packet)->error = trace_read_packet(libtrace, *packet);
411        // Doing this inside the lock ensures the first packet is always
412        // recorded first
413        store_first_packet(libtrace, *packet, t);
414
415        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
416        return (*packet)->error;
417}
418
419/**
420 * For the case that we have a dedicated hasher thread
421 * 1. We read a packet from our buffer
422 * 2. Move that into the packet provided (packet)
423 */
424inline static int trace_pread_packet_hasher_thread(libtrace_t *libtrace, libtrace_packet_t **packet)
425{
426        int this_thread = get_thread_table_num(libtrace); // Could be worth caching ... ?
427        libtrace_thread_t* t = &libtrace->mapper_threads[this_thread];
428
429        if (*packet) // Recycle the old get the new
430                if (!libtrace_ringbuffer_try_swrite_bl(&libtrace->packet_freelist, (void *) *packet))
431                        trace_destroy_packet(*packet);
432        *packet = libtrace_ringbuffer_read(&t->rbuffer);
433
434        if (*packet) {
435                return 1;
436        } else {
437                printf("Got a NULL packet the trace is over\n");
438                return -1; // We are done for some reason
439        }
440}
441
442/**
443 * Tries to read from our queue and returns 1 if a packet was retrieved
444 */
445static inline int try_waiting_queue(libtrace_t *libtrace, libtrace_thread_t * t, libtrace_packet_t **packet, int * ret)
446{
447        libtrace_packet_t* retrived_packet;
448
449        /* Lets see if we have one waiting */
450        if (libtrace_ringbuffer_try_read(&t->rbuffer, (void **) &retrived_packet)) {
451                /* Copy paste from trace_pread_packet_hasher_thread() except that we try read (non-blocking) */
452                assert(retrived_packet);
453
454                if (*packet) // Recycle the old get the new
455                        if (!libtrace_ringbuffer_try_swrite_bl(&libtrace->packet_freelist, (void *) *packet))
456                                trace_destroy_packet(*packet);
457                *packet = retrived_packet;
458                *ret = (*packet)->error;
459                return 1;
460        }
461        return 0;
462}
463
464/**
465 * Allows us to ensure all threads are finished writing to our threads ring_buffer
466 * before returning EOF/error.
467 */
468inline static int trace_handle_finishing_mapper(libtrace_t *libtrace, libtrace_packet_t **packet, libtrace_thread_t * t)
469{
470        /* We are waiting for the condition that another thread ends to check
471         * our queue for new data, once all threads end we can go to finished */
472        bool complete = false;
473        int ret;
474
475        do {
476                // Wait for a thread to end
477                assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
478
479                // Check before
480                if (libtrace->mappers_finishing == libtrace->mapper_thread_count) {
481                        complete = true;
482                        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
483                        continue;
484                }
485
486                assert(pthread_cond_wait(&libtrace->perpkt_cond, &libtrace->libtrace_lock) == 0);
487
488                // Check after
489                if (libtrace->mappers_finishing == libtrace->mapper_thread_count) {
490                        complete = true;
491                        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
492                        continue;
493                }
494
495                assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
496
497                // Always trying to keep our buffer empty for the unlikely case more threads than buffer space want to write into our queue
498                if(try_waiting_queue(libtrace, t, packet, &ret))
499                        return ret;
500        } while (!complete);
501
502        // We can only end up here once all threads complete
503        try_waiting_queue(libtrace, t, packet, &ret);
504
505        return ret;
506        // TODO rethink this logic fix bug here
507}
508
509/**
510 * Expects the libtrace_lock to not be held
511 */
512inline static int trace_finish_mapper(libtrace_t *libtrace, libtrace_packet_t **packet, libtrace_thread_t * t)
513{
514        assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
515        t->state = THREAD_FINISHING;
516        libtrace->mappers_finishing++;
517        pthread_cond_broadcast(&libtrace->perpkt_cond);
518        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
519        return trace_handle_finishing_mapper(libtrace, packet, t);
520}
521
522/**
523 * This case is much like the dedicated hasher, except that we will become
524 * hasher if we don't have a a packet waiting.
525 *
526 * Note: This is only every used if we have are doing hashing.
527 *
528 * TODO: Can block on zero copy formats such as ring: and dpdk: if the
529 * queue sizes in total are larger than the ring size.
530 *
531 * 1. We read a packet from our buffer
532 * 2. Move that into the packet provided (packet)
533 */
534inline static int trace_pread_packet_hash_locked(libtrace_t *libtrace, libtrace_packet_t **packet)
535{
536        int this_thread = get_thread_table_num(libtrace); // Could be worth caching ... ?
537        libtrace_thread_t * t = &libtrace->mapper_threads[this_thread];
538        int thread, ret, psize;
539
540        while (1) {
541                if(try_waiting_queue(libtrace, t, packet, &ret))
542                        return ret;
543                // Can still block here if another thread is writing to a full queue
544                assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
545
546                // Its impossible for our own queue to overfill, because no one can write
547                // when we are in the lock
548                if(try_waiting_queue(libtrace, t, packet, &ret)) {
549                        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
550                        return ret;
551                }
552
553                // Another thread cannot write a packet because a queue has filled up. Is it ours?
554                if (libtrace->mapper_queue_full) {
555                        contention_stats[this_thread].wait_for_fill_complete_hits++;
556                        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
557                        continue;
558                }
559
560                if (!*packet) {
561                        if (!libtrace_ringbuffer_try_sread_bl(&libtrace->packet_freelist, (void **) packet))
562                                *packet = trace_create_packet();
563                }
564                assert(*packet);
565
566                // If we fail here we can guarantee that our queue is empty (and no new data will be added because we hold the lock)
567                if (libtrace_halt || ((*packet)->error = trace_read_packet(libtrace, *packet)) <1 /*&& psize != LIBTRACE_MESSAGE_WAITING*/) {
568                        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
569                        if (libtrace_halt)
570                                return 0;
571                        else
572                                return (*packet)->error;
573                }
574
575                trace_packet_set_hash(*packet, (*libtrace->hasher)(*packet, libtrace->hasher_data));
576                thread = trace_packet_get_hash(*packet) % libtrace->mapper_thread_count;
577                if (thread == this_thread) {
578                        // If it's this thread we must be in order because we checked the buffer once we got the lock
579                        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
580                        return (*packet)->error;
581                }
582
583                if (libtrace->mapper_threads[thread].state != THREAD_FINISHED) {
584                        while (!libtrace_ringbuffer_try_swrite_bl(&libtrace->mapper_threads[thread].rbuffer, *packet)) {
585                                libtrace->mapper_queue_full = true;
586                                assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
587                                contention_stats[this_thread].full_queue_hits++;
588                                assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
589                        }
590                        *packet = NULL;
591                        libtrace->mapper_queue_full = false;
592                } else {
593                        /* We can get here if the user closes the thread before natural completion/or error */
594                        assert (!"packet_hash_locked() The user terminated the trace in a abnormal manner");
595                }
596                assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
597        }
598}
599
600/**
601 * This case is much like the dedicated hasher, except that we will become
602 * hasher if we don't have a a packet waiting.
603 *
604 * TODO: You can loose the tail of a trace if the final thread
605 * fills its own queue and therefore breaks early and doesn't empty the sliding window.
606 *
607 * TODO: Can block on zero copy formats such as ring: and dpdk: if the
608 * queue sizes in total are larger than the ring size.
609 *
610 * 1. We read a packet from our buffer
611 * 2. Move that into the packet provided (packet)
612 */
613inline static int trace_pread_packet_sliding_window(libtrace_t *libtrace, libtrace_packet_t **packet)
614{
615        int this_thread = get_thread_table_num(libtrace); // Could be worth caching ... ?
616        libtrace_thread_t * t = &libtrace->mapper_threads[this_thread];
617        int ret, i, thread, psize;
618
619        if (t->state == THREAD_FINISHING)
620                return trace_handle_finishing_mapper(libtrace, packet, t);
621
622        while (1) {
623                // Check if we have packets ready
624                if(try_waiting_queue(libtrace, t, packet, &ret))
625                        return ret;
626
627                // We limit the number of packets we get to the size of the sliding window
628                // such that it is impossible for any given thread to fail to store a packet
629                assert(sem_wait(&libtrace->sem) == 0);
630                /*~~~~Single threaded read of a packet~~~~*/
631                assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
632
633                /* Re-check our queue things we might have data waiting */
634                if(try_waiting_queue(libtrace, t, packet, &ret)) {
635                        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
636                        assert(sem_post(&libtrace->sem) == 0);
637                        return ret;
638                }
639
640                // TODO put on *proper* condition variable
641                if (libtrace->mapper_queue_full) {
642                        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
643                        assert(sem_post(&libtrace->sem) == 0);
644                        contention_stats[this_thread].wait_for_fill_complete_hits++;
645                        continue;
646                }
647
648                if (!*packet) {
649                        if (!libtrace_ringbuffer_try_sread_bl(&libtrace->packet_freelist, (void **) packet))
650                                *packet = trace_create_packet();
651                }
652                assert(*packet);
653
654                if (libtrace_halt || ((*packet)->error = trace_read_packet(libtrace, *packet)) <1 /*&& psize != LIBTRACE_MESSAGE_WAITING*/) {
655                        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
656                        assert(sem_post(&libtrace->sem) == 0);
657                        // Finish this thread ensuring that any data written later by another thread is retrieved also
658                        if (libtrace_halt)
659                                return 0;
660                        else
661                                return trace_finish_mapper(libtrace, packet, t);
662                }
663                assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
664
665                /* ~~~~Multiple threads can run the hasher~~~~ */
666                trace_packet_set_hash(*packet, (*libtrace->hasher)(*packet, libtrace->hasher_data));
667
668                /* Yes this is correct opposite read lock for a write operation */
669                assert(pthread_rwlock_rdlock(&libtrace->window_lock) == 0);
670                if (!libtrace_slidingwindow_try_write(&libtrace->sliding_window, trace_packet_get_order(*packet), *packet))
671                        assert(!"Semaphore should stop us from ever overfilling the sliding window");
672                assert(pthread_rwlock_unlock(&libtrace->window_lock) == 0);
673                *packet = NULL;
674
675                // Always try read any data from the sliding window
676                while (libtrace_slidingwindow_read_ready(&libtrace->sliding_window)) {
677                        assert(pthread_rwlock_wrlock(&libtrace->window_lock) == 0);
678                        if (libtrace->mapper_queue_full) {
679                                // I might be the holdup in which case if I can read my queue I should do that and return
680                                if(try_waiting_queue(libtrace, t, packet, &ret)) {
681                                        assert(pthread_rwlock_unlock(&libtrace->window_lock) == 0);
682                                        return ret;
683                                }
684                                assert(pthread_rwlock_unlock(&libtrace->window_lock) == 0);
685                                continue;
686                        }
687                        // Read greedily as many as we can
688                        while (libtrace_slidingwindow_try_read(&libtrace->sliding_window, (void **) packet, NULL)) {
689                                thread = trace_packet_get_hash(*packet) % libtrace->mapper_thread_count;
690                                if (libtrace->mapper_threads[thread].state != THREAD_FINISHED) {
691                                        while (!libtrace_ringbuffer_try_swrite_bl(&libtrace->mapper_threads[thread].rbuffer, *packet)) {
692                                                if (this_thread == thread)
693                                                {
694                                                        // TODO think about this case more because we have to stop early if this were to happen on the last read
695                                                        // before EOF/error we might not have emptied the sliding window
696                                                        printf("!~!~!~!~!~!~In this Code~!~!~!~!\n");
697                                                        // Its our queue we must have a packet to read out
698                                                        if(try_waiting_queue(libtrace, t, packet, &ret)) {
699                                                                // We must be able to write this now 100% without fail
700                                                                libtrace_ringbuffer_write(&libtrace->mapper_threads[thread].rbuffer, *packet);
701                                                                assert(sem_post(&libtrace->sem) == 0);
702                                                                assert(pthread_rwlock_unlock(&libtrace->window_lock) == 0);
703                                                                return ret;
704                                                        } else {
705                                                                assert(!"Our queue is full but I cannot read from it??");
706                                                        }
707                                                }
708                                                // Not us we have to give the other threads a chance to write there packets then
709                                                libtrace->mapper_queue_full = true;
710                                                assert(pthread_rwlock_unlock(&libtrace->window_lock) == 0);
711                                                for (i = 0; i < libtrace->mapper_thread_count-1; i++) // Release all other threads to read there packets
712                                                        assert(sem_post(&libtrace->sem) == 0);
713
714                                                contention_stats[this_thread].full_queue_hits++;
715                                                assert(pthread_rwlock_wrlock(&libtrace->window_lock) == 0);
716                                                // Grab these back
717                                                for (i = 0; i < libtrace->mapper_thread_count-1; i++) // Release all other threads to read there packets
718                                                        assert(sem_wait(&libtrace->sem) == 0);
719                                                libtrace->mapper_queue_full = false;
720                                        }
721                                        assert(sem_post(&libtrace->sem) == 0);
722                                        *packet = NULL;
723                                } else {
724                                        // Cannot write to a queue if no ones waiting (I think this is unreachable)
725                                        // in the general case (unless the user ends early without proper clean up).
726                                        assert (!"unreachable code??");
727                                }
728                        }
729                        assert(pthread_rwlock_unlock(&libtrace->window_lock) == 0);
730                }
731                // Now we go back to checking our queue anyways
732        }
733}
734
735
736/**
737 * For the first packet of each queue we keep a copy and note the system
738 * time it was received at.
739 *
740 * This is used for finding the first packet when playing back a trace
741 * in trace time. And can be used by real time applications to print
742 * results out every XXX seconds.
743 */
744inline void store_first_packet(libtrace_t *libtrace, libtrace_packet_t *packet, libtrace_thread_t *t)
745{
746        if (!t->recorded_first) {
747                struct timeval tv;
748                libtrace_packet_t * dup;
749                // For what it's worth we can call these outside of the lock
750                gettimeofday(&tv, NULL);
751                dup = trace_copy_packet(packet);
752                assert(pthread_spin_lock(&libtrace->first_packets.lock) == 0);
753                libtrace->first_packets.packets[t->map_num].packet = dup;
754                //printf("Stored first packet time=%f\n", trace_get_seconds(dup));
755                memcpy(&libtrace->first_packets.packets[t->map_num].tv, &tv, sizeof(tv));
756                // Now update the first
757                libtrace->first_packets.count++;
758                if (libtrace->first_packets.count == 1) {
759                        // We the first entry hence also the first known packet
760                        libtrace->first_packets.first = t->map_num;
761                } else {
762                        // Check if we are newer than the previous 'first' packet
763                        size_t first = libtrace->first_packets.first;
764                        if (trace_get_seconds(dup) <
765                                trace_get_seconds(libtrace->first_packets.packets[first].packet))
766                                libtrace->first_packets.first = t->map_num;
767                }
768                assert(pthread_spin_unlock(&libtrace->first_packets.lock) == 0);
769                libtrace_message_t mesg;
770                mesg.code = MESSAGE_FIRST_PACKET;
771                mesg.additional = NULL;
772                trace_send_message_to_reducer(libtrace, &mesg);
773                t->recorded_first = true;
774        }
775}
776
777/**
778 * Returns 1 if its certain that the first packet is truly the first packet
779 * rather than a best guess based upon threads that have published so far.
780 * Otherwise 0 is returned.
781 * It's recommended that this result is stored rather than calling this
782 * function again.
783 */
784DLLEXPORT int retrive_first_packet(libtrace_t *libtrace, libtrace_packet_t **packet, struct timeval **tv)
785{
786        int ret = 0;
787        assert(pthread_spin_lock(&libtrace->first_packets.lock) == 0);
788        if (libtrace->first_packets.count) {
789                *packet = libtrace->first_packets.packets[libtrace->first_packets.first].packet;
790                *tv = &libtrace->first_packets.packets[libtrace->first_packets.first].tv;
791                if (libtrace->first_packets.count == libtrace->mapper_thread_count) {
792                        ret = 1;
793                } else {
794                        struct timeval curr_tv;
795                        // If a second has passed since the first entry we will assume this is the very first packet
796                        gettimeofday(&curr_tv, NULL);
797                        if (curr_tv.tv_sec > (*tv)->tv_sec) {
798                                if(curr_tv.tv_usec > (*tv)->tv_usec || curr_tv.tv_sec - (*tv)->tv_sec > 1) {
799                                        ret = 1;
800                                }
801                        }
802                }
803        } else {
804                *packet = NULL;
805                *tv = NULL;
806        }
807        assert(pthread_spin_unlock(&libtrace->first_packets.lock) == 0);
808        return ret;
809}
810
811
812DLLEXPORT inline uint64_t tv_to_usec(struct timeval *tv)
813{
814        return (uint64_t) tv->tv_sec*1000000ull + (uint64_t) tv->tv_usec;
815}
816
817inline static struct timeval usec_to_tv(uint64_t usec)
818{
819        struct timeval tv;
820        tv.tv_sec = usec / 1000000;
821        tv.tv_usec = usec % 1000000;
822        return tv;
823}
824
825
826/**
827 * Delays a packets playback so the playback will be in trace time
828 */
829static inline void delay_tracetime(libtrace_t *libtrace, libtrace_packet_t *packet, libtrace_thread_t *t) {
830        struct timeval curr_tv, pkt_tv;
831        uint64_t next_release = t->tracetime_offset_usec; // Time at which to release the packet
832        uint64_t curr_usec;
833        /* Tracetime we might delay releasing this packet */
834        if (!t->tracetime_offset_usec) {
835                libtrace_packet_t * first_pkt;
836                struct timeval *sys_tv;
837                int64_t initial_offset;
838                int stable = retrive_first_packet(libtrace, &first_pkt, &sys_tv);
839                assert(first_pkt);
840                pkt_tv = trace_get_timeval(first_pkt);
841                initial_offset = (int64_t)tv_to_usec(sys_tv) - (int64_t)tv_to_usec(&pkt_tv);
842                if (stable)
843                        // 0->1 because 0 is used to mean unset
844                        t->tracetime_offset_usec = initial_offset ? initial_offset: 1;
845                next_release = initial_offset;
846        }
847        /* next_release == offset */
848        pkt_tv = trace_get_timeval(packet);
849        next_release += tv_to_usec(&pkt_tv);
850        gettimeofday(&curr_tv, NULL);
851        curr_usec = tv_to_usec(&curr_tv);
852        if (next_release > curr_usec) {
853                // We need to wait
854                struct timeval delay_tv = usec_to_tv(next_release-curr_usec);
855                //printf("WAITING for %d.%d next=%"PRIu64" curr=%"PRIu64" seconds packettime %f\n", delay_tv.tv_sec, delay_tv.tv_usec, next_release, curr_usec, trace_get_seconds(packet));
856                select(0, NULL, NULL, NULL, &delay_tv);
857        }
858}
859
860/* Read one packet from the trace into a buffer. Note that this function will
861 * block until a packet is read (or EOF is reached).
862 *
863 * @param libtrace      the libtrace opaque pointer
864 * @param packet        the packet opaque pointer
865 * @returns 0 on EOF, negative value on error
866 *
867 * Note this is identical to read_packet but calls pread_packet instead of
868 * read packet in the format.
869 *
870 */
871static inline int trace_pread_packet_wrapper(libtrace_t *libtrace, libtrace_packet_t *packet) {
872
873        assert(libtrace && "You called trace_read_packet() with a NULL libtrace parameter!\n");
874        if (trace_is_err(libtrace))
875                return -1;
876        if (!libtrace->started) {
877                trace_set_err(libtrace,TRACE_ERR_BAD_STATE,"You must call libtrace_start() before trace_read_packet()\n");
878                return -1;
879        }
880        if (!(packet->buf_control==TRACE_CTRL_PACKET || packet->buf_control==TRACE_CTRL_EXTERNAL)) {
881                trace_set_err(libtrace,TRACE_ERR_BAD_STATE,"Packet passed to trace_read_packet() is invalid\n");
882                return -1;
883        }
884        assert(packet);
885
886        if (libtrace->format->read_packet) {
887                do {
888                        size_t ret;
889                        /* Finalise the packet, freeing any resources the format module
890                         * may have allocated it and zeroing all data associated with it.
891                         */
892                        trace_fin_packet(packet);
893                        /* Store the trace we are reading from into the packet opaque
894                         * structure */
895                        packet->trace = libtrace;
896                        ret=libtrace->format->pread_packet(libtrace,packet);
897                        if (ret==(size_t)-1 || ret==(size_t)-2 || ret==0) {
898                                return ret;
899                        }
900                        if (libtrace->filter) {
901                                /* If the filter doesn't match, read another
902                                 * packet
903                                 */
904                                if (!trace_apply_filter(libtrace->filter,packet)){
905                                        ++libtrace->filtered_packets;
906                                        continue;
907                                }
908                        }
909                        if (libtrace->snaplen>0) {
910                                /* Snap the packet */
911                                trace_set_capture_length(packet,
912                                                libtrace->snaplen);
913                        }
914                        trace_packet_set_order(packet, libtrace->accepted_packets);
915                        ++libtrace->accepted_packets;
916                        return ret;
917                } while(1);
918        }
919        trace_set_err(libtrace,TRACE_ERR_UNSUPPORTED,"This format does not support reading packets\n");
920        return ~0U;
921}
922
923/**
924 * Read a packet from the parallel trace
925 */
926DLLEXPORT int trace_pread_packet(libtrace_t *libtrace, libtrace_packet_t **packet)
927{
928        int ret;
929        libtrace_thread_t *t = get_thread_table(libtrace);
930
931        // Cleanup the packet passed back
932        if (*packet)
933                trace_fin_packet(*packet);
934
935        if (libtrace->format->pread_packet) {
936                if (!*packet)
937                        *packet = trace_create_packet();
938                ret = trace_pread_packet_wrapper(libtrace, *packet);
939        } else  if (!libtrace->hasher) {
940                /* We don't care about which core a packet goes to */
941                ret =  trace_pread_packet_first_in_first_served(libtrace, packet);
942        } else if (trace_has_dedicated_hasher(libtrace)) {
943                ret = trace_pread_packet_hasher_thread(libtrace, packet);
944        } else if (libtrace->reducer_flags & MAPPER_USE_SLIDING_WINDOW) {
945                ret = trace_pread_packet_sliding_window(libtrace, packet);
946        } else {
947                ret = trace_pread_packet_hash_locked(libtrace, packet);
948        }
949
950        // Formats can also optionally do this internally to ensure the first
951        // packet is always reported correctly
952        if (ret > 0) {
953                store_first_packet(libtrace, *packet, t);
954                if (libtrace->tracetime)
955                        delay_tracetime(libtrace, *packet, t);
956        }
957
958        return ret;
959}
960
961/* Starts perpkt threads
962 * @return threads_started
963 */
964static inline int trace_start_perpkt_threads (libtrace_t *libtrace) {
965        int i;
966
967        for (i = 0; i < libtrace->mapper_thread_count; i++) {
968                libtrace_thread_t *t = &libtrace->mapper_threads[i];
969                assert(pthread_create(&t->tid, NULL, mapper_start, (void *) libtrace) == 0);
970        }
971        return libtrace->mapper_thread_count;
972}
973
974/* Start an input trace in a parallel fashion.
975 *
976 * @param libtrace      the input trace to start
977 * @param global_blob some global data you can share with the new thread
978 * @returns 0 on success
979 */
980DLLEXPORT int trace_pstart(libtrace_t *libtrace, void* global_blob, fn_per_pkt per_pkt, fn_reducer reducer)
981{
982        int i;
983        sigset_t sig_before, sig_block_all;
984
985        assert(libtrace);
986        if (trace_is_err(libtrace))
987                return -1;;
988        if (libtrace->perpkt_pausing != 0) {
989                printf("Restarting trace\n");
990                libtrace->format->pstart_input(libtrace);
991                // TODO empty any queues out here //
992                assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
993                libtrace->started = true;
994                assert(pthread_cond_broadcast(&libtrace->perpkt_cond) == 0);
995                assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
996                return 0;
997        }
998
999        // Store the user defined things against the trace
1000        libtrace->global_blob = global_blob;
1001        libtrace->per_pkt = per_pkt;
1002        libtrace->reducer = reducer;
1003        libtrace->mappers_finishing = 0;
1004        // libtrace->hasher = &rand_hash; /* Hasher now set via option */
1005
1006        assert(pthread_mutex_init(&libtrace->libtrace_lock, NULL) == 0);
1007        assert(pthread_cond_init(&libtrace->perpkt_cond, NULL) == 0);
1008        assert(pthread_rwlock_init(&libtrace->window_lock, NULL) == 0);
1009        assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
1010
1011        // Set default buffer sizes
1012        if (libtrace->mapper_buffer_size <= 0)
1013                libtrace->mapper_buffer_size = 1000;
1014
1015        if (libtrace->mapper_thread_count <= 0)
1016                libtrace->mapper_thread_count = 2; // XXX scale to system
1017
1018        if(libtrace->packet_freelist_size <= 0)
1019                libtrace->packet_freelist_size = (libtrace->mapper_buffer_size + 1) * libtrace->mapper_thread_count;
1020
1021        if(libtrace->packet_freelist_size <
1022                (libtrace->mapper_buffer_size + 1) * libtrace->mapper_thread_count)
1023                fprintf(stderr, "WARNING deadlocks may occur and extra memory allocating buffer sizes (packet_freelist_size) mismatched\n");
1024
1025        libtrace->started=true; // Before we start the threads otherwise we could have issues
1026        /* Disable signals - Pthread signal handling */
1027
1028        sigemptyset(&sig_block_all);
1029
1030        assert(pthread_sigmask(SIG_SETMASK, &sig_block_all, &sig_before) == 0);
1031
1032        // If we are using a hasher start it
1033        if (libtrace->hasher && libtrace->hasher_thread.type == THREAD_HASHER) {
1034                libtrace_thread_t *t = &libtrace->hasher_thread;
1035                t->trace = libtrace;
1036                t->ret = NULL;
1037                t->type = THREAD_HASHER;
1038                t->state = THREAD_RUNNING;
1039                assert(pthread_create(&t->tid, NULL, hasher_start, (void *) libtrace) == 0);
1040        } else {
1041                libtrace->hasher_thread.type = THREAD_EMPTY;
1042        }
1043        libtrace_ringbuffer_init(&libtrace->packet_freelist, libtrace->packet_freelist_size, LIBTRACE_RINGBUFFER_POLLING);
1044        libtrace_slidingwindow_init(&libtrace->sliding_window, libtrace->packet_freelist_size, 0);
1045        assert(sem_init(&libtrace->sem, 0, libtrace->packet_freelist_size) == 0);
1046        // This will be applied to every new thread that starts, i.e. they will block all signals
1047        // Lets start a fixed number of reading threads
1048
1049        // For now we never have a dedicated thread for the reducer
1050        // i.e. This main thread is used as the reducer
1051        libtrace->reducer_thread.tid = pthread_self();
1052        libtrace->reducer_thread.type = THREAD_REDUCER;
1053        libtrace->reducer_thread.state = THREAD_RUNNING;
1054        libtrace_message_queue_init(&libtrace->reducer_thread.messages, sizeof(libtrace_message_t));
1055
1056        /* Ready some storages */
1057        libtrace->first_packets.first = 0;
1058        libtrace->first_packets.count = 0;
1059        assert(pthread_spin_init(&libtrace->first_packets.lock, 0) == 0);
1060        libtrace->first_packets.packets = calloc(libtrace->mapper_thread_count, sizeof(struct  __packet_storage_magic_type));
1061
1062
1063        /* Start all of our mapper threads */
1064        libtrace->mapper_threads = calloc(sizeof(libtrace_thread_t), libtrace->mapper_thread_count);
1065        for (i = 0; i < libtrace->mapper_thread_count; i++) {
1066                libtrace_thread_t *t = &libtrace->mapper_threads[i];
1067                t->trace = libtrace;
1068                t->ret = NULL;
1069                t->type = THREAD_MAPPER;
1070                t->state = THREAD_RUNNING;
1071                t->user_data = NULL;
1072                // t->tid DONE on create
1073                t->map_num = i;
1074                if (libtrace->hasher)
1075                        libtrace_ringbuffer_init(&t->rbuffer, libtrace->mapper_buffer_size, LIBTRACE_RINGBUFFER_POLLING);
1076                // Depending on the mode vector or deque might be chosen
1077                libtrace_vector_init(&t->vector, sizeof(libtrace_result_t));
1078                libtrace_deque_init(&t->deque, sizeof(libtrace_result_t));
1079                libtrace_message_queue_init(&t->messages, sizeof(libtrace_message_t));
1080                t->tmp_key = 0;
1081                t->tmp_data = NULL;
1082                t->recorded_first = false;
1083                assert(pthread_spin_init(&t->tmp_spinlock, 0) == 0);
1084                t->tracetime_offset_usec = 0;;
1085        }
1086
1087        int threads_started = 0;
1088        /* Setup the trace and start our threads */
1089        if (libtrace->mapper_thread_count > 1 && libtrace->format->pstart_input) {
1090                printf("This format has direct support for p's\n");
1091                threads_started = libtrace->format->pstart_input(libtrace);
1092        } else {
1093                if (libtrace->format->start_input) {
1094                        threads_started=libtrace->format->start_input(libtrace);
1095                }
1096        }
1097        if (threads_started == 0)
1098                threads_started = trace_start_perpkt_threads(libtrace);
1099
1100
1101        // Revert back - Allow signals again
1102        assert(pthread_sigmask(SIG_SETMASK, &sig_before, NULL) == 0);
1103        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
1104
1105        if (threads_started < 0)
1106                // Error
1107                return threads_started;
1108
1109        // TODO fix these leaks etc
1110        if (libtrace->mapper_thread_count != threads_started)
1111                printf("Warning started threads not equal requested s=%d r=%d", threads_started, libtrace->mapper_thread_count);
1112
1113
1114        return 0;
1115}
1116
1117/**
1118 * Pauses a trace, this should only be called by the main thread
1119 * 1. Set started = false
1120 * 2. All perpkt threads are paused waiting on a condition var
1121 * 3. Then call ppause on the underlying format if found
1122 * 4. Return with perpkt_pausing set to mapper_count (Used when restarting so we reuse the threads)
1123 *
1124 * Once done you should be a able to modify the trace setup and call pstart again
1125 * TODO handle changing thread numbers
1126 */
1127DLLEXPORT int trace_ppause(libtrace_t *libtrace)
1128{
1129        libtrace_thread_t *t;
1130        int i;
1131        assert(libtrace);
1132        if (!libtrace->started) {
1133                trace_set_err(libtrace,TRACE_ERR_BAD_STATE, "You must call trace_start() before calling trace_ppause()");
1134                return -1;
1135        }
1136
1137        t = get_thread_table(libtrace);
1138
1139        // Set paused
1140        assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
1141        libtrace->started = false;
1142        pthread_cond_broadcast(&libtrace->perpkt_cond);
1143        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
1144
1145        printf("Sending messages \n");
1146        // Stop threads, skip this one if its a mapper
1147        for (i = 0; i < libtrace->mapper_thread_count; i++) {
1148                if (&libtrace->mapper_threads[i] != t) {
1149                        libtrace_message_t message;
1150                        message.code = MESSAGE_PAUSE;
1151                        message.additional = NULL;
1152                        trace_send_message_to_thread(libtrace, &libtrace->mapper_threads[i], &message);
1153                }
1154        }
1155
1156        // Formats must support native message handling if a message is ready
1157        // Approach per Perry's suggestion is a non-blocking read
1158        // followed by a blocking read. XXX STRIP THIS OUT
1159
1160        if (t) {
1161                // A mapper is doing the pausing interesting fake a extra thread paused
1162                // We rely on the user to not return before starting the trace again
1163                assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
1164                libtrace->perpkt_pausing++;
1165                assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
1166        }
1167
1168        printf("Threads are pausing\n");
1169
1170        // Do a early pause to kick threads out - XXX testing for int
1171        if (libtrace->format->pause_input)
1172                        libtrace->format->pause_input(libtrace);
1173
1174        // Wait for all threads to pause
1175        assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
1176        while (libtrace->mapper_thread_count != libtrace->perpkt_pausing) {
1177                assert(pthread_cond_wait(&libtrace->perpkt_cond, &libtrace->libtrace_lock) == 0);
1178        }
1179        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
1180
1181        printf("Threads have paused\n");
1182
1183        if (trace_supports_parallel(libtrace)) {
1184                if (libtrace->format->ppause_input)
1185                        libtrace->format->ppause_input(libtrace);
1186                // TODO What happens if we don't have pause input??
1187        } else {
1188                printf("Trace is not parallel so we are doing a normal pause %s\n", libtrace->uridata);
1189                // This doesn't really work because this could be called by any thread
1190                // Maybe we should grab the lock here??
1191                if (libtrace->format->pause_input)
1192                        libtrace->format->pause_input(libtrace);
1193                // TODO What happens if we don't have pause input??
1194        }
1195
1196        return 0;
1197}
1198
1199/**
1200 * Stop trace finish prematurely as though it meet an EOF
1201 * This should only be called by the main thread
1202 * 1. Calls ppause
1203 * 2. Sends a message asking for threads to finish
1204 *
1205 */
1206DLLEXPORT int trace_pstop(libtrace_t *libtrace)
1207{
1208        int i;
1209        if (!libtrace->started) {
1210                trace_set_err(libtrace,TRACE_ERR_BAD_STATE, "You must call trace_start() before calling trace_pstop()");
1211                return -1;
1212        }
1213
1214        // Ensure all threads have paused and the underlying trace format has
1215        // been closed
1216        trace_ppause(libtrace);
1217
1218        // Now send a message asking the threads to stop
1219        // This will be retrieved before trying to read another packet
1220        for (i = 0; i < libtrace->mapper_thread_count; i++) {
1221                libtrace_message_t message;
1222                message.code = MESSAGE_STOP;
1223                message.additional = NULL;
1224                trace_send_message_to_thread(libtrace, &libtrace->mapper_threads[i], &message);
1225        }
1226
1227        // Now release the threads and let them stop
1228        assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
1229        libtrace->started = true;
1230        assert(pthread_cond_broadcast(&libtrace->perpkt_cond) == 0);
1231        assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
1232        return 0;
1233}
1234
1235/**
1236 * Set the hasher type along with a selected function, if hardware supports
1237 * that generic type of hashing it will be used otherwise the supplied
1238 * hasher function will be used and passed data when called.
1239 *
1240 * @return 0 if successful otherwise -1 on error
1241 */
1242DLLEXPORT int trace_set_hasher(libtrace_t *trace, enum hasher_types type, fn_hasher hasher, void *data) {
1243        int ret = -1;
1244        if (type == HASHER_HARDWARE || (type == HASHER_CUSTOM && !hasher) || (type == HASHER_BALANCE && hasher)) {
1245                return -1;
1246        }
1247
1248        // Save the requirements
1249        trace->hasher_type = type;
1250        if (hasher) {
1251                trace->hasher = hasher;
1252                trace->hasher_data = hasher;
1253        } else {
1254                trace->hasher = NULL;
1255                // TODO consider how to handle freeing this
1256                trace->hasher_data = NULL;
1257        }
1258
1259        // Try push this to hardware - NOTE hardware could do custom if
1260        // there is a more efficient way to apply it, in this case
1261        // it will simply grab the function out of libtrace_t
1262        if (trace->format->pconfig_input)
1263                ret = trace->format->pconfig_input(trace, TRACE_OPTION_SET_HASHER, &type);
1264
1265        if (ret == -1) {
1266                // We have to deal with this ourself
1267                // This most likely means single threaded reading of the trace
1268                if (!hasher) {
1269                        switch (type)
1270                        {
1271                                case HASHER_CUSTOM:
1272                                case HASHER_BALANCE:
1273                                        return 0;
1274                                case HASHER_BIDIRECTIONAL:
1275                                        trace->hasher = toeplitz_hash_packet;
1276                                        trace->hasher_data = calloc(1, sizeof(toeplitz_conf_t));
1277                                        toeplitz_init_config(trace->hasher_data, 1);
1278                                        return 0;
1279                                case HASHER_UNIDIRECTIONAL:
1280                                        trace->hasher = toeplitz_hash_packet;
1281                                        trace->hasher_data = calloc(1, sizeof(toeplitz_conf_t));
1282                                        toeplitz_init_config(trace->hasher_data, 1);
1283                                        return 0;
1284                                case HASHER_HARDWARE:
1285                                        return -1;
1286                        }
1287                        return -1;
1288                }
1289        } else {
1290                // The hardware is dealing with this yay
1291                trace->hasher_type = HASHER_HARDWARE;
1292        }
1293
1294        return 0;
1295}
1296
1297// Waits for all threads to finish
1298DLLEXPORT void trace_join(libtrace_t *libtrace) {
1299        int i;
1300
1301        /* Firstly wait for the mapper threads to finish, since these are
1302         * user controlled */
1303        for (i=0; i< libtrace->mapper_thread_count; i++) {
1304                //printf("Waiting to join with mapper #%d\n", i);
1305                assert(pthread_join(libtrace->mapper_threads[i].tid, NULL) == 0);
1306                //printf("Joined with mapper #%d\n", i);
1307                // So we must do our best effort to empty the queue - so
1308                // the producer (or any other threads) don't block.
1309                libtrace_packet_t * packet;
1310                // Mark that we are no longer accepting packets
1311                assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
1312                libtrace->mapper_threads[i].state = THREAD_FINISHED; // Important we are finished before we empty the buffer
1313                assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
1314                while(libtrace_ringbuffer_try_read(&libtrace->mapper_threads[i].rbuffer, (void **) &packet))
1315                        if (packet) // This could be NULL iff the mapper finishes early
1316                                trace_destroy_packet(packet);
1317        }
1318
1319        /* Now the hasher */
1320        // XXX signal it to stop
1321        if (trace_has_dedicated_hasher(libtrace)) {
1322                printf("Waiting to join with the hasher\n");
1323                pthread_join(libtrace->hasher_thread.tid, NULL);
1324                printf("Joined with with the hasher\n");
1325                libtrace->hasher_thread.state = THREAD_FINISHED;
1326        }
1327
1328        // Now that everything is finished nothing can be touching our
1329        // buffers so clean them up
1330        for (i = 0; i < libtrace->mapper_thread_count; i++) {
1331                // Its possible 1 packet got added by the reducer (or 1 per any other thread) since we cleaned up
1332                // if they lost timeslice before-during a write
1333                libtrace_packet_t * packet;
1334                while(libtrace_ringbuffer_try_read(&libtrace->mapper_threads[i].rbuffer, (void **) &packet))
1335                        trace_destroy_packet(packet);
1336                if (libtrace->hasher) {
1337                        assert(libtrace_ringbuffer_is_empty(&libtrace->mapper_threads[i].rbuffer));
1338                        libtrace_ringbuffer_destroy(&libtrace->mapper_threads[i].rbuffer);
1339                }
1340                // Cannot destroy vector yet, this happens with trace_destroy
1341        }
1342
1343        // Lets mark this as done for now
1344        libtrace->joined = true;
1345}
1346
1347// Don't use extra overhead = :( directly place in storage structure using
1348// post
1349DLLEXPORT libtrace_result_t *trace_create_result()
1350{
1351        libtrace_result_t *result = malloc(sizeof(libtrace_result_t));
1352        assert(result);
1353        result->key = 0;
1354        result->value = NULL;
1355        // TODO automatically back with a free list!!
1356        return result;
1357}
1358
1359DLLEXPORT int libtrace_thread_get_message_count(libtrace_t * libtrace)
1360{
1361        libtrace_thread_t * t = get_thread_descriptor(libtrace);
1362        assert(t);
1363        return libtrace_message_queue_count(&t->messages);
1364}
1365
1366DLLEXPORT int libtrace_thread_get_message(libtrace_t * libtrace, libtrace_message_t * message)
1367{
1368        libtrace_thread_t * t = get_thread_descriptor(libtrace);
1369        assert(t);
1370        return libtrace_message_queue_get(&t->messages, message);
1371}
1372
1373DLLEXPORT int libtrace_thread_try_get_message(libtrace_t * libtrace, libtrace_message_t * message)
1374{
1375        libtrace_thread_t * t = get_thread_descriptor(libtrace);
1376        assert(t);
1377        return libtrace_message_queue_try_get(&t->messages, message);
1378}
1379
1380/**
1381 * Return backlog indicator
1382 */
1383DLLEXPORT int trace_post_reduce(libtrace_t *libtrace)
1384{
1385        libtrace_message_t message = {0};
1386        message.code = MESSAGE_POST_REDUCE;
1387        message.additional = NULL;
1388        message.sender = get_thread_descriptor(libtrace);
1389        return libtrace_message_queue_put(&libtrace->reducer_thread.messages, (void *) &message);
1390}
1391
1392/**
1393 * Return backlog indicator
1394 */
1395DLLEXPORT int trace_send_message_to_reducer(libtrace_t * libtrace, libtrace_message_t * message)
1396{
1397        //printf("Sending message code=%d to reducer\n", message->code);
1398        message->sender = get_thread_descriptor(libtrace);
1399        return libtrace_message_queue_put(&libtrace->reducer_thread.messages, message);
1400}
1401
1402/**
1403 *
1404 */
1405DLLEXPORT int trace_send_message_to_thread(libtrace_t * libtrace, libtrace_thread_t *t, libtrace_message_t * message)
1406{
1407        //printf("Sending message code=%d to reducer\n", message->code);
1408        message->sender = get_thread_descriptor(libtrace);
1409        return libtrace_message_queue_put(&t->messages, message);
1410}
1411
1412DLLEXPORT inline void libtrace_result_set_key(libtrace_result_t * result, uint64_t key) {
1413        result->key = key;
1414}
1415DLLEXPORT inline uint64_t libtrace_result_get_key(libtrace_result_t * result) {
1416        return result->key;
1417}
1418DLLEXPORT inline void libtrace_result_set_value(libtrace_result_t * result, void * value) {
1419        result->value = value;
1420}
1421DLLEXPORT inline void* libtrace_result_get_value(libtrace_result_t * result) {
1422        return result->value;
1423}
1424DLLEXPORT inline void libtrace_result_set_key_value(libtrace_result_t * result, uint64_t key, void * value) {
1425        result->key = key;
1426        result->value = value;
1427}
1428DLLEXPORT void trace_destroy_result(libtrace_result_t ** result) {
1429        free(*result);
1430        result = NULL;
1431        // TODO automatically back with a free list!!
1432}
1433
1434DLLEXPORT void * trace_get_global(libtrace_t *trace)
1435{
1436        return trace->global_blob;
1437}
1438
1439DLLEXPORT void * trace_set_global(libtrace_t *trace, void * data)
1440{
1441        if (trace->global_blob && trace->global_blob != data) {
1442                void * ret = trace->global_blob;
1443                trace->global_blob = data;
1444                return ret;
1445        } else {
1446                trace->global_blob = data;
1447                return NULL;
1448        }
1449}
1450
1451DLLEXPORT void * trace_get_tls(libtrace_thread_t *t)
1452{
1453        return t->user_data;
1454}
1455
1456DLLEXPORT void * trace_set_tls(libtrace_thread_t *t, void * data)
1457{
1458        if(t->user_data && t->user_data != data) {
1459                void *ret = t->user_data;
1460                t->user_data = data;
1461                return ret;
1462        } else {
1463                t->user_data = data;
1464                return NULL;
1465        }
1466}
1467
1468/**
1469 * Note: This function grabs a lock and expects trace_update_inprogress_result
1470 * to be called to release the lock.
1471 *
1472 * Expected to be used in trace-time situations to allow a result to be pending
1473 * a publish that can be taken by the reducer before publish if it wants to
1474 * publish a result. Such as publish a result every second but a queue hasn't
1475 * processed a packet (or is overloaded) and hasn't published yet.
1476 *
1477 * Currently this only supports a single temporary result,
1478 * as such if a key is different to the current temporary result the existing
1479 * one will be published and NULL returned.
1480 */
1481DLLEXPORT void * trace_retrive_inprogress_result(libtrace_t *libtrace, uint64_t key)
1482{
1483        int this_thread = get_thread_table_num(libtrace); // Could be worth caching ... ?
1484        libtrace_thread_t * t = &libtrace->mapper_threads[this_thread];
1485
1486        assert (pthread_spin_lock(&t->tmp_spinlock) == 0);
1487        if (t->tmp_key != key) {
1488                if (t->tmp_data) {
1489                        //printf("publising data key=%"PRIu64"\n", t->tmp_key);
1490                        trace_publish_result(libtrace, t->tmp_key, t->tmp_data);
1491                }
1492                t->tmp_data = NULL;
1493                t->tmp_key = key;
1494        }
1495        return t->tmp_data;
1496}
1497
1498/**
1499 * Updates a temporary result and releases the lock previously grabbed by trace_retrive_inprogress_result
1500 */
1501DLLEXPORT void trace_update_inprogress_result(libtrace_t *libtrace, uint64_t key, void * value)
1502{
1503        int this_thread = get_thread_table_num(libtrace); // Could be worth caching ... ?
1504        libtrace_thread_t * t = &libtrace->mapper_threads[this_thread];
1505        if (t->tmp_key != key) {
1506                if (t->tmp_data) {
1507                        printf("BAD publihsing data key=%"PRIu64"\n", t->tmp_key);
1508                        trace_publish_result(libtrace, t->tmp_key, t->tmp_data);
1509                }
1510                t->tmp_key = key;
1511        }
1512        t->tmp_data = value;
1513        assert (pthread_spin_unlock(&t->tmp_spinlock) == 0);
1514}
1515
1516/**
1517 * Publish to the reduce queue, return
1518 */
1519DLLEXPORT void trace_publish_result(libtrace_t *libtrace, uint64_t key, void * value) {
1520        libtrace_result_t res;
1521        // Who am I???
1522        int this_thread = get_thread_table_num(libtrace); // Could be worth caching ... ?
1523        libtrace_thread_t * t = &libtrace->mapper_threads[this_thread];
1524        // Now put it into my table
1525        static __thread int count = 0;
1526
1527
1528        libtrace_result_set_key_value(&res, key, value);
1529        /*
1530        if (count == 1)
1531                printf("My vector size is %d\n", libtrace_vector_get_size(&t->vector));
1532        count = (count+1) %1000;
1533        libtrace_vector_push_back(&t->vector, &res); // Automatically locking for us :)
1534        */
1535        /*if (count == 1)
1536                printf("My vector size is %d\n", libtrace_deque_get_size(&t->deque));
1537        count = (count+1)%1000;*/
1538        if (libtrace->reducer_flags & (REDUCE_SEQUENTIAL | REDUCE_ORDERED)) {
1539                if (libtrace_deque_get_size(&t->deque) >= 800) {
1540                        trace_post_reduce(libtrace);
1541                }
1542                //while (libtrace_deque_get_size(&t->deque) >= 1000)
1543                //      sched_yield();
1544                libtrace_deque_push_back(&t->deque, &res); // Automatically locking for us :)
1545        } else {
1546                //while (libtrace_vector_get_size(&t->vector) >= 1000)
1547                //      sched_yield();
1548
1549                if (libtrace_deque_get_size(&t->deque) >= 800) {
1550                        trace_post_reduce(libtrace);
1551                }
1552                libtrace_vector_push_back(&t->vector, &res); // Automatically locking for us :)
1553        }
1554}
1555
1556
1557static int compareres(const void* p1, const void* p2)
1558{
1559        if (libtrace_result_get_key((libtrace_result_t *) p1) < libtrace_result_get_key((libtrace_result_t *) p2))
1560                return -1;
1561        if (libtrace_result_get_key((libtrace_result_t *) p1) == libtrace_result_get_key((libtrace_result_t *) p2))
1562                return 0;
1563        else
1564                return 1;
1565}
1566
1567/* Retrieves all results with the key requested from the temporary result
1568 * holding zone.
1569 */
1570DLLEXPORT int trace_get_results_check_temp(libtrace_t *libtrace, libtrace_vector_t *results, uint64_t key)
1571{
1572        int i;
1573
1574        libtrace_vector_empty(results);
1575        // Check all of the temp queues
1576        for (i = 0; i < libtrace->mapper_thread_count; ++i) {
1577                libtrace_result_t r = {0,0};
1578                assert (pthread_spin_lock(&libtrace->mapper_threads[i].tmp_spinlock) == 0);
1579                if (libtrace->mapper_threads[i].tmp_key == key) {
1580                        libtrace_result_set_key_value(&r, key, libtrace->mapper_threads[i].tmp_data);
1581                        libtrace->mapper_threads[i].tmp_data = NULL;
1582                        printf("Found in temp queue\n");
1583                }
1584                assert (pthread_spin_unlock(&libtrace->mapper_threads[i].tmp_spinlock) == 0);
1585                if (libtrace_result_get_value(&r)) {
1586                        // Got a result still in temporary
1587                        printf("Publishing from temp queue\n");
1588                        libtrace_vector_push_back(results, &r);
1589                } else {
1590                        // This might be waiting on the actual queue
1591                        libtrace_queue_t *v = &libtrace->mapper_threads[i].deque;
1592                        if (libtrace_deque_peek_front(v, (void *) &r) &&
1593                                        libtrace_result_get_value(&r)) {
1594                                assert (libtrace_deque_pop_front(&libtrace->mapper_threads[i].deque, (void *) &r) == 1);
1595                                printf("Found in real queue\n");
1596                                libtrace_vector_push_back(results, &r);
1597                        } // else no data (probably means no packets)
1598                        else {
1599                                printf("Result missing in real queue\n");
1600                        }
1601                }
1602        }
1603        //printf("Loop done yo, that means we've got #%d results to print fool!\n", libtrace_vector_get_size(results));
1604        return libtrace_vector_get_size(results);
1605}
1606
1607DLLEXPORT int trace_get_results(libtrace_t *libtrace, libtrace_vector_t * results) {
1608        int i;
1609        int flags = libtrace->reducer_flags; // Hint these aren't a changing
1610
1611        libtrace_vector_empty(results);
1612
1613        /* Here we assume queues are in order ascending order and they want
1614         * the smallest result first. If they are not in order the results
1615         * may not be in order.
1616         */
1617        if (flags & (REDUCE_SEQUENTIAL | REDUCE_ORDERED)) {
1618                int live_count = 0;
1619                bool live[libtrace->mapper_thread_count]; // Set if a trace is alive
1620                uint64_t key[libtrace->mapper_thread_count]; // Cached keys
1621                uint64_t min_key = UINT64_MAX; // XXX use max int here stdlimit.h?
1622                int min_queue = -1;
1623
1624                /* Loop through check all are alive (have data) and find the smallest */
1625                for (i = 0; i < libtrace->mapper_thread_count; ++i) {
1626                        libtrace_queue_t *v = &libtrace->mapper_threads[i].deque;
1627                        if (libtrace_deque_get_size(v) != 0) {
1628                                libtrace_result_t r;
1629                                libtrace_deque_peek_front(v, (void *) &r);
1630                                live_count++;
1631                                live[i] = 1;
1632                                key[i] = libtrace_result_get_key(&r);
1633                                if (i==0 || min_key > key[i]) {
1634                                        min_key = key[i];
1635                                        min_queue = i;
1636                                }
1637                        } else {
1638                                live[i] = 0;
1639                        }
1640                }
1641
1642                /* Now remove the smallest and loop - special case if all threads have joined we always flush whats left */
1643                while ((live_count == libtrace->mapper_thread_count) || (live_count &&
1644                                ((flags & REDUCE_SEQUENTIAL && min_key == libtrace->expected_key) ||
1645                                libtrace->joined))) {
1646                        /* Get the minimum queue and then do stuff */
1647                        libtrace_result_t r;
1648
1649                        assert (libtrace_deque_pop_front(&libtrace->mapper_threads[min_queue].deque, (void *) &r) == 1);
1650                        libtrace_vector_push_back(results, &r);
1651
1652                        // We expect the key we read +1 now
1653                        libtrace->expected_key = key[min_queue] + 1;
1654
1655                        // Now update the one we just removed
1656                        if (libtrace_deque_get_size(&libtrace->mapper_threads[min_queue].deque) )
1657                        {
1658                                libtrace_deque_peek_front(&libtrace->mapper_threads[min_queue].deque, (void *) &r);
1659                                key[min_queue] = libtrace_result_get_key(&r);
1660                                if (key[min_queue] <= min_key) {
1661                                        // We are still the smallest, might be out of order though :(
1662                                        min_key = key[min_queue];
1663                                } else {
1664                                        min_key = key[min_queue]; // Update our minimum
1665                                        // Check all find the smallest again - all are alive
1666                                        for (i = 0; i < libtrace->mapper_thread_count; ++i) {
1667                                                if (live[i] && min_key > key[i]) {
1668                                                        min_key = key[i];
1669                                                        min_queue = i;
1670                                                }
1671                                        }
1672                                }
1673                        } else {
1674                                live[min_queue] = 0;
1675                                live_count--;
1676                                min_key = UINT64_MAX; // Update our minimum
1677                                // Check all find the smallest again - all are alive
1678                                for (i = 0; i < libtrace->mapper_thread_count; ++i) {
1679                                        // Still not 100% TODO (what if order is wrong or not increasing)
1680                                        if (live[i] && min_key >= key[i]) {
1681                                                min_key = key[i];
1682                                                min_queue = i;
1683                                        }
1684                                }
1685                        }
1686                }
1687        } else { // Queues are not in order - return all results in the queue
1688                for (i = 0; i < libtrace->mapper_thread_count; i++) {
1689                        libtrace_vector_append(results, &libtrace->mapper_threads[i].vector);
1690                }
1691                if (flags & REDUCE_SORT) {
1692                        qsort(results->elements, results->size, results->element_size, &compareres);
1693                }
1694        }
1695        return libtrace_vector_get_size(results);
1696}
1697
1698DLLEXPORT uint64_t trace_packet_get_order(libtrace_packet_t * packet) {
1699        return packet->order;
1700}
1701
1702DLLEXPORT uint64_t trace_packet_get_hash(libtrace_packet_t * packet) {
1703        return packet->hash;
1704}
1705
1706DLLEXPORT void trace_packet_set_order(libtrace_packet_t * packet, uint64_t order) {
1707        packet->order = order;
1708}
1709
1710DLLEXPORT void trace_packet_set_hash(libtrace_packet_t * packet, uint64_t hash) {
1711        packet->hash = hash;
1712}
1713
1714DLLEXPORT int trace_finished(libtrace_t * libtrace) {
1715        int i;
1716        int b = 0;
1717        // TODO I don't like using this so much
1718        //assert(pthread_mutex_lock(&libtrace->libtrace_lock) == 0);
1719        for (i = 0; i < libtrace->mapper_thread_count; i++) {
1720                if (libtrace->mapper_threads[i].state == THREAD_RUNNING)
1721                        b++;
1722        }
1723        //assert(pthread_mutex_unlock(&libtrace->libtrace_lock) == 0);
1724        return !b;
1725}
1726
1727DLLEXPORT int trace_parallel_config(libtrace_t *libtrace, trace_parallel_option_t option, void *value)
1728{
1729        int ret = -1;
1730        switch (option) {
1731                case TRACE_OPTION_SET_HASHER:
1732                        return trace_set_hasher(libtrace, (enum hasher_types) *((int *) value), NULL, NULL);
1733                case TRACE_OPTION_SET_MAPPER_BUFFER_SIZE:
1734                        libtrace->mapper_buffer_size = *((int *) value);
1735                        return 1;
1736                case TRACE_OPTION_SET_PACKET_FREELIST_SIZE:
1737                        libtrace->packet_freelist_size = *((int *) value);
1738                        return 1;
1739                case TRACE_OPTION_SET_MAPPER_THREAD_COUNT:
1740                        libtrace->mapper_thread_count = *((int *) value);
1741                        return 1;
1742                case TRACE_DROP_OUT_OF_ORDER:
1743                        if (*((int *) value))
1744                                libtrace->reducer_flags |= REDUCE_DROP_OOO;
1745                        else
1746                                libtrace->reducer_flags &= ~REDUCE_DROP_OOO;
1747                        return 1;
1748                case TRACE_OPTION_SEQUENTIAL:
1749                        if (*((int *) value))
1750                                libtrace->reducer_flags |= REDUCE_SEQUENTIAL;
1751                        else
1752                                libtrace->reducer_flags &= ~REDUCE_SEQUENTIAL;
1753                        return 1;
1754                case TRACE_OPTION_ORDERED:
1755                        if (*((int *) value))
1756                                libtrace->reducer_flags |= REDUCE_ORDERED;
1757                        else
1758                                libtrace->reducer_flags &= ~REDUCE_ORDERED;
1759                        return 1;
1760                case TRACE_OPTION_USE_DEDICATED_HASHER:
1761                        if (*((int *) value))
1762                                libtrace->hasher_thread.type = THREAD_HASHER;
1763                        else
1764                                libtrace->hasher_thread.type = THREAD_EMPTY;
1765                        return 1;
1766                case TRACE_OPTION_USE_SLIDING_WINDOW_BUFFER:
1767                        if (*((int *) value))
1768                                libtrace->reducer_flags |= MAPPER_USE_SLIDING_WINDOW;
1769                        else
1770                                libtrace->reducer_flags &= ~MAPPER_USE_SLIDING_WINDOW;
1771                        return 1;
1772                case TRACE_OPTION_TRACETIME:
1773                        if(*((int *) value))
1774                                libtrace->tracetime = 1;
1775                        else
1776                                libtrace->tracetime = 0;
1777                        return 0;
1778        }
1779        return 0;
1780}
1781
1782DLLEXPORT libtrace_packet_t* trace_result_packet(libtrace_t * libtrace, libtrace_packet_t * packet) {
1783        libtrace_packet_t* result;
1784        if (!libtrace_ringbuffer_try_sread_bl(&libtrace->packet_freelist, (void **) &result))
1785                result = trace_create_packet();
1786        assert(result);
1787        swap_packets(result, packet); // Move the current packet into our copy
1788        return result;
1789}
1790
1791DLLEXPORT void trace_free_result_packet(libtrace_t * libtrace, libtrace_packet_t * packet) {
1792        // Try write back the packet
1793        assert(packet);
1794        // Always release any resources this might be holding such as a slot in a ringbuffer
1795        trace_fin_packet(packet);
1796        if (!libtrace_ringbuffer_try_swrite_bl(&libtrace->packet_freelist, packet)) {
1797                /* We couldn't, oh well lets just destroy it - XXX consider non managed formats i.e. rings buffers loosing packets and jamming up :( */
1798                //assert(1 == 90);
1799                trace_destroy_packet(packet);
1800        }
1801}
Note: See TracBrowser for help on using the repository browser.