245 lines
6.1 KiB
C
245 lines
6.1 KiB
C
/*
|
|
* 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);
|
|
}
|