aboutsummaryrefslogtreecommitdiff
path: root/broccoli/src
diff options
context:
space:
mode:
Diffstat (limited to 'broccoli/src')
-rw-r--r--broccoli/src/broevent.c65
-rw-r--r--broccoli/src/fifoqueue.c114
-rw-r--r--broccoli/src/midbropasad.c33
3 files changed, 212 insertions, 0 deletions
diff --git a/broccoli/src/broevent.c b/broccoli/src/broevent.c
new file mode 100644
index 0000000..3813270
--- /dev/null
+++ b/broccoli/src/broevent.c
@@ -0,0 +1,65 @@
+#include "fifoqueue.h"
+#include "broevent.h"
+#ifdef BROCCOLI
+#include <broccoli.h>
+#endif
+
+char *host_default = "127.0.0.1";
+char *port_default = "47760";
+Fifo_q * q;
+
+ static void
+bro_response(BroConn *conn, void *data, uint64* registers, uint64* uid)
+{
+ add_to_queue(q,create_sensor_object(*registers,*uid));
+ //printf("Received value %"PRIu64" from uid=%"PRIu64"\n",*registers,*uid);
+
+ conn = NULL;
+ data = NULL;
+}
+
+ void *
+bro_event_listener(void * args)
+{
+ q = (Fifo_q *) args;
+ int fd = -1;
+ BroConn *bc = NULL;
+ bro_init(NULL);
+ char hostname[512];
+
+ snprintf(hostname, 512, "%s:%s", host_default, port_default);
+ if (! (bc = bro_conn_new_str(hostname, BRO_CFLAG_RECONNECT | BRO_CFLAG_ALWAYS_QUEUE)))
+ {
+ printf("Could not get Bro connection handle.\n");
+ exit(-1);
+ }
+ bro_debug_calltrace = 0;
+ bro_debug_messages = 0;
+
+ bro_event_registry_add(bc, "response",(BroEventFunc) bro_response, NULL);
+
+ if (! bro_conn_connect(bc))
+ {
+ printf("Could not connect to Bro at %s:%s.\n", host_default,
+ port_default);
+ exit(-1);
+ }
+
+ fd =bro_conn_get_fd(bc);
+ fd_set rfds;
+ setbuf(stdout,NULL);
+
+ while(true)
+ {
+ FD_ZERO(&rfds);
+ FD_SET(fd,&rfds);
+ if(select(fd+1,&rfds,NULL,NULL,NULL) == -1){
+ perror("select()");
+ break;
+ }
+
+ bro_conn_process_input(bc);
+ }
+
+ bro_conn_delete(bc);
+}
diff --git a/broccoli/src/fifoqueue.c b/broccoli/src/fifoqueue.c
new file mode 100644
index 0000000..9b972e7
--- /dev/null
+++ b/broccoli/src/fifoqueue.c
@@ -0,0 +1,114 @@
+#include <pthread.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include "types.h"
+#include "fifoqueue.h"
+
+pthread_mutex_t lock;
+pthread_mutex_t bufferEmptyBlock;
+
+ Fifo_q *
+init_queue(int size)
+{
+ Fifo_q * q = (Fifo_q *) malloc(sizeof(Fifo_q));
+ q->head = NULL;
+ q->tail = NULL;
+ q->maxSize = size;
+ q->currentSize = 0;
+ if (pthread_mutex_init(&lock, NULL) != 0)
+ {
+ printf("WARNING: Couldn't initialize lock\n");
+ }
+ if (pthread_mutex_init(&bufferEmptyBlock, NULL) != 0)
+ {
+ printf("WARNING: Couldn't initialize blocking lock\n");
+ }
+ pthread_mutex_lock(&bufferEmptyBlock);
+ return q;
+}
+
+ boolean
+is_full(Fifo_q * q)
+{
+ if(q->currentSize < q->maxSize)
+ return false;
+ else
+ return true;
+}
+
+ boolean
+is_empty(Fifo_q * q)
+{
+ if(q->head==NULL)
+ return true;
+ else
+ return false;
+}
+
+ int
+add_to_queue(Fifo_q * q, Sensor_t * sensor)
+{
+
+ pthread_mutex_lock(&lock);
+ /* TODO delete first one if full */
+ if(q == NULL){
+ return -1;
+ }
+ else if(is_full(q)){
+ return -1;
+ }
+ Queue_t * new_elem = (Queue_t *) malloc(sizeof(Queue_t));
+ new_elem->next = NULL;
+ new_elem->sensor = sensor;
+ if(is_empty(q)){
+ q->head = new_elem;
+ pthread_mutex_unlock(&bufferEmptyBlock);
+ }else
+ q->tail->next = new_elem;
+ q->tail = new_elem;
+ q->currentSize++;
+ pthread_mutex_unlock(&lock);
+ return 1;
+}
+
+ Sensor_t *
+pop_from_queue(Fifo_q * q)
+{
+
+ if(is_empty(q)){
+ perror("The queue is empty");
+ pthread_mutex_lock(&bufferEmptyBlock);
+ }
+ pthread_mutex_lock(&lock);
+ Queue_t * head = q->head;
+ q->head = head->next;
+ Sensor_t * sensor = head->sensor;
+ free(head);
+ q->currentSize--;
+ pthread_mutex_unlock(&lock);
+ return sensor;
+}
+
+ Sensor_t *
+create_sensor_object(int value, int uid){
+ Sensor_t * sensor = (Sensor_t *) malloc(sizeof(Sensor_t));
+ sensor->value = value;
+ sensor->uid = uid;
+ return sensor;
+}
+ void
+print_queue(Fifo_q * q)
+{
+ pthread_mutex_lock(&lock);
+ Queue_t * current = q->head;
+ if(current == NULL){
+ printf("The queue is empty!");
+ return;
+ }
+ while(current != NULL){
+ printf("sensor value=%d, sensor uid=%d\n",
+ current->sensor->value, current->sensor->uid);
+ current = current->next;
+ }
+ pthread_mutex_unlock(&lock);
+}
diff --git a/broccoli/src/midbropasad.c b/broccoli/src/midbropasad.c
new file mode 100644
index 0000000..756e558
--- /dev/null
+++ b/broccoli/src/midbropasad.c
@@ -0,0 +1,33 @@
+#include <pthread.h>
+#include <unistd.h>
+#include "fifoqueue.h"
+#include "broevent.h"
+#ifdef BROCCOLI
+#include <broccoli.h>
+#endif
+
+ void
+start_data_capture(Fifo_q * q)
+{
+ int res;
+ pthread_t event_listener;
+ res = pthread_create(&event_listener, NULL, bro_event_listener, q);
+ if(res){
+ perror("Unable to create thread");
+ exit(-1);
+ }
+}
+
+ int
+main(int argc, char **argv)
+{
+ Fifo_q * q = init_queue(50);
+ start_data_capture(q);
+ while(true){
+ printf("Main thread\n");
+ sleep(10);
+ print_queue(q);
+ }
+ free(q);
+ return 0;
+}