/*
 *   Copyright (c) International Business Machines  Corp., 2001
 *    06/20/2001 Robbie Williamson (robbiew@us.ibm.com)
 *    11/08/2001 Manoj Iyer (manjo@austin.ibm.com)
 *
 *   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.
 *
 *   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.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program;  if not, write to the Free Software Foundation,
 *   Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/*
 * Usage:	stress_cd [-n n] [-f file] [-m xx] [-d]
 *		where:
 *		  -n n     Number of threads to create
 *		  -f file  File or device to read from
 *		  -m xx    Number of MB to read from file
 *		  -b xx    Number of bytes to read from file
 *		  -d       Enable debugging messages
 */

#include <pthread.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
#include <errno.h>
#include <sys/types.h>
#include <unistd.h>
#include <string.h>

#define DEFAULT_NUM_THREADS	10
#define DEFAULT_NUM_BYTES	(1024*1024*100)	/* 100Mb */
#define DEFAULT_FILE		"/dev/cdrom"

static void sys_error(const char *, int);
static void parse_args(int, char **);
static void *thread(int *);
static int read_data(int, unsigned long *);

static int num_threads = DEFAULT_NUM_THREADS;
static int num_bytes = DEFAULT_NUM_BYTES;
static char *file = DEFAULT_FILE;
static unsigned long checksum;
static int debug;

int main(int argc, char **argv)
{
	pthread_attr_t attr;
	int rc = 0, i;

	/* Parse command line arguments and print out program header */
	parse_args(argc, argv);

	/* Read data from CDROM & compute checksum */
	read_data(0, &checksum);
	if (debug)
		printf("Thread [main] checksum: %-#12lx\n", checksum);

	if (pthread_attr_init(&attr))
		sys_error("pthread_attr_init failed", __LINE__);
	if (pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE))
		sys_error("pthread_attr_setdetachstate failed", __LINE__);

	printf("\tThread [main] Creating %d threads\n", num_threads);

	pthread_t array[num_threads];
	int arg[num_threads];

	for (i = 0; i < num_threads; i++) {
		if (debug)
			printf("\tThread [main]: creating thread %d\n", i + 1);
		arg[i] = i + 1;
		if (pthread_create((pthread_t *)&array[i], &attr,
				   (void *)thread, (void *)&arg[i])) {
			if (errno == EAGAIN) {
				fprintf(stderr,
					"\tThread [main]: unable to create "
					"thread %d\n", i);
			} else {
				sys_error("pthread_create failed", __LINE__);
			}
		}
		if (debug)
			printf("\tThread [main]: created thread %d\n", i + 1);
	}
	if (pthread_attr_destroy(&attr))
		sys_error("pthread_attr_destroy failed", __LINE__);

	for (i = 0; i < num_threads; i++) {
		void *exit_value;
		printf("\tThread [main]: waiting for thread: %d\n", i + 1);
		if (pthread_join(array[i], &exit_value))
			sys_error("pthread_join failed", __LINE__);

		if (debug)
			printf("\tThread [%d]: return %ld\n", i + 1,
			       (long)exit_value);
		rc += (long)exit_value;
	}

	if (rc != 0) {
		printf("test failed!\n");
		exit(-1);
	}

	printf("\tThread [main] All threads completed successfully...\n");
	exit(0);
}

static void *thread(int *parm)
{
	int num = *parm;
	unsigned long cksum = 0;

	if (debug)
		printf("\tThread [%d]: begin\n", num);

	read_data(num, &cksum);
	if (checksum != cksum) {
		fprintf(stderr, "\tThread [%d]: checksum mismatch!\n", num);
		pthread_exit((void *)-1);
	}

	if (debug)
		printf("\tThread [%d]: done\n", num);

	pthread_exit(NULL);
}

static int read_data(int num, unsigned long *cksum)
{
	int fd;
	const int bufSize = 1024;
	char *buffer;
	int bytes_read = 0;
	int n;
	char *p;

	if (debug)
		printf("\tThread [%d]: read_data()\n", num);

	if ((fd = open(file, O_RDONLY, NULL)) < 0)
		sys_error("open failed /dev/cdrom", __LINE__);

	buffer = malloc(sizeof(char) * bufSize);
	assert(buffer);

	lseek(fd, 1024 * 36, SEEK_SET);
	while (bytes_read < num_bytes) {
		n = read(fd, buffer, bufSize);
		if (n < 0)
			sys_error("read failed", __LINE__);
		else if (n == 0)
			sys_error("End of file", __LINE__);
		bytes_read += n;

		for (p = buffer; p < buffer + n; p++)
			*cksum += *p;

		if (debug)
			printf("\tThread [%d] bytes read: %5d checksum: "
			       "%-#12lx\n", num, bytes_read, *cksum);
	}
	free(buffer);

	if (debug)
		printf("\tThread [%d] bytes read: %5d checksum: %-#12lx\n",
		       num, bytes_read, *cksum);

	if (close(fd) < 0)
		sys_error("close failed", __LINE__);

	if (debug)
		printf("\tThread [%d]: done\n", num);

	return (0);
}

static void parse_args(int argc, char **argv)
{
	int i;
	int errflag = 0;
	char *program_name = *argv;
	extern char *optarg;	/* Command line option */

	while ((i = getopt(argc, argv, "df:n:b:m:?")) != EOF) {
		switch (i) {
		case 'd':	/* debug option */
			debug++;
			break;
		case 'f':	/* file to read from */
			file = optarg;
			break;
		case 'm':	/* num MB to read */
			num_bytes = atoi(optarg) * 1024 * 1024;
			break;
		case 'b':	/* num bytes to read */
			num_bytes = atoi(optarg);
			break;
		case 'n':	/* number of threads */
			num_threads = atoi(optarg);
			break;
		case '?':	/* help */
			errflag++;
			break;
		}
	}
	if (num_bytes < 0) {
		errflag++;
		fprintf(stderr, "ERROR: num_bytes must be greater than 0");
	}
	if (num_threads < 0) {
		errflag++;
		fprintf(stderr, "ERROR: num_threads must be greater than 0");
	}

	if (errflag) {
		fprintf(stderr, "\nUsage: %s"
			" [-n xx] [-m|b xx] [-d]\n\n"
			"\t-n xx    Number of threads to create (up to %d)\n"
			"\t-f file  File to read from\n"
			"\t-m xx    Number of MB to read\n"
			"\t-b xx    Number of bytes to read\n"
			"\t-d       Debug option\n", program_name,
			DEFAULT_NUM_THREADS);
		exit(2);
	}
}

static void sys_error(const char *msg, int line)
{
	fprintf(stderr, "ERROR [%d: %s: %s]\n", line, msg, strerror(errno));
	exit(-1);
}