2005-11-07 12:07:56 +00:00
|
|
|
/*
|
2006-07-07 08:49:06 +00:00
|
|
|
* Copyright (C) 2005-2006 Martin Willi
|
|
|
|
* Copyright (C) 2005 Jan Hutter
|
2005-11-07 12:07:56 +00:00
|
|
|
* Hochschule fuer Technik Rapperswil
|
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
|
|
* under the terms of the GNU General Public License as published by the
|
|
|
|
* Free Software Foundation; either version 2 of the License, or (at your
|
|
|
|
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful, but
|
|
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
|
|
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
|
|
* for more details.
|
2008-03-13 14:14:44 +00:00
|
|
|
*
|
|
|
|
* $Id$
|
2005-11-07 12:07:56 +00:00
|
|
|
*/
|
2005-11-09 09:11:06 +00:00
|
|
|
|
2005-11-07 12:07:56 +00:00
|
|
|
#include <stdlib.h>
|
|
|
|
#include <pthread.h>
|
2005-11-09 09:11:06 +00:00
|
|
|
|
2005-11-07 12:07:56 +00:00
|
|
|
#include "sender.h"
|
2005-11-11 13:31:52 +00:00
|
|
|
|
2005-11-29 10:25:07 +00:00
|
|
|
#include <daemon.h>
|
2005-11-23 09:43:14 +00:00
|
|
|
#include <network/socket.h>
|
2007-06-11 10:57:19 +00:00
|
|
|
#include <processing/jobs/callback_job.h>
|
2005-11-09 09:11:06 +00:00
|
|
|
|
2005-12-06 16:04:39 +00:00
|
|
|
|
2005-11-24 11:30:19 +00:00
|
|
|
typedef struct private_sender_t private_sender_t;
|
|
|
|
|
2005-11-07 12:07:56 +00:00
|
|
|
/**
|
2005-12-06 16:04:39 +00:00
|
|
|
* Private data of a sender_t object.
|
2005-11-07 12:07:56 +00:00
|
|
|
*/
|
2005-11-24 11:30:19 +00:00
|
|
|
struct private_sender_t {
|
2005-11-07 12:07:56 +00:00
|
|
|
/**
|
2005-12-06 16:04:39 +00:00
|
|
|
* Public part of a sender_t object.
|
2005-11-07 12:07:56 +00:00
|
|
|
*/
|
2007-06-11 10:57:19 +00:00
|
|
|
sender_t public;
|
2005-11-09 09:11:06 +00:00
|
|
|
|
2007-06-11 10:57:19 +00:00
|
|
|
/**
|
|
|
|
* Sender threads job.
|
|
|
|
*/
|
|
|
|
callback_job_t *job;
|
2007-03-28 13:34:02 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* The packets are stored in a linked list
|
|
|
|
*/
|
|
|
|
linked_list_t *list;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* mutex to synchronize access to list
|
|
|
|
*/
|
|
|
|
pthread_mutex_t mutex;
|
2005-11-07 12:07:56 +00:00
|
|
|
|
2007-03-28 13:34:02 +00:00
|
|
|
/**
|
2008-04-03 09:19:12 +00:00
|
|
|
* condvar to signal for packets added to list
|
2007-03-28 13:34:02 +00:00
|
|
|
*/
|
2008-04-03 09:19:12 +00:00
|
|
|
pthread_cond_t gotone;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* condvar to signal for packets sent
|
|
|
|
*/
|
|
|
|
pthread_cond_t sentone;
|
2005-11-07 12:07:56 +00:00
|
|
|
};
|
|
|
|
|
2007-03-28 13:34:02 +00:00
|
|
|
/**
|
|
|
|
* implements sender_t.send
|
|
|
|
*/
|
|
|
|
static void send_(private_sender_t *this, packet_t *packet)
|
|
|
|
{
|
|
|
|
host_t *src, *dst;
|
|
|
|
|
|
|
|
src = packet->get_source(packet);
|
|
|
|
dst = packet->get_destination(packet);
|
|
|
|
DBG1(DBG_NET, "sending packet: from %#H to %#H", src, dst);
|
|
|
|
|
|
|
|
pthread_mutex_lock(&this->mutex);
|
|
|
|
this->list->insert_last(this->list, packet);
|
2008-04-03 09:19:12 +00:00
|
|
|
pthread_cond_signal(&this->gotone);
|
2007-03-28 13:34:02 +00:00
|
|
|
pthread_mutex_unlock(&this->mutex);
|
|
|
|
}
|
|
|
|
|
2005-11-07 12:07:56 +00:00
|
|
|
/**
|
2005-12-06 16:04:39 +00:00
|
|
|
* Implementation of private_sender_t.send_packets.
|
2005-11-07 12:07:56 +00:00
|
|
|
*/
|
2007-06-11 10:57:19 +00:00
|
|
|
static job_requeue_t send_packets(private_sender_t * this)
|
2005-11-07 12:07:56 +00:00
|
|
|
{
|
2007-06-11 10:57:19 +00:00
|
|
|
packet_t *packet;
|
|
|
|
int oldstate;
|
2007-05-03 14:21:22 +00:00
|
|
|
|
2007-06-11 10:57:19 +00:00
|
|
|
pthread_mutex_lock(&this->mutex);
|
|
|
|
while (this->list->get_count(this->list) == 0)
|
2005-11-07 12:07:56 +00:00
|
|
|
{
|
2007-06-11 10:57:19 +00:00
|
|
|
/* add cleanup handler, wait for packet, remove cleanup handler */
|
|
|
|
pthread_cleanup_push((void(*)(void*))pthread_mutex_unlock, (void*)&this->mutex);
|
|
|
|
pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, &oldstate);
|
|
|
|
|
2008-04-03 09:19:12 +00:00
|
|
|
pthread_cond_wait(&this->gotone, &this->mutex);
|
2007-03-28 13:34:02 +00:00
|
|
|
|
2007-06-11 10:57:19 +00:00
|
|
|
pthread_setcancelstate(oldstate, NULL);
|
|
|
|
pthread_cleanup_pop(0);
|
2005-11-07 12:07:56 +00:00
|
|
|
}
|
2007-06-11 10:57:19 +00:00
|
|
|
this->list->remove_first(this->list, (void**)&packet);
|
2008-04-03 09:19:12 +00:00
|
|
|
pthread_cond_signal(&this->sentone);
|
2007-06-11 10:57:19 +00:00
|
|
|
pthread_mutex_unlock(&this->mutex);
|
|
|
|
|
|
|
|
charon->socket->send(charon->socket, packet);
|
|
|
|
packet->destroy(packet);
|
|
|
|
return JOB_REQUEUE_DIRECT;
|
2005-11-07 12:07:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2005-12-06 16:04:39 +00:00
|
|
|
* Implementation of sender_t.destroy.
|
2005-11-07 12:07:56 +00:00
|
|
|
*/
|
2005-11-28 20:29:47 +00:00
|
|
|
static void destroy(private_sender_t *this)
|
2005-11-07 12:07:56 +00:00
|
|
|
{
|
2007-05-03 14:21:22 +00:00
|
|
|
/* send all packets in the queue */
|
2008-04-03 09:19:12 +00:00
|
|
|
pthread_mutex_lock(&this->mutex);
|
2007-05-03 14:21:22 +00:00
|
|
|
while (this->list->get_count(this->list))
|
|
|
|
{
|
2008-04-03 09:19:12 +00:00
|
|
|
pthread_cond_wait(&this->sentone, &this->mutex);
|
2007-05-03 14:21:22 +00:00
|
|
|
}
|
2008-04-03 09:19:12 +00:00
|
|
|
pthread_mutex_unlock(&this->mutex);
|
|
|
|
pthread_mutex_destroy(&this->mutex);
|
2007-06-11 10:57:19 +00:00
|
|
|
this->job->cancel(this->job);
|
2007-05-03 14:21:22 +00:00
|
|
|
this->list->destroy(this->list);
|
2006-04-10 08:07:38 +00:00
|
|
|
free(this);
|
2005-11-07 12:07:56 +00:00
|
|
|
}
|
|
|
|
|
2005-11-25 13:42:58 +00:00
|
|
|
/*
|
2005-12-06 16:04:39 +00:00
|
|
|
* Described in header.
|
2005-11-25 13:42:58 +00:00
|
|
|
*/
|
2005-11-07 12:07:56 +00:00
|
|
|
sender_t * sender_create()
|
|
|
|
{
|
2006-04-10 08:07:38 +00:00
|
|
|
private_sender_t *this = malloc_thing(private_sender_t);
|
2005-11-09 09:11:06 +00:00
|
|
|
|
2007-03-28 13:34:02 +00:00
|
|
|
this->public.send = (void(*)(sender_t*,packet_t*))send_;
|
2005-11-28 20:29:47 +00:00
|
|
|
this->public.destroy = (void(*)(sender_t*)) destroy;
|
|
|
|
|
2007-03-28 13:34:02 +00:00
|
|
|
this->list = linked_list_create();
|
|
|
|
pthread_mutex_init(&this->mutex, NULL);
|
2008-04-03 09:19:12 +00:00
|
|
|
pthread_cond_init(&this->gotone, NULL);
|
|
|
|
pthread_cond_init(&this->sentone, NULL);
|
2007-03-28 13:34:02 +00:00
|
|
|
|
2007-06-11 10:57:19 +00:00
|
|
|
this->job = callback_job_create((callback_job_cb_t)send_packets,
|
|
|
|
this, NULL, NULL);
|
|
|
|
charon->processor->queue_job(charon->processor, (job_t*)this->job);
|
2005-11-09 09:11:06 +00:00
|
|
|
|
2007-06-11 10:57:19 +00:00
|
|
|
return &this->public;
|
2005-11-07 12:07:56 +00:00
|
|
|
}
|
2007-06-11 10:57:19 +00:00
|
|
|
|