#include <config.h>
#endif
+#define _REENTRANT
#define _GNU_SOURCE
+#include <features.h>
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <sys/poll.h>
#include <sys/mman.h>
#include <signal.h>
+#include <pthread.h>
/* Relayfs IOCTL */
#include <asm/ioctl.h>
unsigned int n_subbufs;
unsigned int subbuf_size;
void *mmap;
+ pthread_mutex_t mutex;
};
struct channel_trace_fd {
static char *channel_name = NULL;
static int daemon_mode = 0;
static int append_mode = 0;
+static unsigned long num_threads = 1;
volatile static int quit_program = 0; /* For signal handler */
/* Args :
printf("-c directory Root directory of the relayfs trace channels.\n");
printf("-d Run in background (daemon).\n");
printf("-a Append to an possibly existing trace.\n");
+ printf("-N Number of threads to start.\n");
printf("\n");
}
case 'a':
append_mode = 1;
break;
+ case 'N':
+ if(argn+1 < argc) {
+ num_threads = strtoul(argv[argn+1], NULL, 0);
+ argn++;
+ }
+ break;
default:
printf("Invalid argument '%s'.\n", argv[argn]);
printf("\n");
int read_subbuffer(struct fd_pair *pair)
{
unsigned int consumed_old;
- int err, ret;
+ int err, ret=0;
err = ioctl(pair->channel, RELAYFS_GET_SUBBUF,
&consumed_old);
printf("cookie : %u\n", consumed_old);
if(err != 0) {
- perror("Error in reserving sub buffer");
- ret = -EPERM;
+ ret = errno;
+ perror("Reserving sub buffer failed (everything is normal)");
goto get_error;
}
err = TEMP_FAILURE_RETRY(write(pair->trace,
- pair->mmap + (consumed_old & (~(pair->subbuf_size-1))),
+ pair->mmap
+ + (consumed_old & ((pair->n_subbufs * pair->subbuf_size)-1)),
pair->subbuf_size));
if(err < 0) {
+ ret = errno;
perror("Error in writing to file");
- ret = err;
goto write_error;
}
-
-
+#if 0
+ err = fsync(pair->trace);
+ if(err < 0) {
+ ret = errno;
+ perror("Error in writing to file");
+ goto write_error;
+ }
+#endif //0
write_error:
err = ioctl(pair->channel, RELAYFS_PUT_SUBBUF, &consumed_old);
if(err != 0) {
+ ret = errno;
if(errno == -EFAULT) {
- perror("Error in unreserving sub buffer");
- ret = -EFAULT;
+ perror("Error in unreserving sub buffer\n");
} else if(errno == -EIO) {
perror("Reader has been pushed by the writer, last subbuffer corrupted.");
/* FIXME : we may delete the last written buffer if we wish. */
- ret = -EIO;
}
goto get_error;
}
}
-/* read_channels
- *
- * Read the realyfs channels and write them in the paired tracefiles.
- *
- * @fd_pairs : paired channels and trace files.
- *
- * returns 0 on success, -1 on error.
- *
- * Note that the high priority polled channels are consumed first. We then poll
- * again to see if these channels are still in priority. Only when no
- * high priority channel is left, we start reading low priority channels.
- *
- * Note that a channel is considered high priority when the buffer is almost
- * full.
- */
-int read_channels(struct channel_trace_fd *fd_pairs)
+int map_channels(struct channel_trace_fd *fd_pairs)
{
- struct pollfd *pollfd;
int i,j;
- int num_rdy, num_hup;
- int high_prio;
- int ret;
+ int ret=0;
if(fd_pairs->num_pairs <= 0) {
printf("No channel to read\n");
perror("Error in getting the size of the subbuffers");
goto end;
}
+ ret = pthread_mutex_init(&pair->mutex, NULL); /* Fast mutex */
+ if(ret != 0) {
+ perror("Error in mutex init");
+ goto end;
+ }
}
/* Mmap each FD */
}
}
+ goto end; /* success */
+
+ /* Error handling */
+ /* munmap only the successfully mmapped indexes */
+munmap:
+ /* Munmap each FD */
+ for(j=0;j<i;j++) {
+ struct fd_pair *pair = &fd_pairs->pair[j];
+ int err_ret;
+
+ err_ret = munmap(pair->mmap, pair->subbuf_size * pair->n_subbufs);
+ if(err_ret != 0) {
+ perror("Error in munmap");
+ }
+ ret |= err_ret;
+ }
+
+end:
+ return ret;
+
+
+}
+
+
+int unmap_channels(struct channel_trace_fd *fd_pairs)
+{
+ int j;
+ int ret=0;
+
+ /* Munmap each FD */
+ for(j=0;j<fd_pairs->num_pairs;j++) {
+ struct fd_pair *pair = &fd_pairs->pair[j];
+ int err_ret;
+
+ err_ret = munmap(pair->mmap, pair->subbuf_size * pair->n_subbufs);
+ if(err_ret != 0) {
+ perror("Error in munmap");
+ }
+ ret |= err_ret;
+ err_ret = pthread_mutex_destroy(&pair->mutex);
+ if(err_ret != 0) {
+ perror("Error in mutex destroy");
+ }
+ ret |= err_ret;
+ }
+
+ return ret;
+}
+
+
+/* read_channels
+ *
+ * Thread worker.
+ *
+ * Read the relayfs channels and write them in the paired tracefiles.
+ *
+ * @fd_pairs : paired channels and trace files.
+ *
+ * returns (void*)0 on success, (void*)-1 on error.
+ *
+ * Note that the high priority polled channels are consumed first. We then poll
+ * again to see if these channels are still in priority. Only when no
+ * high priority channel is left, we start reading low priority channels.
+ *
+ * Note that a channel is considered high priority when the buffer is almost
+ * full.
+ */
+
+void * read_channels(void *arg)
+{
+ struct pollfd *pollfd;
+ int i,j;
+ int num_rdy, num_hup;
+ int high_prio;
+ int ret = 0;
+ struct channel_trace_fd *fd_pairs = (struct channel_trace_fd *)arg;
/* Start polling the FD */
num_hup++;
break;
case POLLPRI:
- printf("Urgent read on fd %d\n", pollfd[i].fd);
- /* Take care of high priority channels first. */
- high_prio = 1;
- ret |= read_subbuffer(&fd_pairs->pair[i]);
+ if(pthread_mutex_trylock(&fd_pairs->pair[i].mutex) == 0) {
+ printf("Urgent read on fd %d\n", pollfd[i].fd);
+ /* Take care of high priority channels first. */
+ high_prio = 1;
+ /* it's ok to have an unavailable subbuffer */
+ ret = read_subbuffer(&fd_pairs->pair[i]);
+ if(ret == -EAGAIN) ret = 0;
+
+ ret = pthread_mutex_unlock(&fd_pairs->pair[i].mutex);
+ if(ret)
+ printf("Error in mutex unlock : %s\n", strerror(ret));
+ }
break;
}
}
for(i=0;i<fd_pairs->num_pairs;i++) {
switch(pollfd[i].revents) {
case POLLIN:
- /* Take care of low priority channels. */
- printf("Normal read on fd %d\n", pollfd[i].fd);
- ret |= read_subbuffer(&fd_pairs->pair[i]);
+ if(pthread_mutex_trylock(&fd_pairs->pair[i].mutex) == 0) {
+ /* Take care of low priority channels. */
+ printf("Normal read on fd %d\n", pollfd[i].fd);
+ /* it's ok to have an unavailable subbuffer */
+ ret = read_subbuffer(&fd_pairs->pair[i]);
+ if(ret == -EAGAIN) ret = 0;
+
+ ret = pthread_mutex_unlock(&fd_pairs->pair[i].mutex);
+ if(ret)
+ printf("Error in mutex unlock : %s\n", strerror(ret));
+ }
break;
}
}
free_fd:
free(pollfd);
- /* munmap only the successfully mmapped indexes */
- i = fd_pairs->num_pairs;
-munmap:
- /* Munmap each FD */
- for(j=0;j<i;j++) {
- struct fd_pair *pair = &fd_pairs->pair[j];
- int err_ret;
-
- err_ret = munmap(pair->mmap, pair->subbuf_size * pair->n_subbufs);
- if(err_ret != 0) {
- perror("Error in munmap");
- }
- ret |= err_ret;
- }
-
end:
- return ret;
+ return (void*)ret;
}
int main(int argc, char ** argv)
{
- int ret;
+ int ret = 0;
struct channel_trace_fd fd_pairs = { NULL, 0 };
struct sigaction act;
+ pthread_t *tids;
+ unsigned int i;
+ void *tret;
ret = parse_arguments(argc, argv);
sigaction(SIGQUIT, &act, NULL);
sigaction(SIGINT, &act, NULL);
- //return 0;
+
if(ret = open_channel_trace_pairs(channel_name, trace_name, &fd_pairs))
goto close_channel;
- ret = read_channels(&fd_pairs);
+ if(ret = map_channels(&fd_pairs))
+ goto close_channel;
+
+ tids = malloc(sizeof(pthread_t) * num_threads);
+ for(i=0; i<num_threads; i++) {
+ ret = pthread_create(&tids[i], NULL, read_channels, &fd_pairs);
+ if(ret) {
+ perror("Error creating thread");
+ break;
+ }
+ }
+
+ for(i=0; i<num_threads; i++) {
+ ret = pthread_join(tids[i], &tret);
+ if(ret) {
+ perror("Error joining thread");
+ break;
+ }
+ if((int)tret != 0) {
+ printf("Error %s occured in thread %u\n", strerror((int)tret), i);
+ }
+ }
+
+ free(tids);
+
+ ret |= unmap_channels(&fd_pairs);
close_channel:
close_channel_trace_pairs(&fd_pairs);
projects / ltt-control.git / blobdiff