[lttv.git] / ltt / branches / poly / lttd / lttd.c

/* lttd
 *
 * Linux Trace Toolkit Daemon
 *
 * This is a simple daemon that reads a few relayfs channels and save them in a
 * trace.
 *
 *
 * Copyright 2005 -
 * 	Mathieu Desnoyers <mathieu.desnoyers@polymtl.ca>
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#define _GNU_SOURCE
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <stdlib.h>
#include <dirent.h>
#include <string.h>
#include <fcntl.h>
#include <sys/poll.h>
#include <sys/mman.h>
#include <signal.h>

/* Relayfs IOCTL */
#include <asm/ioctl.h>
#include <asm/types.h>

/* Get the next sub buffer that can be read. */
#define RELAYFS_GET_SUBBUF        _IOR(0xF4, 0x00,__u32)
/* Release the oldest reserved (by "get") sub buffer. */
#define RELAYFS_PUT_SUBBUF        _IOW(0xF4, 0x01,__u32)
/* returns the number of sub buffers in the per cpu channel. */
#define RELAYFS_GET_N_SUBBUFS     _IOR(0xF4, 0x02,__u32)
/* returns the size of the sub buffers. */
#define RELAYFS_GET_SUBBUF_SIZE   _IOR(0xF4, 0x03,__u32)


enum {
	GET_SUBBUF,
	PUT_SUBBUF,
	GET_N_BUBBUFS,
	GET_SUBBUF_SIZE
};

struct fd_pair {
	int channel;
	int trace;
	unsigned int n_subbufs;
	unsigned int subbuf_size;
	void *mmap;
};

struct channel_trace_fd {
	struct fd_pair *pair;
	int num_pairs;
};

static char *trace_name = NULL;
static char *channel_name = NULL;
static int	daemon_mode = 0;
static int	append_mode = 0;
volatile static int	quit_program = 0;	/* For signal handler */

/* Args :
 *
 * -t directory		Directory name of the trace to write to. Will be created.
 * -c directory		Root directory of the relayfs trace channels.
 * -d          		Run in background (daemon).
 * -a							Trace append mode.
 * -s							Send SIGUSR1 to parent when ready for IO.
 */
void show_arguments(void)
{
	printf("Please use the following arguments :\n");
	printf("\n");
	printf("-t directory  Directory name of the trace to write to.\n"
				 "              It will be created.\n");
	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");
}


/* parse_arguments
 *
 * Parses the command line arguments.
 *
 * Returns 1 if the arguments were correct, but doesn't ask for program
 * continuation. Returns -1 if the arguments are incorrect, or 0 if OK.
 */
int parse_arguments(int argc, char **argv)
{
	int ret = 0;
	int argn = 1;
	
	if(argc == 2) {
		if(strcmp(argv[1], "-h") == 0) {
			return 1;
		}
	}

	while(argn < argc) {

		switch(argv[argn][0]) {
			case '-':
				switch(argv[argn][1]) {
					case 't':
						if(argn+1 < argc) {
							trace_name = argv[argn+1];
							argn++;
						}
						break;
					case 'c':
						if(argn+1 < argc) {
							channel_name = argv[argn+1];
							argn++;
						}
						break;
					case 'd':
						daemon_mode = 1;
						break;
					case 'a':
						append_mode = 1;
						break;
					default:
						printf("Invalid argument '%s'.\n", argv[argn]);
						printf("\n");
						ret = -1;
				}
				break;
			default:
				printf("Invalid argument '%s'.\n", argv[argn]);
				printf("\n");
				ret = -1;
		}
		argn++;
	}
	
	if(trace_name == NULL) {
		printf("Please specify a trace name.\n");
		printf("\n");
		ret = -1;
	}
	
	if(channel_name == NULL) {
		printf("Please specify a channel name.\n");
		printf("\n");
		ret = -1;
	}
	
	return ret;
}

void show_info(void)
{
	printf("Linux Trace Toolkit Trace Daemon\n");
	printf("\n");
	printf("Reading from relayfs directory : %s\n", channel_name);
	printf("Writing to trace directory : %s\n", trace_name);
	printf("\n");
}


/* signal handling */

static void handler(int signo)
{
	printf("Signal %d received : exiting cleanly\n", signo);
	quit_program = 1;
}


int open_channel_trace_pairs(char *subchannel_name, char *subtrace_name,
		struct channel_trace_fd *fd_pairs)
{
	DIR *channel_dir = opendir(subchannel_name);
	struct dirent *entry;
	struct stat stat_buf;
	int ret;
	char path_channel[PATH_MAX];
	int path_channel_len;
	char *path_channel_ptr;
	char path_trace[PATH_MAX];
	int path_trace_len;
	char *path_trace_ptr;
  int open_ret = 0;

	if(channel_dir == NULL) {
		perror(subchannel_name);
		open_ret = ENOENT;
    goto end;
	}

	printf("Creating trace subdirectory %s\n", subtrace_name);
	ret = mkdir(subtrace_name, S_IRWXU|S_IRWXG|S_IRWXO);
	if(ret == -1) {
		if(errno != EEXIST) {
			perror(subtrace_name);
      open_ret = -1;
			goto end;
		}
	}

	strncpy(path_channel, subchannel_name, PATH_MAX-1);
	path_channel_len = strlen(path_channel);
	path_channel[path_channel_len] = '/';
	path_channel_len++;
	path_channel_ptr = path_channel + path_channel_len;

	strncpy(path_trace, subtrace_name, PATH_MAX-1);
	path_trace_len = strlen(path_trace);
	path_trace[path_trace_len] = '/';
	path_trace_len++;
	path_trace_ptr = path_trace + path_trace_len;
	
	while((entry = readdir(channel_dir)) != NULL) {

		if(entry->d_name[0] == '.') continue;
		
		strncpy(path_channel_ptr, entry->d_name, PATH_MAX - path_channel_len);
		strncpy(path_trace_ptr, entry->d_name, PATH_MAX - path_trace_len);
		
		ret = stat(path_channel, &stat_buf);
		if(ret == -1) {
			perror(path_channel);
			continue;
		}
		
		printf("Channel file : %s\n", path_channel);
		
		if(S_ISDIR(stat_buf.st_mode)) {

			printf("Entering channel subdirectory...\n");
			ret = open_channel_trace_pairs(path_channel, path_trace, fd_pairs);
			if(ret < 0) continue;
		} else if(S_ISREG(stat_buf.st_mode)) {
			printf("Opening file.\n");
			
			fd_pairs->pair = realloc(fd_pairs->pair,
					++fd_pairs->num_pairs * sizeof(struct fd_pair));

			/* Open the channel in read mode */
			fd_pairs->pair[fd_pairs->num_pairs-1].channel = 
				open(path_channel, O_RDONLY | O_NONBLOCK);
			if(fd_pairs->pair[fd_pairs->num_pairs-1].channel == -1) {
				perror(path_channel);
				fd_pairs->num_pairs--;
				continue;
			}
			/* Open the trace in write mode, only append if append_mode */
			ret = stat(path_trace, &stat_buf);
			if(ret == 0) {
				if(append_mode) {
					printf("Appending to file %s as requested\n", path_trace);

					fd_pairs->pair[fd_pairs->num_pairs-1].trace = 
						open(path_trace, O_WRONLY|O_APPEND,
								S_IRWXU|S_IRWXG|S_IRWXO);

					if(fd_pairs->pair[fd_pairs->num_pairs-1].trace == -1) {
						perror(path_trace);
					}
				} else {
					printf("File %s exists, cannot open. Try append mode.\n", path_trace);
					open_ret = -1;
          goto end;
				}
			} else {
				if(errno == ENOENT) {
					fd_pairs->pair[fd_pairs->num_pairs-1].trace = 
						open(path_trace, O_WRONLY|O_CREAT|O_EXCL,
								S_IRWXU|S_IRWXG|S_IRWXO);
					if(fd_pairs->pair[fd_pairs->num_pairs-1].trace == -1) {
						perror(path_trace);
					}
				}
			}
		}
	}
	
end:
	closedir(channel_dir);

	return open_ret;
}


int read_subbuffer(struct fd_pair *pair)
{
	unsigned int	consumed_old;
	int err, ret;


	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;
		goto get_error;
	}
	
	err = TEMP_FAILURE_RETRY(write(pair->trace,
				pair->mmap 
					+ (consumed_old & ((pair->n_subbufs * pair->subbuf_size)-1)),
				pair->subbuf_size));

	if(err < 0) {
		perror("Error in writing to file");
		ret = err;
		goto write_error;
	}


write_error:
	err = ioctl(pair->channel, RELAYFS_PUT_SUBBUF, &consumed_old);
	if(err != 0) {
		if(errno == -EFAULT) {
			perror("Error in unreserving sub buffer");
			ret = -EFAULT;
		} 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;
	}

get_error:
	return ret;
}


/* 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)
{
	struct pollfd *pollfd;
	int i,j;
	int num_rdy, num_hup;
	int high_prio;
	int ret;

	if(fd_pairs->num_pairs <= 0) {
		printf("No channel to read\n");
		goto end;
	}
	
	/* Get the subbuf sizes and number */

	for(i=0;i<fd_pairs->num_pairs;i++) {
		struct fd_pair *pair = &fd_pairs->pair[i];

		ret = ioctl(pair->channel, RELAYFS_GET_N_SUBBUFS, 
							&pair->n_subbufs);
		if(ret != 0) {
			perror("Error in getting the number of subbuffers");
			goto end;
		}
		ret = ioctl(pair->channel, RELAYFS_GET_SUBBUF_SIZE, 
							&pair->subbuf_size);
		if(ret != 0) {
			perror("Error in getting the size of the subbuffers");
			goto end;
		}
	}

	/* Mmap each FD */
	for(i=0;i<fd_pairs->num_pairs;i++) {
		struct fd_pair *pair = &fd_pairs->pair[i];

		pair->mmap = mmap(0, pair->subbuf_size * pair->n_subbufs, PROT_READ,
				MAP_SHARED, pair->channel, 0);
		if(pair->mmap == MAP_FAILED) {
			perror("Mmap error");
			goto munmap;
		}
	}


	/* Start polling the FD */
	
	pollfd = malloc(fd_pairs->num_pairs * sizeof(struct pollfd));

	/* Note : index in pollfd is the same index as fd_pair->pair */
	for(i=0;i<fd_pairs->num_pairs;i++) {
		pollfd[i].fd = fd_pairs->pair[i].channel;
		pollfd[i].events = POLLIN|POLLPRI;
	}

	while(1) {
		high_prio = 0;
		num_hup = 0; 
#ifdef DEBUG
		printf("Press a key for next poll...\n");
		char buf[1];
		read(STDIN_FILENO, &buf, 1);
		printf("Next poll (polling %d fd) :\n", fd_pairs->num_pairs);
#endif //DEBUG
		
		/* Have we received a signal ? */
		if(quit_program) break;
		
		num_rdy = poll(pollfd, fd_pairs->num_pairs, -1);
		if(num_rdy == -1) {
			perror("Poll error");
			goto free_fd;
		}

		printf("Data received\n");

		for(i=0;i<fd_pairs->num_pairs;i++) {
			switch(pollfd[i].revents) {
				case POLLERR:
					printf("Error returned in polling fd %d.\n", pollfd[i].fd);
					num_hup++;
					break;
				case POLLHUP:
					printf("Polling fd %d tells it has hung up.\n", pollfd[i].fd);
					num_hup++;
					break;
				case POLLNVAL:
					printf("Polling fd %d tells fd is not open.\n", pollfd[i].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]);
					break;
			}
		}
		/* If every FD has hung up, we end the read loop here */
		if(num_hup == fd_pairs->num_pairs) break;

		if(!high_prio) {
			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]);
						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;
}


void close_channel_trace_pairs(struct channel_trace_fd *fd_pairs)
{
	int i;
	int ret;

	for(i=0;i<fd_pairs->num_pairs;i++) {
		ret = close(fd_pairs->pair[i].channel);
		if(ret == -1) perror("Close error on channel");
		ret = close(fd_pairs->pair[i].trace);
		if(ret == -1) perror("Close error on trace");
	}
	free(fd_pairs->pair);
}

int main(int argc, char ** argv)
{
	int ret;
	struct channel_trace_fd fd_pairs = { NULL, 0 };
	struct sigaction act;
	
	ret = parse_arguments(argc, argv);

	if(ret != 0) show_arguments();
	if(ret < 0) return EINVAL;
	if(ret > 0) return 0;

	show_info();

	if(daemon_mode) {
		ret = daemon(0, 0);
    
    if(ret == -1) {
      perror("An error occured while daemonizing.");
      exit(-1);
    }
  }

	/* Connect the signal handlers */
	act.sa_handler = handler;
	act.sa_flags = 0;
	sigemptyset(&(act.sa_mask));
	sigaddset(&(act.sa_mask), SIGTERM);
	sigaddset(&(act.sa_mask), SIGQUIT);
	sigaddset(&(act.sa_mask), SIGINT);
	sigaction(SIGTERM, &act, NULL);
	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);

close_channel:
	close_channel_trace_pairs(&fd_pairs);

	return ret;
}
Commit	Line	Data
6cecb9cf	1	/* lttd
	2	*
	3	* Linux Trace Toolkit Daemon
	4	*
	5	* This is a simple daemon that reads a few relayfs channels and save them in a
	6	* trace.
	7	*
	8	*
	9	* Copyright 2005 -
	10	* Mathieu Desnoyers <mathieu.desnoyers@polymtl.ca>
	11	*/
	12
4e4d11b3	13	#ifdef HAVE_CONFIG_H
13474e75	14	#include <config.h>
4e4d11b3	15	#endif
	16
	17	#define _GNU_SOURCE
6cecb9cf	18	#include <stdio.h>
af4061b2	19	#include <unistd.h>
6cecb9cf	20	#include <errno.h>
	21	#include <sys/types.h>
	22	#include <sys/stat.h>
6cecb9cf	23	#include <stdlib.h>
	24	#include <dirent.h>
	25	#include <string.h>
9e7dbe44	26	#include <fcntl.h>
9e7dbe44	27	#include <sys/poll.h>
af4061b2	28	#include <sys/mman.h>
	29	#include <signal.h>
	30
	31	/* Relayfs IOCTL */
	32	#include <asm/ioctl.h>
	33	#include <asm/types.h>
	34
	35	/* Get the next sub buffer that can be read. */
	36	#define RELAYFS_GET_SUBBUF _IOR(0xF4, 0x00,__u32)
	37	/* Release the oldest reserved (by "get") sub buffer. */
b2c7cb27	38	#define RELAYFS_PUT_SUBBUF _IOW(0xF4, 0x01,__u32)
af4061b2	39	/* returns the number of sub buffers in the per cpu channel. */
	40	#define RELAYFS_GET_N_SUBBUFS _IOR(0xF4, 0x02,__u32)
	41	/* returns the size of the sub buffers. */
	42	#define RELAYFS_GET_SUBBUF_SIZE _IOR(0xF4, 0x03,__u32)
	43
	44
6cecb9cf	45
	46	enum {
	47	GET_SUBBUF,
	48	PUT_SUBBUF,
	49	GET_N_BUBBUFS,
	50	GET_SUBBUF_SIZE
	51	};
	52
	53	struct fd_pair {
	54	int channel;
	55	int trace;
af4061b2	56	unsigned int n_subbufs;
	57	unsigned int subbuf_size;
	58	void *mmap;
6cecb9cf	59	};
	60
	61	struct channel_trace_fd {
	62	struct fd_pair *pair;
	63	int num_pairs;
	64	};
	65
	66	static char *trace_name = NULL;
	67	static char *channel_name = NULL;
	68	static int daemon_mode = 0;
9e7dbe44	69	static int append_mode = 0;
af4061b2	70	volatile static int quit_program = 0; /* For signal handler */
6cecb9cf	71
	72	/* Args :
	73	*
	74	* -t directory Directory name of the trace to write to. Will be created.
	75	* -c directory Root directory of the relayfs trace channels.
	76	* -d Run in background (daemon).
9e7dbe44	77	* -a Trace append mode.
86a65fdb	78	* -s Send SIGUSR1 to parent when ready for IO.
6cecb9cf	79	*/
	80	void show_arguments(void)
	81	{
	82	printf("Please use the following arguments :\n");
	83	printf("\n");
	84	printf("-t directory Directory name of the trace to write to.\n"
	85	" It will be created.\n");
	86	printf("-c directory Root directory of the relayfs trace channels.\n");
	87	printf("-d Run in background (daemon).\n");
9e7dbe44	88	printf("-a Append to an possibly existing trace.\n");
6cecb9cf	89	printf("\n");
	90	}
	91
	92
	93	/* parse_arguments
	94	*
	95	* Parses the command line arguments.
	96	*
	97	* Returns 1 if the arguments were correct, but doesn't ask for program
	98	* continuation. Returns -1 if the arguments are incorrect, or 0 if OK.
	99	*/
	100	int parse_arguments(int argc, char **argv)
	101	{
	102	int ret = 0;
	103	int argn = 1;
	104
	105	if(argc == 2) {
	106	if(strcmp(argv[1], "-h") == 0) {
	107	return 1;
	108	}
	109	}
	110
9e7dbe44	111	while(argn < argc) {
6cecb9cf	112
	113	switch(argv[argn][0]) {
	114	case '-':
	115	switch(argv[argn][1]) {
	116	case 't':
9e7dbe44	117	if(argn+1 < argc) {
	118	trace_name = argv[argn+1];
	119	argn++;
	120	}
6cecb9cf	121	break;
6cecb9cf	122	case 'c':
9e7dbe44	123	if(argn+1 < argc) {
	124	channel_name = argv[argn+1];
	125	argn++;
	126	}
6cecb9cf	127	break;
	128	case 'd':
	129	daemon_mode = 1;
	130	break;
9e7dbe44	131	case 'a':
	132	append_mode = 1;
	133	break;
6cecb9cf	134	default:
	135	printf("Invalid argument '%s'.\n", argv[argn]);
	136	printf("\n");
	137	ret = -1;
	138	}
	139	break;
	140	default:
	141	printf("Invalid argument '%s'.\n", argv[argn]);
	142	printf("\n");
	143	ret = -1;
	144	}
	145	argn++;
	146	}
	147
	148	if(trace_name == NULL) {
	149	printf("Please specify a trace name.\n");
	150	printf("\n");
	151	ret = -1;
	152	}
	153
	154	if(channel_name == NULL) {
	155	printf("Please specify a channel name.\n");
	156	printf("\n");
	157	ret = -1;
	158	}
	159
	160	return ret;
	161	}
	162
	163	void show_info(void)
	164	{
	165	printf("Linux Trace Toolkit Trace Daemon\n");
	166	printf("\n");
	167	printf("Reading from relayfs directory : %s\n", channel_name);
	168	printf("Writing to trace directory : %s\n", trace_name);
	169	printf("\n");
	170	}
	171
	172
af4061b2	173	/* signal handling */
	174
	175	static void handler(int signo)
	176	{
	177	printf("Signal %d received : exiting cleanly\n", signo);
	178	quit_program = 1;
	179	}
	180
	181
	182
6cecb9cf	183	int open_channel_trace_pairs(char subchannel_name, char subtrace_name,
	184	struct channel_trace_fd *fd_pairs)
	185	{
	186	DIR *channel_dir = opendir(subchannel_name);
	187	struct dirent *entry;
	188	struct stat stat_buf;
	189	int ret;
	190	char path_channel[PATH_MAX];
	191	int path_channel_len;
	192	char *path_channel_ptr;
	193	char path_trace[PATH_MAX];
	194	int path_trace_len;
	195	char *path_trace_ptr;
02ba971e	196	int open_ret = 0;
6cecb9cf	197
	198	if(channel_dir == NULL) {
	199	perror(subchannel_name);
02ba971e	200	open_ret = ENOENT;
02ba971e	201	goto end;
6cecb9cf	202	}
6cecb9cf	203
6cecb9cf	204	printf("Creating trace subdirectory %s\n", subtrace_name);
	205	ret = mkdir(subtrace_name, S_IRWXU\|S_IRWXG\|S_IRWXO);
	206	if(ret == -1) {
3986d00b	207	if(errno != EEXIST) {
9e7dbe44	208	perror(subtrace_name);
02ba971e	209	open_ret = -1;
02ba971e	210	goto end;
9e7dbe44	211	}
6cecb9cf	212	}
	213
	214	strncpy(path_channel, subchannel_name, PATH_MAX-1);
	215	path_channel_len = strlen(path_channel);
	216	path_channel[path_channel_len] = '/';
	217	path_channel_len++;
	218	path_channel_ptr = path_channel + path_channel_len;
	219
	220	strncpy(path_trace, subtrace_name, PATH_MAX-1);
	221	path_trace_len = strlen(path_trace);
	222	path_trace[path_trace_len] = '/';
	223	path_trace_len++;
	224	path_trace_ptr = path_trace + path_trace_len;
	225
	226	while((entry = readdir(channel_dir)) != NULL) {
	227
	228	if(entry->d_name[0] == '.') continue;
	229
	230	strncpy(path_channel_ptr, entry->d_name, PATH_MAX - path_channel_len);
	231	strncpy(path_trace_ptr, entry->d_name, PATH_MAX - path_trace_len);
	232
	233	ret = stat(path_channel, &stat_buf);
	234	if(ret == -1) {
	235	perror(path_channel);
	236	continue;
	237	}
	238
	239	printf("Channel file : %s\n", path_channel);
	240
	241	if(S_ISDIR(stat_buf.st_mode)) {
	242
	243	printf("Entering channel subdirectory...\n");
	244	ret = open_channel_trace_pairs(path_channel, path_trace, fd_pairs);
	245	if(ret < 0) continue;
9e7dbe44	246	} else if(S_ISREG(stat_buf.st_mode)) {
	247	printf("Opening file.\n");
	248
	249	fd_pairs->pair = realloc(fd_pairs->pair,
	250	++fd_pairs->num_pairs * sizeof(struct fd_pair));
	251
	252	/* Open the channel in read mode */
	253	fd_pairs->pair[fd_pairs->num_pairs-1].channel =
	254	open(path_channel, O_RDONLY \| O_NONBLOCK);
	255	if(fd_pairs->pair[fd_pairs->num_pairs-1].channel == -1) {
	256	perror(path_channel);
13474e75	257	fd_pairs->num_pairs--;
13474e75	258	continue;
9e7dbe44	259	}
	260	/* Open the trace in write mode, only append if append_mode */
	261	ret = stat(path_trace, &stat_buf);
	262	if(ret == 0) {
	263	if(append_mode) {
251b5c5e	264	printf("Appending to file %s as requested\n", path_trace);
9e7dbe44	265
	266	fd_pairs->pair[fd_pairs->num_pairs-1].trace =
	267	open(path_trace, O_WRONLY\|O_APPEND,
	268	S_IRWXU\|S_IRWXG\|S_IRWXO);
	269
	270	if(fd_pairs->pair[fd_pairs->num_pairs-1].trace == -1) {
	271	perror(path_trace);
	272	}
	273	} else {
	274	printf("File %s exists, cannot open. Try append mode.\n", path_trace);
02ba971e	275	open_ret = -1;
02ba971e	276	goto end;
9e7dbe44	277	}
	278	} else {
	279	if(errno == ENOENT) {
	280	fd_pairs->pair[fd_pairs->num_pairs-1].trace =
	281	open(path_trace, O_WRONLY\|O_CREAT\|O_EXCL,
	282	S_IRWXU\|S_IRWXG\|S_IRWXO);
	283	if(fd_pairs->pair[fd_pairs->num_pairs-1].trace == -1) {
	284	perror(path_trace);
	285	}
	286	}
	287	}
6cecb9cf	288	}
6cecb9cf	289	}
6cecb9cf	290
02ba971e	291	end:
6cecb9cf	292	closedir(channel_dir);
6cecb9cf	293
02ba971e	294	return open_ret;
6cecb9cf	295	}
6cecb9cf	296
af4061b2	297
	298	int read_subbuffer(struct fd_pair *pair)
	299	{
12bcec97	300	unsigned int consumed_old;
13474e75	301	int err, ret;
af4061b2	302
af4061b2	303
13474e75	304	err = ioctl(pair->channel, RELAYFS_GET_SUBBUF,
12bcec97	305	&consumed_old);
12bcec97	306	printf("cookie : %u\n", consumed_old);
13474e75	307	if(err != 0) {
af4061b2	308	perror("Error in reserving sub buffer");
13474e75	309	ret = -EPERM;
13474e75	310	goto get_error;
af4061b2	311	}
13474e75	312
13474e75	313	err = TEMP_FAILURE_RETRY(write(pair->trace,
2312de30	314	pair->mmap
2312de30	315	+ (consumed_old & ((pair->n_subbufs * pair->subbuf_size)-1)),
af4061b2	316	pair->subbuf_size));
13474e75	317
13474e75	318	if(err < 0) {
af4061b2	319	perror("Error in writing to file");
13474e75	320	ret = err;
13474e75	321	goto write_error;
af4061b2	322	}
	323
	324
13474e75	325	write_error:
12bcec97	326	err = ioctl(pair->channel, RELAYFS_PUT_SUBBUF, &consumed_old);
13474e75	327	if(err != 0) {
12bcec97	328	if(errno == -EFAULT) {
	329	perror("Error in unreserving sub buffer");
	330	ret = -EFAULT;
	331	} else if(errno == -EIO) {
	332	perror("Reader has been pushed by the writer, last subbuffer corrupted.");
db9d75c0	333	/* FIXME : we may delete the last written buffer if we wish. */
12bcec97	334	ret = -EIO;
12bcec97	335	}
13474e75	336	goto get_error;
af4061b2	337	}
af4061b2	338
13474e75	339	get_error:
13474e75	340	return ret;
af4061b2	341	}
	342
	343
9e7dbe44	344	/* read_channels
	345	*
	346	* Read the realyfs channels and write them in the paired tracefiles.
	347	*
	348	* @fd_pairs : paired channels and trace files.
	349	*
	350	* returns 0 on success, -1 on error.
	351	*
	352	* Note that the high priority polled channels are consumed first. We then poll
	353	* again to see if these channels are still in priority. Only when no
	354	* high priority channel is left, we start reading low priority channels.
	355	*
	356	* Note that a channel is considered high priority when the buffer is almost
	357	* full.
	358	*/
6cecb9cf	359
	360	int read_channels(struct channel_trace_fd *fd_pairs)
	361	{
9e7dbe44	362	struct pollfd *pollfd;
af4061b2	363	int i,j;
af4061b2	364	int num_rdy, num_hup;
68047576	365	int high_prio;
af4061b2	366	int ret;
af4061b2	367
13474e75	368	if(fd_pairs->num_pairs <= 0) {
	369	printf("No channel to read\n");
	370	goto end;
	371	}
	372
af4061b2	373	/* Get the subbuf sizes and number */
	374
	375	for(i=0;i<fd_pairs->num_pairs;i++) {
	376	struct fd_pair *pair = &fd_pairs->pair[i];
9e7dbe44	377
af4061b2	378	ret = ioctl(pair->channel, RELAYFS_GET_N_SUBBUFS,
	379	&pair->n_subbufs);
	380	if(ret != 0) {
	381	perror("Error in getting the number of subbuffers");
	382	goto end;
	383	}
	384	ret = ioctl(pair->channel, RELAYFS_GET_SUBBUF_SIZE,
	385	&pair->subbuf_size);
	386	if(ret != 0) {
	387	perror("Error in getting the size of the subbuffers");
	388	goto end;
	389	}
	390	}
	391
	392	/* Mmap each FD */
	393	for(i=0;i<fd_pairs->num_pairs;i++) {
	394	struct fd_pair *pair = &fd_pairs->pair[i];
	395
	396	pair->mmap = mmap(0, pair->subbuf_size * pair->n_subbufs, PROT_READ,
	397	MAP_SHARED, pair->channel, 0);
	398	if(pair->mmap == MAP_FAILED) {
	399	perror("Mmap error");
	400	goto munmap;
	401	}
	402	}
	403
	404
	405	/* Start polling the FD */
	406
9e7dbe44	407	pollfd = malloc(fd_pairs->num_pairs * sizeof(struct pollfd));
9e7dbe44	408
af4061b2	409	/* Note : index in pollfd is the same index as fd_pair->pair */
9e7dbe44	410	for(i=0;i<fd_pairs->num_pairs;i++) {
	411	pollfd[i].fd = fd_pairs->pair[i].channel;
	412	pollfd[i].events = POLLIN\|POLLPRI;
	413	}
	414
	415	while(1) {
68047576	416	high_prio = 0;
af4061b2	417	num_hup = 0;
	418	#ifdef DEBUG
	419	printf("Press a key for next poll...\n");
	420	char buf[1];
	421	read(STDIN_FILENO, &buf, 1);
13474e75	422	printf("Next poll (polling %d fd) :\n", fd_pairs->num_pairs);
af4061b2	423	#endif //DEBUG
	424
	425	/* Have we received a signal ? */
	426	if(quit_program) break;
	427
9e7dbe44	428	num_rdy = poll(pollfd, fd_pairs->num_pairs, -1);
	429	if(num_rdy == -1) {
	430	perror("Poll error");
af4061b2	431	goto free_fd;
9e7dbe44	432	}
	433
	434	printf("Data received\n");
	435
	436	for(i=0;i<fd_pairs->num_pairs;i++) {
	437	switch(pollfd[i].revents) {
	438	case POLLERR:
	439	printf("Error returned in polling fd %d.\n", pollfd[i].fd);
af4061b2	440	num_hup++;
9e7dbe44	441	break;
	442	case POLLHUP:
	443	printf("Polling fd %d tells it has hung up.\n", pollfd[i].fd);
af4061b2	444	num_hup++;
9e7dbe44	445	break;
	446	case POLLNVAL:
	447	printf("Polling fd %d tells fd is not open.\n", pollfd[i].fd);
af4061b2	448	num_hup++;
9e7dbe44	449	break;
9e7dbe44	450	case POLLPRI:
af4061b2	451	printf("Urgent read on fd %d\n", pollfd[i].fd);
9e7dbe44	452	/* Take care of high priority channels first. */
68047576	453	high_prio = 1;
af4061b2	454	ret \|= read_subbuffer(&fd_pairs->pair[i]);
9e7dbe44	455	break;
68047576	456	}
9e7dbe44	457	}
af4061b2	458	/* If every FD has hung up, we end the read loop here */
af4061b2	459	if(num_hup == fd_pairs->num_pairs) break;
9e7dbe44	460
68047576	461	if(!high_prio) {
	462	for(i=0;i<fd_pairs->num_pairs;i++) {
	463	switch(pollfd[i].revents) {
	464	case POLLIN:
	465	/* Take care of low priority channels. */
af4061b2	466	printf("Normal read on fd %d\n", pollfd[i].fd);
af4061b2	467	ret \|= read_subbuffer(&fd_pairs->pair[i]);
68047576	468	break;
	469	}
	470	}
9e7dbe44	471	}
	472
	473	}
	474
af4061b2	475	free_fd:
9e7dbe44	476	free(pollfd);
9e7dbe44	477
af4061b2	478	/* munmap only the successfully mmapped indexes */
	479	i = fd_pairs->num_pairs;
	480	munmap:
	481	/* Munmap each FD */
	482	for(j=0;j<i;j++) {
	483	struct fd_pair *pair = &fd_pairs->pair[j];
	484	int err_ret;
	485
	486	err_ret = munmap(pair->mmap, pair->subbuf_size * pair->n_subbufs);
	487	if(err_ret != 0) {
	488	perror("Error in munmap");
	489	}
	490	ret \|= err_ret;
	491	}
	492
	493	end:
	494	return ret;
6cecb9cf	495	}
	496
	497
	498	void close_channel_trace_pairs(struct channel_trace_fd *fd_pairs)
	499	{
9e7dbe44	500	int i;
9e7dbe44	501	int ret;
6cecb9cf	502
9e7dbe44	503	for(i=0;i<fd_pairs->num_pairs;i++) {
	504	ret = close(fd_pairs->pair[i].channel);
	505	if(ret == -1) perror("Close error on channel");
	506	ret = close(fd_pairs->pair[i].trace);
	507	if(ret == -1) perror("Close error on trace");
	508	}
	509	free(fd_pairs->pair);
6cecb9cf	510	}
	511
	512	int main(int argc, char ** argv)
	513	{
	514	int ret;
6cecb9cf	515	struct channel_trace_fd fd_pairs = { NULL, 0 };
af4061b2	516	struct sigaction act;
6cecb9cf	517
	518	ret = parse_arguments(argc, argv);
	519
	520	if(ret != 0) show_arguments();
	521	if(ret < 0) return EINVAL;
	522	if(ret > 0) return 0;
	523
	524	show_info();
	525
	526	if(daemon_mode) {
4d20e474	527	ret = daemon(0, 0);
	528
	529	if(ret == -1) {
	530	perror("An error occured while daemonizing.");
	531	exit(-1);
	532	}
	533	}
6cecb9cf	534
af4061b2	535	/* Connect the signal handlers */
	536	act.sa_handler = handler;
	537	act.sa_flags = 0;
	538	sigemptyset(&(act.sa_mask));
	539	sigaddset(&(act.sa_mask), SIGTERM);
	540	sigaddset(&(act.sa_mask), SIGQUIT);
	541	sigaddset(&(act.sa_mask), SIGINT);
	542	sigaction(SIGTERM, &act, NULL);
	543	sigaction(SIGQUIT, &act, NULL);
	544	sigaction(SIGINT, &act, NULL);
	545
	546	//return 0;
6cecb9cf	547	if(ret = open_channel_trace_pairs(channel_name, trace_name, &fd_pairs))
9e7dbe44	548	goto close_channel;
6cecb9cf	549
	550	ret = read_channels(&fd_pairs);
	551
9e7dbe44	552	close_channel:
6cecb9cf	553	close_channel_trace_pairs(&fd_pairs);
6cecb9cf	554
6cecb9cf	555	return ret;
6cecb9cf	556	}