|  | // SPDX-License-Identifier: GPL-2.0 | 
|  | /* | 
|  | * Copyright (c) 2000 Silicon Graphics, Inc. | 
|  | * All Rights Reserved. | 
|  | */ | 
|  | /* | 
|  | * doio -	a general purpose io initiator with system call and | 
|  | *		write logging.  See doio.h for the structure which defines | 
|  | *		what doio requests should look like. | 
|  | * | 
|  | * programming | 
|  | * notes: | 
|  | * ----------- | 
|  | *	messages should generally be printed using doio_fprintf(). | 
|  | * | 
|  | */ | 
|  |  | 
|  | #include "global.h" | 
|  |  | 
|  | #include <stdarg.h> | 
|  | #include <sys/uio.h>	/* for struct iovec (readv)*/ | 
|  | #include <sys/mman.h>	/* for mmap(2) */ | 
|  | #include <sys/ipc.h>	/* for i/o buffer in shared memory */ | 
|  | #include <sys/shm.h>	/* for i/o buffer in shared memory */ | 
|  | #include <sys/wait.h> | 
|  | #include <sys/time.h>	/* for delays */ | 
|  | #include <ctype.h> | 
|  |  | 
|  | struct io_req; | 
|  | int do_xfsctl(struct io_req *); | 
|  |  | 
|  | #include "doio.h" | 
|  | #include "pattern.h" | 
|  | #include "write_log.h" | 
|  | #include "random_range.h" | 
|  | #include "string_to_tokens.h" | 
|  |  | 
|  | #ifndef O_SSD | 
|  | #define O_SSD 0                /* so code compiles on a CRAY2 */ | 
|  | #endif | 
|  |  | 
|  | #define UINT64_T unsigned long long | 
|  |  | 
|  | #ifndef O_PARALLEL | 
|  | #define O_PARALLEL 0	/* so O_PARALLEL may be used in expressions */ | 
|  | #endif | 
|  |  | 
|  | #define PPID_CHECK_INTERVAL 5		/* check ppid every <-- iterations */ | 
|  | #define	MAX_AIO		256		/* maximum number of async I/O ops */ | 
|  | #define	MPP_BUMP	0 | 
|  |  | 
|  |  | 
|  | #define	SYSERR strerror(errno) | 
|  |  | 
|  | /* | 
|  | * getopt() string of supported cmdline arguments. | 
|  | */ | 
|  |  | 
|  | #define OPTS	"aC:d:ehm:n:kr:w:vU:V:M:N:" | 
|  |  | 
|  | #define DEF_RELEASE_INTERVAL	0 | 
|  |  | 
|  | /* | 
|  | * Flags set in parse_cmdline() to indicate which options were selected | 
|  | * on the cmdline. | 
|  | */ | 
|  |  | 
|  | int 	a_opt = 0;  	    /* abort on data compare errors 	*/ | 
|  | int	e_opt = 0;	    /* exec() after fork()'ing	        */ | 
|  | int	C_opt = 0;	    /* Data Check Type			*/ | 
|  | int	d_opt = 0;	    /* delay between operations		*/ | 
|  | int 	k_opt = 0;  	    /* lock file regions during writes	*/ | 
|  | int	m_opt = 0;	    /* generate periodic messages	*/ | 
|  | int 	n_opt = 0;  	    /* nprocs	    	    	    	*/ | 
|  | int 	r_opt = 0;  	    /* resource release interval    	*/ | 
|  | int 	w_opt = 0;  	    /* file write log file  	    	*/ | 
|  | int 	v_opt = 0;  	    /* verify writes if set 	    	*/ | 
|  | int 	U_opt = 0;  	    /* upanic() on varios conditions	*/ | 
|  | int	V_opt = 0;	    /* over-ride default validation fd type */ | 
|  | int	M_opt = 0;	    /* data buffer allocation types     */ | 
|  | char	TagName[40];	    /* name of this doio (see Monster)  */ | 
|  |  | 
|  |  | 
|  | /* | 
|  | * Misc globals initialized in parse_cmdline() | 
|  | */ | 
|  |  | 
|  | char	*Prog = NULL;	    /* set up in parse_cmdline()		*/ | 
|  | int 	Upanic_Conditions;  /* set by args to -U    	    		*/ | 
|  | int 	Release_Interval;   /* arg to -r    	    	    		*/ | 
|  | int 	Nprocs;	    	    /* arg to -n    	    	    		*/ | 
|  | char	*Write_Log; 	    /* arg to -w    	    	    		*/ | 
|  | char	*Infile;    	    /* input file (defaults to stdin)		*/ | 
|  | int	*Children;	    /* pids of child procs			*/ | 
|  | int	Nchildren = 0; | 
|  | int	Nsiblings = 0;	    /* tfork'ed siblings			*/ | 
|  | int	Execd = 0; | 
|  | int	Message_Interval = 0; | 
|  | int	Npes = 0;	    /* non-zero if built as an mpp multi-pe app */ | 
|  | int	Vpe = -1;	    /* Virtual pe number if Npes >= 0           */ | 
|  | int	Reqno = 1;	    /* request # - used in some error messages  */ | 
|  | int	Reqskipcnt = 0;	    /* count of I/O requests that are skipped   */ | 
|  | int	Validation_Flags; | 
|  | char	*(*Data_Check)();   /* function to call for data checking       */ | 
|  | int	(*Data_Fill)();     /* function to call for data filling        */ | 
|  | int	Nmemalloc = 0;	    /* number of memory allocation strategies   */ | 
|  | int	delayop = 0;	    /* delay between operations - type of delay */ | 
|  | int	delaytime = 0;	    /* delay between operations - how long      */ | 
|  |  | 
|  | struct wlog_file	Wlog; | 
|  |  | 
|  | int	active_mmap_rw = 0; /* Indicates that mmapped I/O is occurring. */ | 
|  | /* Used by sigbus_action() in the child doio. */ | 
|  | int	havesigint = 0; | 
|  |  | 
|  | #define SKIP_REQ	-2	/* skip I/O request */ | 
|  |  | 
|  | #define	NMEMALLOC	32 | 
|  | #define	MEM_DATA	1	/* data space 				*/ | 
|  | #define	MEM_SHMEM	2	/* System V shared memory 		*/ | 
|  | #define	MEM_T3ESHMEM	3	/* T3E Shared Memory 			*/ | 
|  | #define	MEM_MMAP	4	/* mmap(2) 				*/ | 
|  |  | 
|  | #define	MEMF_PRIVATE	0001 | 
|  | #define	MEMF_AUTORESRV	0002 | 
|  | #define	MEMF_LOCAL	0004 | 
|  | #define	MEMF_SHARED	0010 | 
|  |  | 
|  | #define	MEMF_FIXADDR	0100 | 
|  | #define	MEMF_ADDR	0200 | 
|  | #define	MEMF_AUTOGROW	0400 | 
|  | #define	MEMF_FILE	01000	/* regular file -- unlink on close	*/ | 
|  | #define MEMF_MPIN	010000	/* use mpin(2) to lock pages in memory */ | 
|  |  | 
|  | struct memalloc { | 
|  | int	memtype; | 
|  | int	flags; | 
|  | int	nblks; | 
|  | char	*name; | 
|  | void	*space;		/* memory address of allocated space */ | 
|  | int	fd;		/* FD open for mmaping */ | 
|  | int	size; | 
|  | }	Memalloc[NMEMALLOC]; | 
|  |  | 
|  | /* | 
|  | * Global file descriptors | 
|  | */ | 
|  |  | 
|  | int 	Wfd_Append; 	    /* for appending to the write-log	    */ | 
|  | int 	Wfd_Random; 	    /* for overlaying write-log entries	    */ | 
|  |  | 
|  | /* | 
|  | * Structure for maintaining open file test descriptors.  Used by | 
|  | * alloc_fd(). | 
|  | */ | 
|  |  | 
|  | struct fd_cache { | 
|  | char    c_file[MAX_FNAME_LENGTH+1]; | 
|  | int	c_oflags; | 
|  | int	c_fd; | 
|  | long    c_rtc; | 
|  | int	c_memalign;	/* from xfsctl(XFS_IOC_DIOINFO) */ | 
|  | int	c_miniosz; | 
|  | int	c_maxiosz; | 
|  | void	*c_memaddr;	/* mmapped address */ | 
|  | int	c_memlen;	/* length of above region */ | 
|  | }; | 
|  |  | 
|  | #define FD_ALLOC_INCR	32      /* allocate this many fd_map structs	*/ | 
|  | /* at a time */ | 
|  |  | 
|  | /* | 
|  | * Globals for tracking Sds and Core usage | 
|  | */ | 
|  |  | 
|  | char	*Memptr;		/* ptr to core buffer space	    	*/ | 
|  | int 	Memsize;		/* # bytes pointed to by Memptr 	*/ | 
|  | /* maintained by alloc_mem()    	*/ | 
|  |  | 
|  | int 	Sdsptr;			/* sds offset (always 0)	    	*/ | 
|  | int 	Sdssize;		/* # bytes of allocated sds space	*/ | 
|  | /* Maintained by alloc_sds()    	*/ | 
|  | char	Host[16]; | 
|  | char	Pattern[128]; | 
|  | int	Pattern_Length; | 
|  |  | 
|  | /* | 
|  | * Signal handlers, and related globals | 
|  | */ | 
|  |  | 
|  | void	sigint_handler();	/* Catch SIGINT in parent doio, propagate | 
|  | * to children, does not die. */ | 
|  |  | 
|  | void	die_handler();		/* Bad sig in child doios, exit 1. */ | 
|  | void	cleanup_handler();	/* Normal kill, exit 0. */ | 
|  |  | 
|  | void	sigbus_handler();	/* Handle sigbus--check active_mmap_rw to | 
|  | decide if this should be a normal exit. */ | 
|  |  | 
|  | void	cb_handler();		/* Posix aio callback handler. */ | 
|  | void	noop_handler();		/* Delayop alarm, does nothing. */ | 
|  | char	*hms(time_t  t); | 
|  | char	*format_rw(); | 
|  | char	*format_sds(); | 
|  | char	*format_listio(); | 
|  | char	*check_file(char *file, int offset, int length, char *pattern, | 
|  | int pattern_length, int patshift, int fsa); | 
|  | int	doio_fprintf(FILE *stream, char *format, ...); | 
|  | void	doio_upanic(int mask); | 
|  | void	doio(); | 
|  | void	help(FILE *stream); | 
|  | void	doio_delay(); | 
|  | int     alloc_fd( char *, int ); | 
|  | int     alloc_mem( int ); | 
|  | int     do_read( struct io_req * ); | 
|  | int     do_write( struct io_req * ); | 
|  | int     do_rw( struct io_req * ); | 
|  | int     do_sync( struct io_req * ); | 
|  | int     usage( FILE * ); | 
|  | int     aio_unregister( int ); | 
|  | int     parse_cmdline( int, char **, char * ); | 
|  | int     lock_file_region( char *, int, int, int, int ); | 
|  | struct	fd_cache *alloc_fdcache(char *, int); | 
|  | int     aio_register( int, int, int ); | 
|  | #ifndef linux | 
|  | int aio_wait(int); | 
|  | #endif | 
|  |  | 
|  | /* | 
|  | * Upanic conditions, and a map from symbolics to values | 
|  | */ | 
|  |  | 
|  | #define U_CORRUPTION	0001	    /* upanic on data corruption    */ | 
|  | #define U_IOSW	    	0002	    /* upanic on bad iosw   	    */ | 
|  | #define U_RVAL	    	0004	    /* upanic on bad rval   	    */ | 
|  |  | 
|  | #define U_ALL	    	(U_CORRUPTION | U_IOSW | U_RVAL) | 
|  |  | 
|  | /* | 
|  | * Name-To-Value map | 
|  | * Used to map cmdline arguments to values | 
|  | */ | 
|  | struct smap { | 
|  | char    *string; | 
|  | int	value; | 
|  | }; | 
|  |  | 
|  | struct smap Upanic_Args[] = { | 
|  | { "corruption",	U_CORRUPTION	}, | 
|  | { "iosw",	U_IOSW		}, | 
|  | { "rval",	U_RVAL  	}, | 
|  | { "all",	U_ALL   	}, | 
|  | { NULL,         0               } | 
|  | }; | 
|  |  | 
|  | struct aio_info { | 
|  | int			busy; | 
|  | int			id; | 
|  | int			fd; | 
|  | int			strategy; | 
|  | volatile int		done; | 
|  | int			sig; | 
|  | int			signalled; | 
|  | struct sigaction	osa; | 
|  | }; | 
|  |  | 
|  | struct aio_info	Aio_Info[MAX_AIO]; | 
|  |  | 
|  | struct aio_info	*aio_slot(); | 
|  | int     aio_done( struct aio_info * ); | 
|  |  | 
|  | /* -C data-fill/check type */ | 
|  | #define	C_DEFAULT	1 | 
|  | struct smap checkmap[] = { | 
|  | { "default",	C_DEFAULT }, | 
|  | { NULL,		0 }, | 
|  | }; | 
|  |  | 
|  | /* -d option delay types */ | 
|  | #define	DELAY_SELECT	1 | 
|  | #define	DELAY_SLEEP	2 | 
|  | #define	DELAY_SGINAP	3 | 
|  | #define	DELAY_ALARM	4 | 
|  | #define	DELAY_ITIMER	5	/* POSIX timer				*/ | 
|  |  | 
|  | struct smap delaymap[] = { | 
|  | { "select",	DELAY_SELECT }, | 
|  | { "sleep",	DELAY_SLEEP }, | 
|  | { "alarm",	DELAY_ALARM }, | 
|  | { NULL,	0 }, | 
|  | }; | 
|  |  | 
|  | /****** | 
|  | * | 
|  | * strerror() does similar actions. | 
|  |  | 
|  | char * | 
|  | syserrno(int err) | 
|  | { | 
|  | static char sys_errno[10]; | 
|  | sprintf(sys_errno, "%d", errno); | 
|  | return(sys_errno); | 
|  | } | 
|  |  | 
|  | ******/ | 
|  |  | 
|  | int | 
|  | main(argc, argv) | 
|  | int 	argc; | 
|  | char	**argv; | 
|  | { | 
|  | int	    	    	i, pid, stat, ex_stat; | 
|  | struct sigaction	sa; | 
|  | sigset_t		block_mask, old_mask; | 
|  | umask(0);		/* force new file modes to known values */ | 
|  |  | 
|  | TagName[0] = '\0'; | 
|  | parse_cmdline(argc, argv, OPTS); | 
|  |  | 
|  | random_range_seed(getpid());       /* initialize random number generator */ | 
|  |  | 
|  | /* | 
|  | * If this is a re-exec of doio, jump directly into the doio function. | 
|  | */ | 
|  |  | 
|  | if (Execd) { | 
|  | doio(); | 
|  | exit(E_SETUP); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Stop on all but a few signals... | 
|  | */ | 
|  | sigemptyset(&sa.sa_mask); | 
|  | sa.sa_handler = sigint_handler; | 
|  | sa.sa_flags = SA_RESETHAND;	/* sigint is ignored after the */ | 
|  | /* first time */ | 
|  | for (i = 1; i <= NSIG; i++) { | 
|  | switch(i) { | 
|  | #ifdef SIGRECOVERY | 
|  | case SIGRECOVERY: | 
|  | break; | 
|  | #endif | 
|  | #ifdef SIGCKPT | 
|  | case SIGCKPT: | 
|  | #endif | 
|  | #ifdef SIGRESTART | 
|  | case SIGRESTART: | 
|  | #endif | 
|  | case SIGTSTP: | 
|  | case SIGSTOP: | 
|  | case SIGCONT: | 
|  | case SIGCHLD: | 
|  | case SIGBUS: | 
|  | case SIGSEGV: | 
|  | case SIGQUIT: | 
|  | break; | 
|  | default: | 
|  | sigaction(i, &sa, NULL); | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * If we're logging write operations, make a dummy call to wlog_open | 
|  | * to initialize the write history file.  This call must be done in | 
|  | * the parent, to ensure that the history file exists and/or has | 
|  | * been truncated before any children attempt to open it, as the doio | 
|  | * children are not allowed to truncate the file. | 
|  | */ | 
|  |  | 
|  | if (w_opt) { | 
|  | strcpy(Wlog.w_file, Write_Log); | 
|  |  | 
|  | if (wlog_open(&Wlog, 1, 0666) < 0) { | 
|  | doio_fprintf(stderr, | 
|  | "Could not create/truncate write log %s\n", | 
|  | Write_Log); | 
|  | exit(2); | 
|  | } | 
|  |  | 
|  | wlog_close(&Wlog); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Malloc space for the children pid array.  Initialize all entries | 
|  | * to -1. | 
|  | */ | 
|  |  | 
|  | Children = (int *)malloc(sizeof(int) * Nprocs); | 
|  | for (i = 0; i < Nprocs; i++) { | 
|  | Children[i] = -1; | 
|  | } | 
|  |  | 
|  | sigemptyset(&block_mask); | 
|  | sigaddset(&block_mask, SIGCHLD); | 
|  | sigprocmask(SIG_BLOCK, &block_mask, &old_mask); | 
|  |  | 
|  | /* | 
|  | * Fork Nprocs.  This [parent] process is a watchdog, to notify the | 
|  | * invoker of procs which exit abnormally, and to make sure that all | 
|  | * child procs get cleaned up.  If the -e option was used, we will also | 
|  | * re-exec.  This is mostly for unicos/mk on mpp's, to ensure that not | 
|  | * all of the doio's don't end up in the same pe. | 
|  | * | 
|  | * Note - if Nprocs is 1, or this doio is a multi-pe app (Npes > 1), | 
|  | * jump directly to doio().  multi-pe apps can't fork(), and there is | 
|  | * no reason to fork() for 1 proc. | 
|  | */ | 
|  |  | 
|  | if (Nprocs == 1 || Npes > 1) { | 
|  | doio(); | 
|  | exit(0); | 
|  | } else { | 
|  | for (i = 0; i < Nprocs; i++) { | 
|  | if ((pid = fork()) == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "(parent) Could not fork %d children:  %s (%d)\n", | 
|  | i+1, SYSERR, errno); | 
|  | exit(E_SETUP); | 
|  | } | 
|  |  | 
|  | Children[Nchildren] = pid; | 
|  | Nchildren++; | 
|  |  | 
|  | if (pid == 0) { | 
|  | if (e_opt) { | 
|  | char *exec_path; | 
|  |  | 
|  | exec_path = argv[0]; | 
|  | argv[0] = (char *)malloc(strlen(exec_path + 1)); | 
|  | sprintf(argv[0], "-%s", exec_path); | 
|  |  | 
|  | execvp(exec_path, argv); | 
|  | doio_fprintf(stderr, | 
|  | "(parent) Could not execvp %s:  %s (%d)\n", | 
|  | exec_path, SYSERR, errno); | 
|  | exit(E_SETUP); | 
|  | } else { | 
|  | doio(); | 
|  | exit(E_SETUP); | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Parent spins on wait(), until all children exit. | 
|  | */ | 
|  |  | 
|  | ex_stat = E_NORMAL; | 
|  |  | 
|  | while (Nprocs) { | 
|  | if ((pid = wait(&stat)) == -1) { | 
|  | if (errno == EINTR) | 
|  | continue; | 
|  | } | 
|  |  | 
|  | for (i = 0; i < Nchildren; i++) | 
|  | if (Children[i] == pid) | 
|  | Children[i] = -1; | 
|  |  | 
|  | Nprocs--; | 
|  |  | 
|  | if (WIFEXITED(stat)) { | 
|  | switch (WEXITSTATUS(stat)) { | 
|  | case E_NORMAL: | 
|  | /* noop */ | 
|  | break; | 
|  |  | 
|  | case E_INTERNAL: | 
|  | doio_fprintf(stderr, | 
|  | "(parent) pid %d exited because of an internal error\n", | 
|  | pid); | 
|  | ex_stat |= E_INTERNAL; | 
|  | break; | 
|  |  | 
|  | case E_SETUP: | 
|  | doio_fprintf(stderr, | 
|  | "(parent) pid %d exited because of a setup error\n", | 
|  | pid); | 
|  | ex_stat |= E_SETUP; | 
|  | break; | 
|  |  | 
|  | case E_COMPARE: | 
|  | doio_fprintf(stderr, | 
|  | "(parent) pid %d exited because of data compare errors\n", | 
|  | pid); | 
|  |  | 
|  | ex_stat |= E_COMPARE; | 
|  |  | 
|  | if (a_opt) | 
|  | kill(0, SIGINT); | 
|  |  | 
|  | break; | 
|  |  | 
|  | case E_USAGE: | 
|  | doio_fprintf(stderr, | 
|  | "(parent) pid %d exited because of a usage error\n", | 
|  | pid); | 
|  |  | 
|  | ex_stat |= E_USAGE; | 
|  | break; | 
|  |  | 
|  | default: | 
|  | doio_fprintf(stderr, | 
|  | "(parent) pid %d exited with unknown status %d\n", | 
|  | pid, WEXITSTATUS(stat)); | 
|  | ex_stat |= E_INTERNAL; | 
|  | break; | 
|  | } | 
|  | } else if (WIFSIGNALED(stat) && WTERMSIG(stat) != SIGINT) { | 
|  | doio_fprintf(stderr, | 
|  | "(parent) pid %d terminated by signal %d\n", | 
|  | pid, WTERMSIG(stat)); | 
|  |  | 
|  | ex_stat |= E_SIGNAL; | 
|  | } | 
|  |  | 
|  | fflush(NULL); | 
|  | } | 
|  | } | 
|  |  | 
|  | exit(ex_stat); | 
|  |  | 
|  | }  /* main */ | 
|  |  | 
|  | /* | 
|  | * main doio function.  Each doio child starts here, and never returns. | 
|  | */ | 
|  |  | 
|  | void | 
|  | doio() | 
|  | { | 
|  | int	    	    	rval, i, infd, nbytes; | 
|  | char			*cp; | 
|  | struct io_req   	ioreq; | 
|  | struct sigaction	sa, def_action, ignore_action, exit_action; | 
|  | struct sigaction	sigbus_action; | 
|  |  | 
|  | Memsize = Sdssize = 0; | 
|  |  | 
|  | /* | 
|  | * Initialize the Pattern - write-type syscalls will replace Pattern[1] | 
|  | * with the pattern passed in the request.  Make sure that | 
|  | * strlen(Pattern) is not mod 16 so that out of order words will be | 
|  | * detected. | 
|  | */ | 
|  |  | 
|  | gethostname(Host, sizeof(Host)); | 
|  | if ((cp = strchr(Host, '.')) != NULL) | 
|  | *cp = '\0'; | 
|  |  | 
|  | Pattern_Length = sprintf(Pattern, "-:%d:%s:%s*", (int)getpid(), Host, Prog); | 
|  |  | 
|  | if (!(Pattern_Length % 16)) { | 
|  | Pattern_Length = sprintf(Pattern, "-:%d:%s:%s**", | 
|  | (int)getpid(), Host, Prog); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Open a couple of descriptors for the write-log file.  One descriptor | 
|  | * is for appending, one for random access.  Write logging is done for | 
|  | * file corruption detection.  The program doio_check is capable of | 
|  | * doing corruption detection based on a doio write-log. | 
|  | */ | 
|  |  | 
|  | if (w_opt) { | 
|  |  | 
|  | strcpy(Wlog.w_file, Write_Log); | 
|  |  | 
|  | if (wlog_open(&Wlog, 0, 0666) == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "Could not open write log file (%s): wlog_open() failed\n", | 
|  | Write_Log); | 
|  | exit(E_SETUP); | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Open the input stream - either a file or stdin | 
|  | */ | 
|  |  | 
|  | if (Infile == NULL) { | 
|  | infd = 0; | 
|  | } else { | 
|  | if ((infd = open(Infile, O_RDWR)) == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "Could not open input file (%s):  %s (%d)\n", | 
|  | Infile, SYSERR, errno); | 
|  | exit(E_SETUP); | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Define a set of signals that should never be masked.  Receipt of | 
|  | * these signals generally indicates a programming error, and we want | 
|  | * a corefile at the point of error.  We put SIGQUIT in this list so | 
|  | * that ^\ will force a user core dump. | 
|  | * | 
|  | * Note:  the handler for these should be SIG_DFL, all of them | 
|  | * produce a corefile as the default action. | 
|  | */ | 
|  |  | 
|  | ignore_action.sa_handler = SIG_IGN; | 
|  | ignore_action.sa_flags = 0; | 
|  | sigemptyset(&ignore_action.sa_mask); | 
|  |  | 
|  | def_action.sa_handler = SIG_DFL; | 
|  | def_action.sa_flags = 0; | 
|  | sigemptyset(&def_action.sa_mask); | 
|  |  | 
|  | exit_action.sa_handler = cleanup_handler; | 
|  | exit_action.sa_flags = 0; | 
|  | sigemptyset(&exit_action.sa_mask); | 
|  |  | 
|  | sa.sa_handler = die_handler; | 
|  | sa.sa_flags = 0; | 
|  | sigemptyset(&sa.sa_mask); | 
|  |  | 
|  | sigbus_action.sa_handler = sigbus_handler; | 
|  | sigbus_action.sa_flags = 0; | 
|  | sigemptyset(&sigbus_action.sa_mask); | 
|  |  | 
|  | for (i = 1; i <= NSIG; i++) { | 
|  | switch(i) { | 
|  | /* Signals to terminate program on */ | 
|  | case SIGINT: | 
|  | sigaction(i, &exit_action, NULL); | 
|  | break; | 
|  |  | 
|  | /* This depends on active_mmap_rw */ | 
|  | case SIGBUS: | 
|  | sigaction(i, &sigbus_action, NULL); | 
|  | break; | 
|  |  | 
|  | /* Signals to Ignore... */ | 
|  | case SIGSTOP: | 
|  | case SIGCONT: | 
|  | #ifdef SIGRECOVERY | 
|  | case SIGRECOVERY: | 
|  | #endif | 
|  | sigaction(i, &ignore_action, NULL); | 
|  | break; | 
|  |  | 
|  | /* Signals to trap & report & die */ | 
|  | /*case SIGTRAP:*/ | 
|  | /*case SIGABRT:*/ | 
|  | #ifdef SIGERR	/* cray only signals */ | 
|  | case SIGERR: | 
|  | case SIGBUFIO: | 
|  | case SIGINFO: | 
|  | #endif | 
|  | /*case SIGFPE:*/ | 
|  | case SIGURG: | 
|  | case SIGHUP: | 
|  | case SIGTERM: | 
|  | case SIGPIPE: | 
|  | case SIGIO: | 
|  | case SIGUSR1: | 
|  | case SIGUSR2: | 
|  | sigaction(i, &sa, NULL); | 
|  | break; | 
|  |  | 
|  |  | 
|  | /* Default Action for all other signals */ | 
|  | default: | 
|  | sigaction(i, &def_action, NULL); | 
|  | break; | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Main loop - each doio proc does this until the read returns eof (0). | 
|  | * Call the appropriate io function based on the request type. | 
|  | */ | 
|  |  | 
|  | while ((nbytes = read(infd, (char *)&ioreq, sizeof(ioreq)))) { | 
|  |  | 
|  | /* | 
|  | * Periodically check our ppid.  If it is 1, the child exits to | 
|  | * help clean up in the case that the main doio process was | 
|  | * killed. | 
|  | */ | 
|  |  | 
|  | if (Reqno && ((Reqno % PPID_CHECK_INTERVAL) == 0)) { | 
|  | if (getppid() == 1) { | 
|  | doio_fprintf(stderr, | 
|  | "Parent doio process has exited\n"); | 
|  | alloc_mem(-1); | 
|  | exit(E_SETUP); | 
|  | } | 
|  | } | 
|  |  | 
|  | if (nbytes == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "read of %d bytes from input failed:  %s (%d)\n", | 
|  | sizeof(ioreq), SYSERR, errno); | 
|  | alloc_mem(-1); | 
|  | exit(E_SETUP); | 
|  | } | 
|  |  | 
|  | if (nbytes != sizeof(ioreq)) { | 
|  | doio_fprintf(stderr, | 
|  | "read wrong # bytes from input stream, expected %d, got %d\n", | 
|  | sizeof(ioreq), nbytes); | 
|  | alloc_mem(-1); | 
|  | exit(E_SETUP); | 
|  | } | 
|  |  | 
|  | if (ioreq.r_magic != DOIO_MAGIC) { | 
|  | doio_fprintf(stderr, | 
|  | "got a bad magic # from input stream.  Expected 0%o, got 0%o\n", | 
|  | DOIO_MAGIC, ioreq.r_magic); | 
|  | alloc_mem(-1); | 
|  | exit(E_SETUP); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * If we're on a Release_Interval multiple, relase all ssd and | 
|  | * core space, and close all fd's in Fd_Map[]. | 
|  | */ | 
|  |  | 
|  | if (Reqno && Release_Interval && ! (Reqno%Release_Interval)) { | 
|  | if (Memsize) { | 
|  | #ifdef NOTDEF | 
|  | sbrk(-1 * Memsize); | 
|  | #else | 
|  | alloc_mem(-1); | 
|  | #endif | 
|  | } | 
|  | alloc_fd(NULL, 0); | 
|  | } | 
|  |  | 
|  | switch (ioreq.r_type) { | 
|  | case READ: | 
|  | case READA: | 
|  | rval = do_read(&ioreq); | 
|  | break; | 
|  |  | 
|  | case WRITE: | 
|  | case WRITEA: | 
|  | rval = do_write(&ioreq); | 
|  | break; | 
|  |  | 
|  | case READV: | 
|  | case AREAD: | 
|  | case PREAD: | 
|  | case LREAD: | 
|  | case LREADA: | 
|  | case LSREAD: | 
|  | case LSREADA: | 
|  | case WRITEV: | 
|  | case AWRITE: | 
|  | case PWRITE: | 
|  | case MMAPR: | 
|  | case MMAPW: | 
|  | case LWRITE: | 
|  | case LWRITEA: | 
|  | case LSWRITE: | 
|  | case LSWRITEA: | 
|  | case LEREAD: | 
|  | case LEREADA: | 
|  | case LEWRITE: | 
|  | case LEWRITEA: | 
|  | rval = do_rw(&ioreq); | 
|  | break; | 
|  | case RESVSP: | 
|  | case UNRESVSP: | 
|  | rval = do_xfsctl(&ioreq); | 
|  | break; | 
|  | case FSYNC2: | 
|  | case FDATASYNC: | 
|  | rval = do_sync(&ioreq); | 
|  | break; | 
|  | default: | 
|  | doio_fprintf(stderr, | 
|  | "Don't know how to handle io request type %d\n", | 
|  | ioreq.r_type); | 
|  | alloc_mem(-1); | 
|  | exit(E_SETUP); | 
|  | } | 
|  |  | 
|  | if (rval == SKIP_REQ){ | 
|  | Reqskipcnt++; | 
|  | } | 
|  | else if (rval != 0) { | 
|  | alloc_mem(-1); | 
|  | doio_fprintf(stderr, | 
|  | "doio(): operation %d returned != 0\n", | 
|  | ioreq.r_type); | 
|  | exit(E_SETUP); | 
|  | } | 
|  |  | 
|  | if (Message_Interval && Reqno % Message_Interval == 0) { | 
|  | doio_fprintf(stderr, "Info:  %d requests done (%d skipped) by this process\n", Reqno, Reqskipcnt); | 
|  | } | 
|  |  | 
|  | Reqno++; | 
|  |  | 
|  | if(delayop != 0) | 
|  | doio_delay(); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Child exits normally | 
|  | */ | 
|  | alloc_mem(-1); | 
|  | exit(E_NORMAL); | 
|  |  | 
|  | }  /* doio */ | 
|  |  | 
|  | void | 
|  | doio_delay() | 
|  | { | 
|  | struct timeval tv_delay; | 
|  | struct sigaction sa_al, sa_old; | 
|  | sigset_t al_mask; | 
|  |  | 
|  | switch(delayop) { | 
|  | case DELAY_SELECT: | 
|  | tv_delay.tv_sec = delaytime / 1000000; | 
|  | tv_delay.tv_usec = delaytime % 1000000; | 
|  | /*doio_fprintf(stdout, "delay_select: %d %d\n", | 
|  | tv_delay.tv_sec, tv_delay.tv_usec);*/ | 
|  | select(0, NULL, NULL, NULL, &tv_delay); | 
|  | break; | 
|  |  | 
|  | case DELAY_SLEEP: | 
|  | sleep(delaytime); | 
|  | break; | 
|  |  | 
|  | case DELAY_ALARM: | 
|  | sa_al.sa_flags = 0; | 
|  | sa_al.sa_handler = noop_handler; | 
|  | sigemptyset(&sa_al.sa_mask); | 
|  | sigaction(SIGALRM, &sa_al, &sa_old); | 
|  | sigemptyset(&al_mask); | 
|  | alarm(delaytime); | 
|  | sigsuspend(&al_mask); | 
|  | sigaction(SIGALRM, &sa_old, 0); | 
|  | break; | 
|  | } | 
|  | } | 
|  |  | 
|  |  | 
|  | /* | 
|  | * Format IO requests, returning a pointer to the formatted text. | 
|  | * | 
|  | * format_strat	- formats the async i/o completion strategy | 
|  | * format_rw	- formats a read[a]/write[a] request | 
|  | * format_sds	- formats a ssread/sswrite request | 
|  | * format_listio- formats a listio request | 
|  | * | 
|  | * ioreq is the doio io request structure. | 
|  | */ | 
|  |  | 
|  | struct smap sysnames[] = { | 
|  | { "READ",	READ		}, | 
|  | { "WRITE",	WRITE		}, | 
|  | { "READA",	READA		}, | 
|  | { "WRITEA",	WRITEA		}, | 
|  | { "SSREAD",	SSREAD		}, | 
|  | { "SSWRITE",	SSWRITE		}, | 
|  | { "LISTIO",  	LISTIO		}, | 
|  | { "LREAD",	LREAD		}, | 
|  | { "LREADA",	LREADA		}, | 
|  | { "LWRITE",	LWRITE		}, | 
|  | { "LWRITEA",	LWRITEA		}, | 
|  | { "LSREAD",	LSREAD		}, | 
|  | { "LSREADA",	LSREADA		}, | 
|  | { "LSWRITE",	LSWRITE		}, | 
|  | { "LSWRITEA",	LSWRITEA	}, | 
|  |  | 
|  | /* Irix System Calls */ | 
|  | { "PREAD",	PREAD		}, | 
|  | { "PWRITE",	PWRITE		}, | 
|  | { "AREAD",	AREAD		}, | 
|  | { "AWRITE",	AWRITE		}, | 
|  | { "LLREAD",	LLREAD		}, | 
|  | { "LLAREAD",	LLAREAD		}, | 
|  | { "LLWRITE",	LLWRITE		}, | 
|  | { "LLAWRITE",	LLAWRITE	}, | 
|  | { "RESVSP",	RESVSP		}, | 
|  | { "UNRESVSP",	UNRESVSP	}, | 
|  |  | 
|  | /* Irix and Linux System Calls */ | 
|  | { "READV",	READV		}, | 
|  | { "WRITEV",	WRITEV		}, | 
|  | { "MMAPR",	MMAPR		}, | 
|  | { "MMAPW",	MMAPW		}, | 
|  | { "FSYNC2",	FSYNC2		}, | 
|  | { "FDATASYNC",	FDATASYNC	}, | 
|  |  | 
|  | { "unknown",	-1		}, | 
|  | }; | 
|  |  | 
|  | struct smap aionames[] = { | 
|  | { "poll",	A_POLL		}, | 
|  | { "signal",	A_SIGNAL	}, | 
|  | { "recall",	A_RECALL	}, | 
|  | { "recalla",	A_RECALLA	}, | 
|  | { "recalls",	A_RECALLS	}, | 
|  | { "suspend",	A_SUSPEND	}, | 
|  | { "callback",	A_CALLBACK	}, | 
|  | { "synch",	0		}, | 
|  | { "unknown",	-1		}, | 
|  | }; | 
|  |  | 
|  | char * | 
|  | format_oflags(int oflags) | 
|  | { | 
|  | char flags[255]; | 
|  |  | 
|  |  | 
|  | flags[0]='\0'; | 
|  | switch(oflags & 03) { | 
|  | case O_RDONLY:		strcat(flags,"O_RDONLY,");	break; | 
|  | case O_WRONLY:		strcat(flags,"O_WRONLY,");	break; | 
|  | case O_RDWR:		strcat(flags,"O_RDWR,");	break; | 
|  | default:		strcat(flags,"O_weird");	break; | 
|  | } | 
|  |  | 
|  | if(oflags & O_EXCL) | 
|  | strcat(flags,"O_EXCL,"); | 
|  |  | 
|  | if(oflags & O_SYNC) | 
|  | strcat(flags,"O_SYNC,"); | 
|  |  | 
|  | if(oflags & O_DIRECT) | 
|  | strcat(flags,"O_DIRECT,"); | 
|  |  | 
|  | return(strdup(flags)); | 
|  | } | 
|  |  | 
|  | char * | 
|  | format_strat(int strategy) | 
|  | { | 
|  | char msg[64]; | 
|  | char *aio_strat; | 
|  |  | 
|  | switch (strategy) { | 
|  | case A_POLL:		aio_strat = "POLL";	break; | 
|  | case A_SIGNAL:		aio_strat = "SIGNAL";	break; | 
|  | case A_RECALL:		aio_strat = "RECALL";	break; | 
|  | case A_RECALLA:		aio_strat = "RECALLA";	break; | 
|  | case A_RECALLS:		aio_strat = "RECALLS";	break; | 
|  | case A_SUSPEND:		aio_strat = "SUSPEND";	break; | 
|  | case A_CALLBACK:	aio_strat = "CALLBACK";	break; | 
|  | case 0:			aio_strat = "<zero>";	break; | 
|  | default: | 
|  | sprintf(msg, "<error:%#o>", strategy); | 
|  | aio_strat = strdup(msg); | 
|  | break; | 
|  | } | 
|  |  | 
|  | return(aio_strat); | 
|  | } | 
|  |  | 
|  | char * | 
|  | format_rw( | 
|  | struct	io_req	*ioreq, | 
|  | int		fd, | 
|  | void		*buffer, | 
|  | int		signo, | 
|  | char		*pattern, | 
|  | void		*iosw | 
|  | ) | 
|  | { | 
|  | static char		*errbuf=NULL; | 
|  | char			*aio_strat, *cp; | 
|  | struct read_req		*readp = &ioreq->r_data.read; | 
|  | struct write_req	*writep = &ioreq->r_data.write; | 
|  | struct read_req		*readap = &ioreq->r_data.read; | 
|  | struct write_req	*writeap = &ioreq->r_data.write; | 
|  |  | 
|  | if(errbuf == NULL) | 
|  | errbuf = (char *)malloc(32768); | 
|  |  | 
|  | cp = errbuf; | 
|  | cp += sprintf(cp, "Request number %d\n", Reqno); | 
|  |  | 
|  | switch (ioreq->r_type) { | 
|  | case READ: | 
|  | cp += sprintf(cp, "syscall:  read(%d, %#lo, %d)\n", | 
|  | fd, (unsigned long) buffer, readp->r_nbytes); | 
|  | cp += sprintf(cp, "          fd %d is file %s - open flags are %#o\n", | 
|  | fd, readp->r_file, readp->r_oflags); | 
|  | cp += sprintf(cp, "          read done at file offset %d\n", | 
|  | readp->r_offset); | 
|  | break; | 
|  |  | 
|  | case WRITE: | 
|  | cp += sprintf(cp, "syscall:  write(%d, %#lo, %d)\n", | 
|  | fd, (unsigned long) buffer, writep->r_nbytes); | 
|  | cp += sprintf(cp, "          fd %d is file %s - open flags are %#o\n", | 
|  | fd, writep->r_file, writep->r_oflags); | 
|  | cp += sprintf(cp, "          write done at file offset %d - pattern is %s\n", | 
|  | writep->r_offset, pattern); | 
|  | break; | 
|  |  | 
|  | case READA: | 
|  | aio_strat = format_strat(readap->r_aio_strat); | 
|  |  | 
|  | cp += sprintf(cp, "syscall:  reada(%d, %#lo, %d, %#lo, %d)\n", | 
|  | fd, (unsigned long) buffer, readap->r_nbytes, | 
|  | (unsigned long) iosw, signo); | 
|  | cp += sprintf(cp, "          fd %d is file %s - open flags are %#o\n", | 
|  | fd, readap->r_file, readp->r_oflags); | 
|  | cp += sprintf(cp, "          reada done at file offset %d\n", | 
|  | readap->r_offset); | 
|  | cp += sprintf(cp, "          async io completion strategy is %s\n", | 
|  | aio_strat); | 
|  | break; | 
|  |  | 
|  | case WRITEA: | 
|  | aio_strat = format_strat(writeap->r_aio_strat); | 
|  |  | 
|  | cp += sprintf(cp, "syscall:  writea(%d, %#lo, %d, %#lo, %d)\n", | 
|  | fd, (unsigned long) buffer, writeap->r_nbytes, | 
|  | (unsigned long) iosw, signo); | 
|  | cp += sprintf(cp, "          fd %d is file %s - open flags are %#o\n", | 
|  | fd, writeap->r_file, writeap->r_oflags); | 
|  | cp += sprintf(cp, "          writea done at file offset %d - pattern is %s\n", | 
|  | writeap->r_offset, pattern); | 
|  | cp += sprintf(cp, "          async io completion strategy is %s\n", | 
|  | aio_strat); | 
|  | break; | 
|  |  | 
|  | } | 
|  |  | 
|  | return errbuf; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Perform the various sorts of disk reads | 
|  | */ | 
|  |  | 
|  | int | 
|  | do_read(req) | 
|  | struct io_req	*req; | 
|  | { | 
|  | int	    	    	fd, offset, nbytes, oflags, rval; | 
|  | char    	    	*addr, *file; | 
|  | struct fd_cache		*fdc; | 
|  |  | 
|  | /* | 
|  | * Initialize common fields - assumes r_oflags, r_file, r_offset, and | 
|  | * r_nbytes are at the same offset in the read_req and reada_req | 
|  | * structures. | 
|  | */ | 
|  |  | 
|  | file = req->r_data.read.r_file; | 
|  | oflags = req->r_data.read.r_oflags; | 
|  | offset = req->r_data.read.r_offset; | 
|  | nbytes = req->r_data.read.r_nbytes; | 
|  |  | 
|  | /*printf("read: %s, %#o, %d %d\n", file, oflags, offset, nbytes);*/ | 
|  |  | 
|  | /* | 
|  | * Grab an open file descriptor | 
|  | * Note: must be done before memory allocation so that the direct i/o | 
|  | *	information is available in mem. allocate | 
|  | */ | 
|  |  | 
|  | if ((fd = alloc_fd(file, oflags)) == -1) | 
|  | return -1; | 
|  |  | 
|  | /* | 
|  | * Allocate core or sds - based on the O_SSD flag | 
|  | */ | 
|  |  | 
|  | #ifndef wtob | 
|  | #define wtob(x)	(x * sizeof(UINT64_T)) | 
|  | #endif | 
|  |  | 
|  | /* get memory alignment for using DIRECT I/O */ | 
|  | fdc = alloc_fdcache(file, oflags); | 
|  |  | 
|  | if ((rval = alloc_mem(nbytes + wtob(1) * 2 + fdc->c_memalign)) < 0) { | 
|  | return rval; | 
|  | } | 
|  |  | 
|  | addr = Memptr; | 
|  |  | 
|  |  | 
|  | if( (req->r_data.read.r_uflags & F_WORD_ALIGNED) ) { | 
|  | /* | 
|  | * Force memory alignment for Direct I/O | 
|  | */ | 
|  | if( (oflags & O_DIRECT) && ((long)addr % fdc->c_memalign != 0) ) { | 
|  | addr += fdc->c_memalign - ((long)addr % fdc->c_memalign); | 
|  | } | 
|  | } else { | 
|  | addr += random_range(0, wtob(1) - 1, 1, NULL); | 
|  | } | 
|  |  | 
|  | switch (req->r_type) { | 
|  | case READ: | 
|  | /* move to the desired file position. */ | 
|  | if (lseek(fd, offset, SEEK_SET) == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "lseek(%d, %d, SEEK_SET) failed:  %s (%d)\n", | 
|  | fd, offset, SYSERR, errno); | 
|  | return -1; | 
|  | } | 
|  |  | 
|  | if ((rval = read(fd, addr, nbytes)) == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "read() request failed:  %s (%d)\n%s\n", | 
|  | SYSERR, errno, | 
|  | format_rw(req, fd, addr, -1, NULL, NULL)); | 
|  | doio_upanic(U_RVAL); | 
|  | return -1; | 
|  | } else if (rval != nbytes) { | 
|  | doio_fprintf(stderr, | 
|  | "read() request returned wrong # of bytes - expected %d, got %d\n%s\n", | 
|  | nbytes, rval, | 
|  | format_rw(req, fd, addr, -1, NULL, NULL)); | 
|  | doio_upanic(U_RVAL); | 
|  | return -1; | 
|  | } | 
|  | break; | 
|  | } | 
|  |  | 
|  | return 0;		/* if we get here, everything went ok */ | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Perform the verious types of disk writes. | 
|  | */ | 
|  |  | 
|  | int | 
|  | do_write(req) | 
|  | struct io_req	*req; | 
|  | { | 
|  | static int		pid = -1; | 
|  | int	    	    	fd, nbytes, oflags; | 
|  | /* REFERENCED */ | 
|  | int	    	    	logged_write, rval, got_lock; | 
|  | long    	    	offset, woffset = 0; | 
|  | char    	    	*addr, pattern, *file, *msg; | 
|  | struct wlog_rec		wrec; | 
|  | struct fd_cache		*fdc; | 
|  |  | 
|  | /* | 
|  | * Misc variable setup | 
|  | */ | 
|  |  | 
|  | nbytes	= req->r_data.write.r_nbytes; | 
|  | offset	= req->r_data.write.r_offset; | 
|  | pattern	= req->r_data.write.r_pattern; | 
|  | file	= req->r_data.write.r_file; | 
|  | oflags	= req->r_data.write.r_oflags; | 
|  |  | 
|  | /*printf("pwrite: %s, %#o, %d %d\n", file, oflags, offset, nbytes);*/ | 
|  |  | 
|  | /* | 
|  | * Allocate core memory and possibly sds space.  Initialize the data | 
|  | * to be written. | 
|  | */ | 
|  |  | 
|  | Pattern[0] = pattern; | 
|  |  | 
|  |  | 
|  | /* | 
|  | * Get a descriptor to do the io on | 
|  | */ | 
|  |  | 
|  | if ((fd = alloc_fd(file, oflags)) == -1) | 
|  | return -1; | 
|  |  | 
|  | /*printf("write: %d, %s, %#o, %d %d\n", | 
|  | fd, file, oflags, offset, nbytes);*/ | 
|  |  | 
|  | /* | 
|  | * Allocate SDS space for backdoor write if desired | 
|  | */ | 
|  |  | 
|  | /* get memory alignment for using DIRECT I/O */ | 
|  | fdc = alloc_fdcache(file, oflags); | 
|  |  | 
|  | if ((rval = alloc_mem(nbytes + wtob(1) * 2 + fdc->c_memalign)) < 0) { | 
|  | return rval; | 
|  | } | 
|  |  | 
|  | addr = Memptr; | 
|  |  | 
|  | if( (req->r_data.write.r_uflags & F_WORD_ALIGNED) ) { | 
|  | /* | 
|  | * Force memory alignment for Direct I/O | 
|  | */ | 
|  | if( (oflags & O_DIRECT) && ((long)addr % fdc->c_memalign != 0) ) { | 
|  | addr += fdc->c_memalign - ((long)addr % fdc->c_memalign); | 
|  | } | 
|  | } else { | 
|  | addr += random_range(0, wtob(1) - 1, 1, NULL); | 
|  | } | 
|  |  | 
|  | (*Data_Fill)(Memptr, nbytes, Pattern, Pattern_Length, 0); | 
|  | if( addr != Memptr ) | 
|  | memmove( addr, Memptr, nbytes); | 
|  |  | 
|  | rval = -1; | 
|  | got_lock = 0; | 
|  | logged_write = 0; | 
|  |  | 
|  | if (k_opt) { | 
|  | if (lock_file_region(file, fd, F_WRLCK, offset, nbytes) < 0) { | 
|  | alloc_mem(-1); | 
|  | exit(E_INTERNAL); | 
|  | } | 
|  |  | 
|  | got_lock = 1; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Write a preliminary write-log entry.  This is done so that | 
|  | * doio_check can do corruption detection across an interrupt/crash. | 
|  | * Note that w_done is set to 0.  If doio_check sees this, it | 
|  | * re-creates the file extents as if the write completed, but does not | 
|  | * do any checking - see comments in doio_check for more details. | 
|  | */ | 
|  |  | 
|  | if (w_opt) { | 
|  | if (pid == -1) { | 
|  | pid = getpid(); | 
|  | } | 
|  | wrec.w_async = (req->r_type == WRITEA) ? 1 : 0; | 
|  | wrec.w_oflags = oflags; | 
|  | wrec.w_pid = pid; | 
|  | wrec.w_offset = offset; | 
|  | wrec.w_nbytes = nbytes; | 
|  |  | 
|  | wrec.w_pathlen = strlen(file); | 
|  | memcpy(wrec.w_path, file, wrec.w_pathlen); | 
|  | wrec.w_hostlen = strlen(Host); | 
|  | memcpy(wrec.w_host, Host, wrec.w_hostlen); | 
|  | wrec.w_patternlen = Pattern_Length; | 
|  | memcpy(wrec.w_pattern, Pattern, wrec.w_patternlen); | 
|  |  | 
|  | wrec.w_done = 0; | 
|  |  | 
|  | if ((woffset = wlog_record_write(&Wlog, &wrec, -1)) == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "Could not append to write-log:  %s (%d)\n", | 
|  | SYSERR, errno); | 
|  | } else { | 
|  | logged_write = 1; | 
|  | } | 
|  | } | 
|  |  | 
|  | switch (req->r_type ) { | 
|  | case WRITE: | 
|  | /* | 
|  | * sync write | 
|  | */ | 
|  |  | 
|  | if (lseek(fd, offset, SEEK_SET) == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "lseek(%d, %d, SEEK_SET) failed:  %s (%d)\n", | 
|  | fd, offset, SYSERR, errno); | 
|  | return -1; | 
|  | } | 
|  |  | 
|  | rval = write(fd, addr, nbytes); | 
|  |  | 
|  | if (rval == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "write() failed:  %s (%d)\n%s\n", | 
|  | SYSERR, errno, | 
|  | format_rw(req, fd, addr, -1, Pattern, NULL)); | 
|  | doio_fprintf(stderr, | 
|  | "write() failed:  %s\n\twrite(%d, %#o, %d)\n\toffset %d, nbytes%%miniou(%d)=%d, oflags=%#o memalign=%d, addr%%memalign=%d\n", | 
|  | strerror(errno), | 
|  | fd, addr, nbytes, | 
|  | offset, | 
|  | fdc->c_miniosz, nbytes%fdc->c_miniosz, | 
|  | oflags, fdc->c_memalign, (long)addr%fdc->c_memalign); | 
|  | doio_upanic(U_RVAL); | 
|  | } else if (rval != nbytes) { | 
|  | doio_fprintf(stderr, | 
|  | "write() returned wrong # bytes - expected %d, got %d\n%s\n", | 
|  | nbytes, rval, | 
|  | format_rw(req, fd, addr, -1, Pattern, NULL)); | 
|  | doio_upanic(U_RVAL); | 
|  | rval = -1; | 
|  | } | 
|  |  | 
|  | break; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Verify that the data was written correctly - check_file() returns | 
|  | * a non-null pointer which contains an error message if there are | 
|  | * problems. | 
|  | */ | 
|  |  | 
|  | if (v_opt) { | 
|  | msg = check_file(file, offset, nbytes, Pattern, Pattern_Length, | 
|  | 0, oflags & O_PARALLEL); | 
|  | if (msg != NULL) { | 
|  | doio_fprintf(stderr, "%s%s\n", | 
|  | msg, | 
|  | format_rw(req, fd, addr, -1, Pattern, NULL) | 
|  | ); | 
|  | doio_upanic(U_CORRUPTION); | 
|  | exit(E_COMPARE); | 
|  |  | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * General cleanup ... | 
|  | * | 
|  | * Write extent information to the write-log, so that doio_check can do | 
|  | * corruption detection.  Note that w_done is set to 1, indicating that | 
|  | * the write has been verified as complete.  We don't need to write the | 
|  | * filename on the second logging. | 
|  | */ | 
|  |  | 
|  | if (w_opt && logged_write) { | 
|  | wrec.w_done = 1; | 
|  | wlog_record_write(&Wlog, &wrec, woffset); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Unlock file region if necessary | 
|  | */ | 
|  |  | 
|  | if (got_lock) { | 
|  | if (lock_file_region(file, fd, F_UNLCK, offset, nbytes) < 0) { | 
|  | alloc_mem(-1); | 
|  | exit(E_INTERNAL); | 
|  | } | 
|  | } | 
|  |  | 
|  | return( (rval == -1) ? -1 : 0); | 
|  | } | 
|  |  | 
|  |  | 
|  | /* | 
|  | * Simple routine to lock/unlock a file using fcntl() | 
|  | */ | 
|  |  | 
|  | int | 
|  | lock_file_region(fname, fd, type, start, nbytes) | 
|  | char	*fname; | 
|  | int	fd; | 
|  | int	type; | 
|  | int	start; | 
|  | int	nbytes; | 
|  | { | 
|  | struct flock	flk; | 
|  |  | 
|  | flk.l_type = type; | 
|  | flk.l_whence = 0; | 
|  | flk.l_start = start; | 
|  | flk.l_len = nbytes; | 
|  |  | 
|  | if (fcntl(fd, F_SETLKW, &flk) < 0) { | 
|  | doio_fprintf(stderr, | 
|  | "fcntl(%d, %d, %#o) failed for file %s, lock type %d, offset %d, length %d:  %s (%d), open flags: %#o\n", | 
|  | fd, F_SETLKW, &flk, fname, type, | 
|  | start, nbytes, SYSERR, errno, | 
|  | fcntl(fd, F_GETFL, 0)); | 
|  | return -1; | 
|  | } | 
|  |  | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | int | 
|  | do_listio(req) | 
|  | struct io_req	*req; | 
|  | { | 
|  | return -1; | 
|  | } | 
|  |  | 
|  | /* --------------------------------------------------------------------------- | 
|  | * | 
|  | * A new paradigm of doing the r/w system call where there is a "stub" | 
|  | * function that builds the info for the system call, then does the system | 
|  | * call; this is called by code that is common to all system calls and does | 
|  | * the syscall return checking, async I/O wait, iosw check, etc. | 
|  | * | 
|  | * Flags: | 
|  | *	WRITE, ASYNC, SSD/SDS, | 
|  | *	FILE_LOCK, WRITE_LOG, VERIFY_DATA, | 
|  | */ | 
|  |  | 
|  | struct	status { | 
|  | int	rval;		/* syscall return */ | 
|  | int	err;		/* errno */ | 
|  | int	*aioid;		/* list of async I/O structures */ | 
|  | }; | 
|  |  | 
|  | struct syscall_info { | 
|  | char		*sy_name; | 
|  | int		sy_type; | 
|  | struct status	*(*sy_syscall)(); | 
|  | int		(*sy_buffer)(); | 
|  | char		*(*sy_format)(); | 
|  | int		sy_flags; | 
|  | int		sy_bits; | 
|  | }; | 
|  |  | 
|  | #define	SY_WRITE		00001 | 
|  | #define	SY_ASYNC		00010 | 
|  | #define	SY_IOSW			00020 | 
|  | #define	SY_SDS			00100 | 
|  |  | 
|  | char * | 
|  | fmt_ioreq(struct io_req *ioreq, struct syscall_info *sy, int fd) | 
|  | { | 
|  | static char		*errbuf=NULL; | 
|  | char			*cp; | 
|  | struct rw_req		*io; | 
|  | struct smap		*aname; | 
|  |  | 
|  | if(errbuf == NULL) | 
|  | errbuf = (char *)malloc(32768); | 
|  |  | 
|  | io = &ioreq->r_data.io; | 
|  |  | 
|  | /* | 
|  | * Look up async I/O completion strategy | 
|  | */ | 
|  | for(aname=aionames; | 
|  | aname->value != -1 && aname->value != io->r_aio_strat; | 
|  | aname++) | 
|  | ; | 
|  |  | 
|  | cp = errbuf; | 
|  | cp += sprintf(cp, "Request number %d\n", Reqno); | 
|  |  | 
|  | cp += sprintf(cp, "          fd %d is file %s - open flags are %#o %s\n", | 
|  | fd, io->r_file, io->r_oflags, format_oflags(io->r_oflags)); | 
|  |  | 
|  | if(sy->sy_flags & SY_WRITE) { | 
|  | cp += sprintf(cp, "          write done at file offset %d - pattern is %c (%#o)\n", | 
|  | io->r_offset, | 
|  | (io->r_pattern == '\0') ? '?' : io->r_pattern, | 
|  | io->r_pattern); | 
|  | } else { | 
|  | cp += sprintf(cp, "          read done at file offset %d\n", | 
|  | io->r_offset); | 
|  | } | 
|  |  | 
|  | if(sy->sy_flags & SY_ASYNC) { | 
|  | cp += sprintf(cp, "          async io completion strategy is %s\n", | 
|  | aname->string); | 
|  | } | 
|  |  | 
|  | cp += sprintf(cp, "          number of requests is %d, strides per request is %d\n", | 
|  | io->r_nent, io->r_nstrides); | 
|  |  | 
|  | cp += sprintf(cp, "          i/o byte count = %d\n", | 
|  | io->r_nbytes); | 
|  |  | 
|  | cp += sprintf(cp, "          memory alignment is %s\n", | 
|  | (io->r_uflags & F_WORD_ALIGNED) ? "aligned" : "unaligned"); | 
|  |  | 
|  | if(io->r_oflags & O_DIRECT) { | 
|  | char		*dio_env; | 
|  | struct dioattr	finfo; | 
|  |  | 
|  | if(xfsctl(io->r_file, fd, XFS_IOC_DIOINFO, &finfo) == -1) { | 
|  | cp += sprintf(cp, "          Error %s (%d) getting direct I/O info\n", | 
|  | strerror(errno), errno); | 
|  | finfo.d_mem = 1; | 
|  | finfo.d_miniosz = 1; | 
|  | finfo.d_maxiosz = 1; | 
|  | } | 
|  |  | 
|  | dio_env = getenv("XFS_DIO_MIN"); | 
|  | if (dio_env) | 
|  | finfo.d_mem = finfo.d_miniosz = atoi(dio_env); | 
|  |  | 
|  | cp += sprintf(cp, "          DIRECT I/O: offset %% %d = %d length %% %d = %d\n", | 
|  | finfo.d_miniosz, | 
|  | io->r_offset % finfo.d_miniosz, | 
|  | io->r_nbytes, | 
|  | io->r_nbytes % finfo.d_miniosz); | 
|  | cp += sprintf(cp, "          mem alignment 0x%x xfer size: small: %d large: %d\n", | 
|  | finfo.d_mem, finfo.d_miniosz, finfo.d_maxiosz); | 
|  | } | 
|  | return(errbuf); | 
|  | } | 
|  |  | 
|  | struct status * | 
|  | sy_pread(req, sysc, fd, addr) | 
|  | struct io_req	*req; | 
|  | struct syscall_info *sysc; | 
|  | int fd; | 
|  | char *addr; | 
|  | { | 
|  | int rc; | 
|  | struct status	*status; | 
|  |  | 
|  | rc = pread(fd, addr, req->r_data.io.r_nbytes, | 
|  | req->r_data.io.r_offset); | 
|  |  | 
|  | status = (struct status *)malloc(sizeof(struct status)); | 
|  | if( status == NULL ){ | 
|  | doio_fprintf(stderr, "malloc failed, %s/%d\n", | 
|  | __FILE__, __LINE__); | 
|  | return NULL; | 
|  | } | 
|  | status->aioid = NULL; | 
|  | status->rval = rc; | 
|  | status->err = errno; | 
|  |  | 
|  | return(status); | 
|  | } | 
|  |  | 
|  | struct status * | 
|  | sy_pwrite(req, sysc, fd, addr) | 
|  | struct io_req	*req; | 
|  | struct syscall_info *sysc; | 
|  | int fd; | 
|  | char *addr; | 
|  | { | 
|  | int rc; | 
|  | struct status	*status; | 
|  |  | 
|  | rc = pwrite(fd, addr, req->r_data.io.r_nbytes, | 
|  | req->r_data.io.r_offset); | 
|  |  | 
|  | status = (struct status *)malloc(sizeof(struct status)); | 
|  | if( status == NULL ){ | 
|  | doio_fprintf(stderr, "malloc failed, %s/%d\n", | 
|  | __FILE__, __LINE__); | 
|  | return NULL; | 
|  | } | 
|  | status->aioid = NULL; | 
|  | status->rval = rc; | 
|  | status->err = errno; | 
|  |  | 
|  | return(status); | 
|  | } | 
|  |  | 
|  | char * | 
|  | fmt_pread(struct io_req *req, struct syscall_info *sy, int fd, char *addr) | 
|  | { | 
|  | static char	*errbuf = NULL; | 
|  | char		*cp; | 
|  |  | 
|  | if(errbuf == NULL){ | 
|  | errbuf = (char *)malloc(32768); | 
|  | if( errbuf == NULL ){ | 
|  | doio_fprintf(stderr, "malloc failed, %s/%d\n", | 
|  | __FILE__, __LINE__); | 
|  | return NULL; | 
|  | } | 
|  | } | 
|  |  | 
|  | cp = errbuf; | 
|  | cp += sprintf(cp, "syscall:  %s(%d, 0x%p, %d)\n", | 
|  | sy->sy_name, fd, addr, req->r_data.io.r_nbytes); | 
|  | return(errbuf); | 
|  | } | 
|  |  | 
|  | struct status * | 
|  | sy_rwv(req, sysc, fd, addr, rw) | 
|  | struct io_req	*req; | 
|  | struct syscall_info *sysc; | 
|  | int fd; | 
|  | char *addr; | 
|  | int rw; | 
|  | { | 
|  | int rc; | 
|  | struct status	*status; | 
|  | struct iovec	iov[2]; | 
|  |  | 
|  | status = (struct status *)malloc(sizeof(struct status)); | 
|  | if( status == NULL ){ | 
|  | doio_fprintf(stderr, "malloc failed, %s/%d\n", | 
|  | __FILE__, __LINE__); | 
|  | return NULL; | 
|  | } | 
|  | status->aioid = NULL; | 
|  |  | 
|  | /* move to the desired file position. */ | 
|  | if ((rc=lseek(fd, req->r_data.io.r_offset, SEEK_SET)) == -1) { | 
|  | status->rval = rc; | 
|  | status->err = errno; | 
|  | return(status); | 
|  | } | 
|  |  | 
|  | iov[0].iov_base = addr; | 
|  | iov[0].iov_len = req->r_data.io.r_nbytes; | 
|  |  | 
|  | if(rw) | 
|  | rc = writev(fd, iov, 1); | 
|  | else | 
|  | rc = readv(fd, iov, 1); | 
|  | status->aioid = NULL; | 
|  | status->rval = rc; | 
|  | status->err = errno; | 
|  | return(status); | 
|  | } | 
|  |  | 
|  | struct status * | 
|  | sy_readv(req, sysc, fd, addr) | 
|  | struct io_req	*req; | 
|  | struct syscall_info *sysc; | 
|  | int fd; | 
|  | char *addr; | 
|  | { | 
|  | return sy_rwv(req, sysc, fd, addr, 0); | 
|  | } | 
|  |  | 
|  | struct status * | 
|  | sy_writev(req, sysc, fd, addr) | 
|  | struct io_req	*req; | 
|  | struct syscall_info *sysc; | 
|  | int fd; | 
|  | char *addr; | 
|  | { | 
|  | return sy_rwv(req, sysc, fd, addr, 1); | 
|  | } | 
|  |  | 
|  | char * | 
|  | fmt_readv(struct io_req *req, struct syscall_info *sy, int fd, char *addr) | 
|  | { | 
|  | static char	errbuf[32768]; | 
|  | char		*cp; | 
|  |  | 
|  | cp = errbuf; | 
|  | cp += sprintf(cp, "syscall:  %s(%d, (iov on stack), 1)\n", | 
|  | sy->sy_name, fd); | 
|  | return(errbuf); | 
|  | } | 
|  |  | 
|  | struct status * | 
|  | sy_mmrw(req, sysc, fd, addr, rw) | 
|  | struct io_req *req; | 
|  | struct syscall_info *sysc; | 
|  | int fd; | 
|  | char *addr; | 
|  | int rw; | 
|  | { | 
|  | /* | 
|  | * mmap read/write | 
|  | * This version is oriented towards mmaping the file to memory | 
|  | * ONCE and keeping it mapped. | 
|  | */ | 
|  | struct status		*status; | 
|  | void			*mrc, *memaddr; | 
|  | struct fd_cache		*fdc; | 
|  | struct stat		sbuf; | 
|  |  | 
|  | status = (struct status *)malloc(sizeof(struct status)); | 
|  | if( status == NULL ){ | 
|  | doio_fprintf(stderr, "malloc failed, %s/%d\n", | 
|  | __FILE__, __LINE__); | 
|  | return NULL; | 
|  | } | 
|  | status->aioid = NULL; | 
|  | status->rval = -1; | 
|  |  | 
|  | fdc = alloc_fdcache(req->r_data.io.r_file, req->r_data.io.r_oflags); | 
|  |  | 
|  | if( fdc->c_memaddr == NULL ) { | 
|  | if( fstat(fd, &sbuf) < 0 ){ | 
|  | doio_fprintf(stderr, "fstat failed, errno=%d\n", | 
|  | errno); | 
|  | status->err = errno; | 
|  | return(status); | 
|  | } | 
|  |  | 
|  | fdc->c_memlen = (int)sbuf.st_size; | 
|  | mrc = mmap(NULL, (int)sbuf.st_size, | 
|  | rw ? PROT_WRITE|PROT_READ : PROT_READ, | 
|  | MAP_SHARED, fd, 0); | 
|  |  | 
|  | if( mrc == MAP_FAILED ) { | 
|  | doio_fprintf(stderr, "mmap() failed - 0x%lx %d\n", | 
|  | mrc, errno); | 
|  | status->err = errno; | 
|  | return(status); | 
|  | } | 
|  |  | 
|  | fdc->c_memaddr = mrc; | 
|  | } | 
|  |  | 
|  | memaddr = (void *)((char *)fdc->c_memaddr + req->r_data.io.r_offset); | 
|  |  | 
|  | active_mmap_rw = 1; | 
|  | if(rw) | 
|  | memcpy(memaddr, addr, req->r_data.io.r_nbytes); | 
|  | else | 
|  | memcpy(addr, memaddr, req->r_data.io.r_nbytes); | 
|  | active_mmap_rw = 0; | 
|  |  | 
|  | status->rval = req->r_data.io.r_nbytes; | 
|  | status->err = 0; | 
|  | return(status); | 
|  | } | 
|  |  | 
|  | struct status * | 
|  | sy_mmread(req, sysc, fd, addr) | 
|  | struct io_req *req; | 
|  | struct syscall_info *sysc; | 
|  | int fd; | 
|  | char *addr; | 
|  | { | 
|  | return sy_mmrw(req, sysc, fd, addr, 0); | 
|  | } | 
|  |  | 
|  | struct status * | 
|  | sy_mmwrite(req, sysc, fd, addr) | 
|  | struct io_req *req; | 
|  | struct syscall_info *sysc; | 
|  | int fd; | 
|  | char *addr; | 
|  | { | 
|  | return sy_mmrw(req, sysc, fd, addr, 1); | 
|  | } | 
|  |  | 
|  | char * | 
|  | fmt_mmrw(struct io_req *req, struct syscall_info *sy, int fd, char *addr) | 
|  | { | 
|  | static char	errbuf[32768]; | 
|  | char		*cp; | 
|  | struct fd_cache	*fdc; | 
|  | void		*memaddr; | 
|  |  | 
|  | fdc = alloc_fdcache(req->r_data.io.r_file, req->r_data.io.r_oflags); | 
|  |  | 
|  | cp = errbuf; | 
|  | cp += sprintf(cp, "syscall:  %s(NULL, %d, %s, MAP_SHARED, %d, 0)\n", | 
|  | sy->sy_name, | 
|  | fdc->c_memlen, | 
|  | (sy->sy_flags & SY_WRITE) ? "PROT_WRITE" : "PROT_READ", | 
|  | fd); | 
|  |  | 
|  | cp += sprintf(cp, "\tfile is mmaped to: 0x%lx\n", | 
|  | (unsigned long) fdc->c_memaddr); | 
|  |  | 
|  | memaddr = (void *)((char *)fdc->c_memaddr + req->r_data.io.r_offset); | 
|  |  | 
|  | cp += sprintf(cp, "\tfile-mem=0x%lx, length=%d, buffer=0x%lx\n", | 
|  | (unsigned long) memaddr, req->r_data.io.r_nbytes, | 
|  | (unsigned long) addr); | 
|  |  | 
|  | return(errbuf); | 
|  | } | 
|  |  | 
|  | struct syscall_info syscalls[] = { | 
|  | { "pread",			PREAD, | 
|  | sy_pread,	NULL,		fmt_pread, | 
|  | 0 | 
|  | }, | 
|  | { "pwrite",			PWRITE, | 
|  | sy_pwrite,	NULL,		fmt_pread, | 
|  | SY_WRITE | 
|  | }, | 
|  |  | 
|  | { "readv",			READV, | 
|  | sy_readv,	NULL,		fmt_readv, | 
|  | 0 | 
|  | }, | 
|  | { "writev",			WRITEV, | 
|  | sy_writev,	NULL,		fmt_readv, | 
|  | SY_WRITE | 
|  | }, | 
|  | { "mmap-read",			MMAPR, | 
|  | sy_mmread,	NULL,		fmt_mmrw, | 
|  | 0 | 
|  | }, | 
|  | { "mmap-write",			MMAPW, | 
|  | sy_mmwrite,	NULL,		fmt_mmrw, | 
|  | SY_WRITE | 
|  | }, | 
|  |  | 
|  | { NULL,				0, | 
|  | 0,		0,		0, | 
|  | 0 | 
|  | }, | 
|  | }; | 
|  |  | 
|  | int | 
|  | do_rw(req) | 
|  | struct io_req	*req; | 
|  | { | 
|  | static int		pid = -1; | 
|  | int	    		fd, offset, nbytes, nstrides, nents, oflags; | 
|  | int			rval, mem_needed, i; | 
|  | int	    		logged_write, got_lock, woffset = 0, pattern; | 
|  | int			min_byte, max_byte; | 
|  | char    		*addr, *file, *msg; | 
|  | struct status		*s; | 
|  | struct wlog_rec		wrec; | 
|  | struct syscall_info	*sy; | 
|  | struct fd_cache		*fdc; | 
|  |  | 
|  | /* | 
|  | * Initialize common fields - assumes r_oflags, r_file, r_offset, and | 
|  | * r_nbytes are at the same offset in the read_req and reada_req | 
|  | * structures. | 
|  | */ | 
|  | file	= req->r_data.io.r_file; | 
|  | oflags	= req->r_data.io.r_oflags; | 
|  | offset	= req->r_data.io.r_offset; | 
|  | nbytes	= req->r_data.io.r_nbytes; | 
|  | nstrides= req->r_data.io.r_nstrides; | 
|  | nents   = req->r_data.io.r_nent; | 
|  | pattern	= req->r_data.io.r_pattern; | 
|  |  | 
|  | if( nents >= MAX_AIO ) { | 
|  | doio_fprintf(stderr, "do_rw: too many list requests, %d.  Maximum is %d\n", | 
|  | nents, MAX_AIO); | 
|  | return(-1); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * look up system call info | 
|  | */ | 
|  | for(sy=syscalls; sy->sy_name != NULL && sy->sy_type != req->r_type; sy++) | 
|  | ; | 
|  |  | 
|  | if(sy->sy_name == NULL) { | 
|  | doio_fprintf(stderr, "do_rw: unknown r_type %d.\n", | 
|  | req->r_type); | 
|  | return(-1); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Get an open file descriptor | 
|  | * Note: must be done before memory allocation so that the direct i/o | 
|  | *	information is available in mem. allocate | 
|  | */ | 
|  |  | 
|  | if ((fd = alloc_fd(file, oflags)) == -1) | 
|  | return -1; | 
|  |  | 
|  | /* | 
|  | * Allocate core memory and possibly sds space.  Initialize the | 
|  | * data to be written.  Make sure we get enough, based on the | 
|  | * memstride. | 
|  | * | 
|  | * need: | 
|  | *	1 extra word for possible partial-word address "bump" | 
|  | *	1 extra word for dynamic pattern overrun | 
|  | *	MPP_BUMP extra words for T3E non-hw-aligned memory address. | 
|  | */ | 
|  |  | 
|  | if( sy->sy_buffer != NULL ) { | 
|  | mem_needed = (*sy->sy_buffer)(req, 0, 0, NULL, NULL); | 
|  | } else { | 
|  | mem_needed = nbytes; | 
|  | } | 
|  |  | 
|  | /* get memory alignment for using DIRECT I/O */ | 
|  | fdc = alloc_fdcache(file, oflags); | 
|  |  | 
|  | if ((rval = alloc_mem(mem_needed + wtob(1) * 2 + fdc->c_memalign)) < 0) { | 
|  | return rval; | 
|  | } | 
|  |  | 
|  | Pattern[0] = pattern; | 
|  |  | 
|  | /* | 
|  | * Allocate SDS space for backdoor write if desired | 
|  | */ | 
|  |  | 
|  | if (oflags & O_SSD) { | 
|  | doio_fprintf(stderr, "Invalid O_SSD flag was generated for non-Cray system\n"); | 
|  | fflush(stderr); | 
|  | return -1; | 
|  | } else { | 
|  | addr = Memptr; | 
|  |  | 
|  | /* | 
|  | * if io is not raw, bump the offset by a random amount | 
|  | * to generate non-word-aligned io. | 
|  | * | 
|  | * On MPP systems, raw I/O must start on an 0x80 byte boundary. | 
|  | * For non-aligned I/O, bump the address from 1 to 8 words. | 
|  | */ | 
|  |  | 
|  | if (! (req->r_data.io.r_uflags & F_WORD_ALIGNED)) { | 
|  | addr += random_range(0, wtob(1) - 1, 1, NULL); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Force memory alignment for Direct I/O | 
|  | */ | 
|  | if( (oflags & O_DIRECT) && ((long)addr % fdc->c_memalign != 0) ) { | 
|  | addr += fdc->c_memalign - ((long)addr % fdc->c_memalign); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * FILL must be done on a word-aligned buffer. | 
|  | * Call the fill function with Memptr which is aligned, | 
|  | * then memmove it to the right place. | 
|  | */ | 
|  | if (sy->sy_flags & SY_WRITE) { | 
|  | (*Data_Fill)(Memptr, mem_needed, Pattern, Pattern_Length, 0); | 
|  | if( addr != Memptr ) | 
|  | memmove( addr, Memptr, mem_needed); | 
|  | } | 
|  | } | 
|  |  | 
|  | rval = 0; | 
|  | got_lock = 0; | 
|  | logged_write = 0; | 
|  |  | 
|  | /* | 
|  | * Lock data if this is a write and locking option is set | 
|  | */ | 
|  | if (sy->sy_flags & SY_WRITE && k_opt) { | 
|  | if( sy->sy_buffer != NULL ) { | 
|  | (*sy->sy_buffer)(req, offset, 0, &min_byte, &max_byte); | 
|  | } else { | 
|  | min_byte = offset; | 
|  | max_byte = offset + (nbytes * nstrides * nents); | 
|  | } | 
|  |  | 
|  | if (lock_file_region(file, fd, F_WRLCK, | 
|  | min_byte, (max_byte-min_byte+1)) < 0) { | 
|  | doio_fprintf(stderr, | 
|  | "file lock failed:\n%s\n", | 
|  | fmt_ioreq(req, sy, fd)); | 
|  | doio_fprintf(stderr, | 
|  | "          buffer(req, %d, 0, 0x%x, 0x%x)\n", | 
|  | offset, min_byte, max_byte); | 
|  | alloc_mem(-1); | 
|  | exit(E_INTERNAL); | 
|  | } | 
|  |  | 
|  | got_lock = 1; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Write a preliminary write-log entry.  This is done so that | 
|  | * doio_check can do corruption detection across an interrupt/crash. | 
|  | * Note that w_done is set to 0.  If doio_check sees this, it | 
|  | * re-creates the file extents as if the write completed, but does not | 
|  | * do any checking - see comments in doio_check for more details. | 
|  | */ | 
|  |  | 
|  | if (sy->sy_flags & SY_WRITE && w_opt) { | 
|  | if (pid == -1) { | 
|  | pid = getpid(); | 
|  | } | 
|  |  | 
|  | wrec.w_async = (sy->sy_flags & SY_ASYNC) ? 1 : 0; | 
|  | wrec.w_oflags = oflags; | 
|  | wrec.w_pid = pid; | 
|  | wrec.w_offset = offset; | 
|  | wrec.w_nbytes = nbytes;	/* mem_needed -- total length */ | 
|  |  | 
|  | wrec.w_pathlen = strlen(file); | 
|  | memcpy(wrec.w_path, file, wrec.w_pathlen); | 
|  | wrec.w_hostlen = strlen(Host); | 
|  | memcpy(wrec.w_host, Host, wrec.w_hostlen); | 
|  | wrec.w_patternlen = Pattern_Length; | 
|  | memcpy(wrec.w_pattern, Pattern, wrec.w_patternlen); | 
|  |  | 
|  | wrec.w_done = 0; | 
|  |  | 
|  | if ((woffset = wlog_record_write(&Wlog, &wrec, -1)) == -1) { | 
|  | doio_fprintf(stderr, | 
|  | "Could not append to write-log:  %s (%d)\n", | 
|  | SYSERR, errno); | 
|  | } else { | 
|  | logged_write = 1; | 
|  | } | 
|  | } | 
|  |  | 
|  | s = (*sy->sy_syscall)(req, sy, fd, addr); | 
|  |  | 
|  | if( s->rval == -1 ) { | 
|  | doio_fprintf(stderr, | 
|  | "%s() request failed:  %s (%d)\n%s\n%s\n", | 
|  | sy->sy_name, SYSERR, errno, | 
|  | fmt_ioreq(req, sy, fd), | 
|  | (*sy->sy_format)(req, sy, fd, addr)); | 
|  |  | 
|  | doio_upanic(U_RVAL); | 
|  |  | 
|  | for(i=0; i < nents; i++) { | 
|  | if(s->aioid == NULL) | 
|  | break; | 
|  | aio_unregister(s->aioid[i]); | 
|  | } | 
|  | rval = -1; | 
|  | } else { | 
|  | /* | 
|  | * If the syscall was async, wait for I/O to complete | 
|  | */ | 
|  | #ifndef linux | 
|  | if(sy->sy_flags & SY_ASYNC) { | 
|  | for(i=0; i < nents; i++) { | 
|  | aio_wait(s->aioid[i]); | 
|  | } | 
|  | } | 
|  | #endif | 
|  |  | 
|  | /* | 
|  | * Check the syscall how-much-data-written return.  Look | 
|  | * for this in either the return value or the 'iosw' | 
|  | * structure. | 
|  | */ | 
|  |  | 
|  | if( !(sy->sy_flags & SY_IOSW) ) { | 
|  |  | 
|  | if(s->rval != mem_needed) { | 
|  | doio_fprintf(stderr, | 
|  | "%s() request returned wrong # of bytes - expected %d, got %d\n%s\n%s\n", | 
|  | sy->sy_name, nbytes, s->rval, | 
|  | fmt_ioreq(req, sy, fd), | 
|  | (*sy->sy_format)(req, sy, fd, addr)); | 
|  | rval = -1; | 
|  | doio_upanic(U_RVAL); | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  |  | 
|  | /* | 
|  | * Verify that the data was written correctly - check_file() returns | 
|  | * a non-null pointer which contains an error message if there are | 
|  | * problems. | 
|  | */ | 
|  |  | 
|  | if ( rval == 0 && sy->sy_flags & SY_WRITE && v_opt) { | 
|  | msg = check_file(file, offset, nbytes*nstrides*nents, | 
|  | Pattern, Pattern_Length, 0, | 
|  | oflags & O_PARALLEL); | 
|  | if (msg != NULL) { | 
|  | doio_fprintf(stderr, "%s\n%s\n%s\n", | 
|  | msg, | 
|  | fmt_ioreq(req, sy, fd), | 
|  | (*sy->sy_format)(req, sy, fd, addr)); | 
|  | doio_upanic(U_CORRUPTION); | 
|  | exit(E_COMPARE); | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * General cleanup ... | 
|  | * | 
|  | * Write extent information to the write-log, so that doio_check can do | 
|  | * corruption detection.  Note that w_done is set to 1, indicating that | 
|  | * the write has been verified as complete.  We don't need to write the | 
|  | * filename on the second logging. | 
|  | */ | 
|  |  | 
|  | if (w_opt && logged_write) { | 
|  | wrec.w_done = 1; | 
|  | wlog_record_write(&Wlog, &wrec, woffset); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Unlock file region if necessary | 
|  | */ | 
|  |  | 
|  | if (got_lock) { | 
|  | if (lock_file_region(file, fd, F_UNLCK, | 
|  | min_byte, (max_byte-min_byte+1)) < 0) { | 
|  | alloc_mem(-1); | 
|  | exit(E_INTERNAL); | 
|  | } | 
|  | } | 
|  |  | 
|  | if(s->aioid != NULL) | 
|  | free(s->aioid); | 
|  | free(s); | 
|  | return (rval == -1) ? -1 : 0; | 
|  | } | 
|  |  | 
|  |  | 
|  | /* | 
|  | * xfsctl-based requests | 
|  | *   - XFS_IOC_RESVSP | 
|  | *   - XFS_IOC_UNRESVSP | 
|  | */ | 
|  | int | 
|  | do_xfsctl(req) | 
|  | struct io_req	*req; | 
|  | { | 
|  | int	    		fd, oflags, offset, nbytes; | 
|  | int			rval, op = 0; | 
|  | int	    		got_lock; | 
|  | int			min_byte = 0, max_byte = 0; | 
|  | char    		*file, *msg = NULL; | 
|  | struct xfs_flock64    	flk; | 
|  |  | 
|  | /* | 
|  | * Initialize common fields - assumes r_oflags, r_file, r_offset, and | 
|  | * r_nbytes are at the same offset in the read_req and reada_req | 
|  | * structures. | 
|  | */ | 
|  | file	= req->r_data.io.r_file; | 
|  | oflags	= req->r_data.io.r_oflags; | 
|  | offset	= req->r_data.io.r_offset; | 
|  | nbytes	= req->r_data.io.r_nbytes; | 
|  |  | 
|  | flk.l_type=0; | 
|  | flk.l_whence=SEEK_SET; | 
|  | flk.l_start=offset; | 
|  | flk.l_len=nbytes; | 
|  |  | 
|  | /* | 
|  | * Get an open file descriptor | 
|  | */ | 
|  |  | 
|  | if ((fd = alloc_fd(file, oflags)) == -1) | 
|  | return -1; | 
|  |  | 
|  | rval = 0; | 
|  | got_lock = 0; | 
|  |  | 
|  | /* | 
|  | * Lock data if this is locking option is set | 
|  | */ | 
|  | if (k_opt) { | 
|  | min_byte = offset; | 
|  | max_byte = offset + nbytes; | 
|  |  | 
|  | if (lock_file_region(file, fd, F_WRLCK, | 
|  | min_byte, (nbytes+1)) < 0) { | 
|  | doio_fprintf(stderr, | 
|  | "file lock failed:\n"); | 
|  | doio_fprintf(stderr, | 
|  | "          buffer(req, %d, 0, 0x%x, 0x%x)\n", | 
|  | offset, min_byte, max_byte); | 
|  | alloc_mem(-1); | 
|  | exit(E_INTERNAL); | 
|  | } | 
|  |  | 
|  | got_lock = 1; | 
|  | } | 
|  |  | 
|  | switch (req->r_type) { | 
|  | case RESVSP:	op=XFS_IOC_RESVSP;	msg="resvsp";	break; | 
|  | case UNRESVSP:	op=XFS_IOC_UNRESVSP;	msg="unresvsp";	break; | 
|  | } | 
|  |  | 
|  | rval = xfsctl(file, fd, op, &flk); | 
|  |  | 
|  | if( rval == -1 ) { | 
|  | doio_fprintf(stderr, | 
|  | "xfsctl %s request failed: %s (%d)\n\txfsctl(%d, %s %d, {%d %lld ==> %lld}\n", | 
|  | msg, SYSERR, errno, | 
|  | fd, msg, op, flk.l_whence, | 
|  | (long long)flk.l_start, | 
|  | (long long)flk.l_len); | 
|  |  | 
|  | doio_upanic(U_RVAL); | 
|  | rval = -1; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Unlock file region if necessary | 
|  | */ | 
|  |  | 
|  | if (got_lock) { | 
|  | if (lock_file_region(file, fd, F_UNLCK, | 
|  | min_byte, (max_byte-min_byte+1)) < 0) { | 
|  | alloc_mem(-1); | 
|  | exit(E_INTERNAL); | 
|  | } | 
|  | } | 
|  |  | 
|  | return (rval == -1) ? -1 : 0; | 
|  | } | 
|  |  | 
|  | /* | 
|  | *  fsync(2) and fdatasync(2) | 
|  | */ | 
|  | int | 
|  | do_sync(req) | 
|  | struct io_req	*req; | 
|  | { | 
|  | int	    		fd, oflags; | 
|  | int			rval; | 
|  | char    		*file; | 
|  |  | 
|  | /* | 
|  | * Initialize common fields - assumes r_oflags, r_file, r_offset, and | 
|  | * r_nbytes are at the same offset in the read_req and reada_req | 
|  | * structures. | 
|  | */ | 
|  | file	= req->r_data.io.r_file; | 
|  | oflags	= req->r_data.io.r_oflags; | 
|  |  | 
|  | /* | 
|  | * Get an open file descriptor | 
|  | */ | 
|  |  | 
|  | if ((fd = alloc_fd(file, oflags)) == -1) | 
|  | return -1; | 
|  |  | 
|  | rval = 0; | 
|  | switch(req->r_type) { | 
|  | case FSYNC2: | 
|  | rval = fsync(fd); | 
|  | break; | 
|  | case FDATASYNC: | 
|  | rval = fdatasync(fd); | 
|  | break; | 
|  | default: | 
|  | rval = -1; | 
|  | } | 
|  | return (rval == -1) ? -1 : 0; | 
|  | } | 
|  |  | 
|  | int | 
|  | doio_pat_fill(char *addr, int mem_needed, char *Pattern, int Pattern_Length, | 
|  | int shift) | 
|  | { | 
|  | return pattern_fill(addr, mem_needed, Pattern, Pattern_Length, 0); | 
|  | } | 
|  |  | 
|  | char * | 
|  | doio_pat_check(buf, offset, length, pattern, pattern_length, patshift) | 
|  | char	*buf; | 
|  | int	offset; | 
|  | int 	length; | 
|  | char	*pattern; | 
|  | int	pattern_length; | 
|  | int	patshift; | 
|  | { | 
|  | static char	errbuf[4096]; | 
|  | int		nb, i, pattern_index; | 
|  | char    	*cp, *bufend, *ep; | 
|  | char    	actual[33], expected[33]; | 
|  |  | 
|  | if (pattern_check(buf, length, pattern, pattern_length, patshift) != 0) { | 
|  | ep = errbuf; | 
|  | ep += sprintf(ep, "Corrupt regions follow - unprintable chars are represented as '.'\n"); | 
|  | ep += sprintf(ep, "-----------------------------------------------------------------\n"); | 
|  |  | 
|  | pattern_index = patshift % pattern_length;; | 
|  | cp = buf; | 
|  | bufend = buf + length; | 
|  |  | 
|  | while (cp < bufend) { | 
|  | if (*cp != pattern[pattern_index]) { | 
|  | nb = bufend - cp; | 
|  | if (nb > sizeof(expected)-1) { | 
|  | nb = sizeof(expected)-1; | 
|  | } | 
|  |  | 
|  | ep += sprintf(ep, "corrupt bytes starting at file offset %d\n", offset + (int)(cp-buf)); | 
|  |  | 
|  | /* | 
|  | * Fill in the expected and actual patterns | 
|  | */ | 
|  | bzero(expected, sizeof(expected)); | 
|  | bzero(actual, sizeof(actual)); | 
|  |  | 
|  | for (i = 0; i < nb; i++) { | 
|  | expected[i] = pattern[(pattern_index + i) % pattern_length]; | 
|  | if (! isprint((int)expected[i])) { | 
|  | expected[i] = '.'; | 
|  | } | 
|  |  | 
|  | actual[i] = cp[i]; | 
|  | if (! isprint((int)actual[i])) { | 
|  | actual[i] = '.'; | 
|  | } | 
|  | } | 
|  |  | 
|  | ep += sprintf(ep, "    1st %2d expected bytes:  %s\n", nb, expected); | 
|  | ep += sprintf(ep, "    1st %2d actual bytes:    %s\n", nb, actual); | 
|  | fflush(stderr); | 
|  | return errbuf; | 
|  | } else { | 
|  | cp++; | 
|  | pattern_index++; | 
|  |  | 
|  | if (pattern_index == pattern_length) { | 
|  | pattern_index = 0; | 
|  | } | 
|  | } | 
|  | } | 
|  | return errbuf; | 
|  | } | 
|  |  | 
|  | return(NULL); | 
|  | } | 
|  |  | 
|  |  | 
|  | /* | 
|  | * Check the contents of a file beginning at offset, for length bytes.  It | 
|  | * is assumed that there is a string of pattern bytes in this area of the | 
|  | * file.  Use normal buffered reads to do the verification. | 
|  | * | 
|  | * If there is a data mismatch, write a detailed message into a static buffer | 
|  | * suitable for the caller to print.  Otherwise print NULL. | 
|  | * | 
|  | * The fsa flag is set to non-zero if the buffer should be read back through | 
|  | * the FSA (unicos/mk).  This implies the file will be opened | 
|  | * O_PARALLEL|O_RAW|O_WELLFORMED to do the validation.  We must do this because | 
|  | * FSA will not allow the file to be opened for buffered io if it was | 
|  | * previously opened for O_PARALLEL io. | 
|  | */ | 
|  |  | 
|  | char * | 
|  | check_file(file, offset, length, pattern, pattern_length, patshift, fsa) | 
|  | char 	*file; | 
|  | int 	offset; | 
|  | int 	length; | 
|  | char	*pattern; | 
|  | int	pattern_length; | 
|  | int	patshift; | 
|  | int	fsa; | 
|  | { | 
|  | static char	errbuf[4096]; | 
|  | int	    	fd, nb, flags; | 
|  | char		*buf, *em, *ep; | 
|  | struct fd_cache *fdc; | 
|  |  | 
|  | buf = Memptr; | 
|  |  | 
|  | if (V_opt) { | 
|  | flags = Validation_Flags | O_RDONLY; | 
|  | } else { | 
|  | flags = O_RDONLY; | 
|  | } | 
|  |  | 
|  | if ((fd = alloc_fd(file, flags)) == -1) { | 
|  | sprintf(errbuf, | 
|  | "Could not open file %s with flags %#o (%s) for data comparison:  %s (%d)\n", | 
|  | file, flags, format_oflags(flags), | 
|  | SYSERR, errno); | 
|  | return errbuf; | 
|  | } | 
|  |  | 
|  | if (lseek(fd, offset, SEEK_SET) == -1) { | 
|  | sprintf(errbuf, | 
|  | "Could not lseek to offset %d in %s for verification:  %s (%d)\n", | 
|  | offset, file, SYSERR, errno); | 
|  | return errbuf; | 
|  | } | 
|  |  | 
|  | /* Guarantee a properly aligned address on Direct I/O */ | 
|  | fdc = alloc_fdcache(file, flags); | 
|  | if( (flags & O_DIRECT) && ((long)buf % fdc->c_memalign != 0) ) { | 
|  | buf += fdc->c_memalign - ((long)buf % fdc->c_memalign); | 
|  | } | 
|  |  | 
|  | if ((nb = read(fd, buf, length)) == -1) { | 
|  | sprintf(errbuf, | 
|  | "Could not read %d bytes from %s for verification:  %s (%d)\n\tread(%d, 0x%p, %d)\n\tbuf %% alignment(%d) = %ld\n", | 
|  | length, file, SYSERR, errno, | 
|  | fd, buf, length, | 
|  | fdc->c_memalign, (long)buf % fdc->c_memalign); | 
|  | return errbuf; | 
|  | } | 
|  |  | 
|  | if (nb != length) { | 
|  | sprintf(errbuf, | 
|  | "Read wrong # bytes from %s.  Expected %d, got %d\n", | 
|  | file, length, nb); | 
|  | return errbuf; | 
|  | } | 
|  |  | 
|  | if( (em = (*Data_Check)(buf, offset, length, pattern, pattern_length, patshift)) != NULL ) { | 
|  | ep = errbuf; | 
|  | ep += sprintf(ep, "*** DATA COMPARISON ERROR ***\n"); | 
|  | ep += sprintf(ep, "check_file(%s, %d, %d, %s, %d, %d) failed\n\n", | 
|  | file, offset, length, pattern, pattern_length, patshift); | 
|  | ep += sprintf(ep, "Comparison fd is %d, with open flags %#o\n", | 
|  | fd, flags); | 
|  | strcpy(ep, em); | 
|  | return(errbuf); | 
|  | } | 
|  | return NULL; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Function to single-thread stdio output. | 
|  | */ | 
|  |  | 
|  | int | 
|  | doio_fprintf(FILE *stream, char *format, ...) | 
|  | { | 
|  | static int	pid = -1; | 
|  | char		*date; | 
|  | int		rval; | 
|  | struct flock	flk; | 
|  | va_list		arglist; | 
|  |  | 
|  | date = hms(time(0)); | 
|  |  | 
|  | if (pid == -1) { | 
|  | pid = getpid(); | 
|  | } | 
|  |  | 
|  | flk.l_whence = flk.l_start = flk.l_len = 0; | 
|  | flk.l_type = F_WRLCK; | 
|  | fcntl(fileno(stream), F_SETLKW, &flk); | 
|  |  | 
|  | va_start(arglist, format); | 
|  | rval = fprintf(stream, "\n%s%s (%5d) %s\n", Prog, TagName, pid, date); | 
|  | rval += fprintf(stream, "---------------------\n"); | 
|  | vfprintf(stream, format, arglist); | 
|  | va_end(arglist); | 
|  |  | 
|  | fflush(stream); | 
|  |  | 
|  | flk.l_type = F_UNLCK; | 
|  | fcntl(fileno(stream), F_SETLKW, &flk); | 
|  |  | 
|  | return rval; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Simple function for allocating core memory.  Uses Memsize and Memptr to | 
|  | * keep track of the current amount allocated. | 
|  | */ | 
|  |  | 
|  | int | 
|  | alloc_mem(nbytes) | 
|  | int nbytes; | 
|  | { | 
|  | char    	*cp; | 
|  | void		*addr; | 
|  | int		me = 0, flags, key, shmid; | 
|  | static int	mturn = 0;	/* which memory type to use */ | 
|  | struct memalloc	*M; | 
|  | char		filename[255]; | 
|  | #ifdef linux | 
|  | struct shmid_ds shm_ds; | 
|  | bzero( &shm_ds, sizeof(struct shmid_ds) ); | 
|  | #endif | 
|  |  | 
|  | /* nbytes = -1 means "free all allocated memory" */ | 
|  | if( nbytes == -1 ) { | 
|  |  | 
|  | for(me=0; me < Nmemalloc; me++) { | 
|  | if(Memalloc[me].space == NULL) | 
|  | continue; | 
|  |  | 
|  | switch(Memalloc[me].memtype) { | 
|  | case MEM_DATA: | 
|  | free(Memalloc[me].space); | 
|  | Memalloc[me].space = NULL; | 
|  | Memptr = NULL; | 
|  | Memsize = 0; | 
|  | break; | 
|  | case MEM_SHMEM: | 
|  | shmdt(Memalloc[me].space); | 
|  | Memalloc[me].space = NULL; | 
|  | shmctl(Memalloc[me].fd, IPC_RMID, &shm_ds); | 
|  | break; | 
|  | case MEM_MMAP: | 
|  | munmap(Memalloc[me].space, | 
|  | Memalloc[me].size); | 
|  | close(Memalloc[me].fd); | 
|  | if(Memalloc[me].flags & MEMF_FILE) { | 
|  | unlink(Memalloc[me].name); | 
|  | } | 
|  | Memalloc[me].space = NULL; | 
|  | break; | 
|  | default: | 
|  | doio_fprintf(stderr, "alloc_mem: HELP! Unknown memory space type %d index %d\n", | 
|  | Memalloc[me].memtype, me); | 
|  | break; | 
|  | } | 
|  | } | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Select a memory area (currently round-robbin) | 
|  | */ | 
|  |  | 
|  | if(mturn >= Nmemalloc) | 
|  | mturn=0; | 
|  |  | 
|  | M = &Memalloc[mturn]; | 
|  |  | 
|  | switch(M->memtype) { | 
|  | case MEM_DATA: | 
|  | if( nbytes > M->size ) { | 
|  | if( M->space != NULL ){ | 
|  | free(M->space); | 
|  | } | 
|  | M->space = NULL; | 
|  | M->size = 0; | 
|  | } | 
|  |  | 
|  | if( M->space == NULL ) { | 
|  | if( (cp = malloc( nbytes )) == NULL ) { | 
|  | doio_fprintf(stderr, "malloc(%d) failed:  %s (%d)\n", | 
|  | nbytes, SYSERR, errno); | 
|  | return -1; | 
|  | } | 
|  | M->space = (void *)cp; | 
|  | M->size = nbytes; | 
|  | } | 
|  | break; | 
|  |  | 
|  | case MEM_MMAP: | 
|  | if( nbytes > M->size ) { | 
|  | if( M->space != NULL ) { | 
|  | munmap(M->space, M->size); | 
|  | close(M->fd); | 
|  | if( M->flags & MEMF_FILE ) | 
|  | unlink( M->name ); | 
|  | } | 
|  | M->space = NULL; | 
|  | M->size = 0; | 
|  | } | 
|  |  | 
|  | if( M->space == NULL ) { | 
|  | if(strchr(M->name, '%')) { | 
|  | sprintf(filename, M->name, getpid()); | 
|  | M->name = strdup(filename); | 
|  | } | 
|  |  | 
|  | if( (M->fd = open(M->name, O_CREAT|O_RDWR, 0666)) == -1) { | 
|  | doio_fprintf(stderr, "alloc_mmap: error %d (%s) opening '%s'\n", | 
|  | errno, SYSERR, | 
|  | M->name); | 
|  | return(-1); | 
|  | } | 
|  |  | 
|  | addr = NULL; | 
|  | flags = 0; | 
|  | M->size = nbytes * 4; | 
|  |  | 
|  | /* bias addr if MEMF_ADDR | MEMF_FIXADDR */ | 
|  | /* >>> how to pick a memory address? */ | 
|  |  | 
|  | /* bias flags on MEMF_PRIVATE etc */ | 
|  | if(M->flags & MEMF_PRIVATE) | 
|  | flags |= MAP_PRIVATE; | 
|  | if(M->flags & MEMF_SHARED) | 
|  | flags |= MAP_SHARED; | 
|  |  | 
|  | /*printf("alloc_mem, about to mmap, fd=%d, name=(%s)\n", M->fd, M->name);*/ | 
|  | if( (M->space = mmap(addr, M->size, | 
|  | PROT_READ|PROT_WRITE, | 
|  | flags, M->fd, 0)) | 
|  | == MAP_FAILED) { | 
|  | doio_fprintf(stderr, "alloc_mem: mmap error. errno %d (%s)\n\tmmap(addr 0x%x, size %d, read|write 0x%x, mmap flags 0x%x [%#o], fd %d, 0)\n\tfile %s\n", | 
|  | errno, SYSERR, | 
|  | addr, M->size, | 
|  | PROT_READ|PROT_WRITE, | 
|  | flags, M->flags, M->fd, | 
|  | M->name); | 
|  | doio_fprintf(stderr, "\t%s%s%s%s%s", | 
|  | (flags & MAP_PRIVATE) ? "private " : "", | 
|  | (flags & MAP_SHARED) ? "shared" : ""); | 
|  | return(-1); | 
|  | } | 
|  | } | 
|  | break; | 
|  |  | 
|  | case MEM_SHMEM: | 
|  | if( nbytes > M->size ) { | 
|  | if( M->space != NULL ) { | 
|  | shmctl( M->fd, IPC_RMID, &shm_ds ); | 
|  | } | 
|  | M->space = NULL; | 
|  | M->size = 0; | 
|  | } | 
|  |  | 
|  | if(M->space == NULL) { | 
|  | if(!strcmp(M->name, "private")) { | 
|  | key = IPC_PRIVATE; | 
|  | } else { | 
|  | sscanf(M->name, "%i", &key); | 
|  | } | 
|  |  | 
|  | M->size = M->nblks ? M->nblks * 512 : nbytes; | 
|  |  | 
|  | if( nbytes > M->size ){ | 
|  | #ifdef DEBUG | 
|  | doio_fprintf(stderr, "MEM_SHMEM: nblks(%d) too small:  nbytes=%d  Msize=%d, skipping this req.\n", | 
|  | M->nblks, nbytes, M->size ); | 
|  | #endif | 
|  | return SKIP_REQ; | 
|  | } | 
|  |  | 
|  | shmid = shmget(key, M->size, IPC_CREAT|0666); | 
|  | if( shmid == -1 ) { | 
|  | doio_fprintf(stderr, "shmget(0x%x, %d, CREAT) failed: %s (%d)\n", | 
|  | key, M->size, SYSERR, errno); | 
|  | return(-1); | 
|  | } | 
|  | M->fd = shmid; | 
|  | M->space = shmat(shmid, NULL, SHM_RND); | 
|  | if( M->space == (void *)-1 ) { | 
|  | doio_fprintf(stderr, "shmat(0x%x, NULL, SHM_RND) failed: %s (%d)\n", | 
|  | shmid, SYSERR, errno); | 
|  | return(-1); | 
|  | } | 
|  | } | 
|  | break; | 
|  |  | 
|  | default: | 
|  | doio_fprintf(stderr, "alloc_mem: HELP! Unknown memory space type %d index %d\n", | 
|  | Memalloc[me].memtype, mturn); | 
|  | break; | 
|  | } | 
|  |  | 
|  | Memptr = M->space; | 
|  | Memsize = M->size; | 
|  |  | 
|  | mturn++; | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Function to maintain a file descriptor cache, so that doio does not have | 
|  | * to do so many open() and close() calls.  Descriptors are stored in the | 
|  | * cache by file name, and open flags.  Each entry also has a _rtc value | 
|  | * associated with it which is used in aging.  If doio cannot open a file | 
|  | * because it already has too many open (ie. system limit hit) it will close | 
|  | * the one in the cache that has the oldest _rtc value. | 
|  | * | 
|  | * If alloc_fd() is called with a file of NULL, it will close all descriptors | 
|  | * in the cache, and free the memory in the cache. | 
|  | */ | 
|  |  | 
|  | int | 
|  | alloc_fd(file, oflags) | 
|  | char	*file; | 
|  | int	oflags; | 
|  | { | 
|  | struct fd_cache *fdc; | 
|  | struct fd_cache *alloc_fdcache(char *file, int oflags); | 
|  |  | 
|  | fdc = alloc_fdcache(file, oflags); | 
|  | if(fdc != NULL) | 
|  | return(fdc->c_fd); | 
|  | else | 
|  | return(-1); | 
|  | } | 
|  |  | 
|  | struct fd_cache * | 
|  | alloc_fdcache(file, oflags) | 
|  | char	*file; | 
|  | int	oflags; | 
|  | { | 
|  | int			fd; | 
|  | struct fd_cache		*free_slot, *oldest_slot, *cp; | 
|  | static int		cache_size = 0; | 
|  | static struct fd_cache	*cache = NULL; | 
|  | struct dioattr	finfo; | 
|  |  | 
|  | /* | 
|  | * If file is NULL, it means to free up the fd cache. | 
|  | */ | 
|  |  | 
|  | if (file == NULL && cache != NULL) { | 
|  | for (cp = cache; cp < &cache[cache_size]; cp++) { | 
|  | if (cp->c_fd != -1) { | 
|  | close(cp->c_fd); | 
|  | } | 
|  | if (cp->c_memaddr != NULL) { | 
|  | munmap(cp->c_memaddr, cp->c_memlen); | 
|  | } | 
|  | } | 
|  |  | 
|  | free(cache); | 
|  | cache = NULL; | 
|  | cache_size = 0; | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | free_slot = NULL; | 
|  | oldest_slot = NULL; | 
|  |  | 
|  | /* | 
|  | * Look for a fd in the cache.  If one is found, return it directly. | 
|  | * Otherwise, when this loop exits, oldest_slot will point to the | 
|  | * oldest fd slot in the cache, and free_slot will point to an | 
|  | * unoccupied slot if there are any. | 
|  | */ | 
|  |  | 
|  | for (cp = cache; cp != NULL && cp < &cache[cache_size]; cp++) { | 
|  | if (cp->c_fd != -1 && | 
|  | cp->c_oflags == oflags && | 
|  | strcmp(cp->c_file, file) == 0) { | 
|  | cp->c_rtc = Reqno; | 
|  | return cp; | 
|  | } | 
|  |  | 
|  | if (cp->c_fd == -1) { | 
|  | if (free_slot == NULL) { | 
|  | free_slot = cp; | 
|  | } | 
|  | } else { | 
|  | if (oldest_slot == NULL || | 
|  | cp->c_rtc < oldest_slot->c_rtc) { | 
|  | oldest_slot = cp; | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * No matching file/oflags pair was found in the cache.  Attempt to | 
|  | * open a new fd. | 
|  | */ | 
|  |  | 
|  | if ((fd = open(file, oflags, 0666)) < 0) { | 
|  | if (errno != EMFILE) { | 
|  | doio_fprintf(stderr, | 
|  | "Could not open file %s with flags %#o (%s): %s (%d)\n", | 
|  | file, oflags, format_oflags(oflags), | 
|  | SYSERR, errno); | 
|  | alloc_mem(-1); | 
|  | exit(E_SETUP); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * If we get here, we have as many open fd's as we can have. | 
|  | * Close the oldest one in the cache (pointed to by | 
|  | * oldest_slot), and attempt to re-open. | 
|  | */ | 
|  |  | 
|  | close(oldest_slot->c_fd); | 
|  | oldest_slot->c_fd = -1; | 
|  | free_slot = oldest_slot; | 
|  |  | 
|  | if ((fd = open(file, oflags, 0666)) < 0) { | 
|  | doio_fprintf(stderr, | 
|  | "Could not open file %s with flags %#o (%s):  %s (%d)\n", | 
|  | file, oflags, format_oflags(oflags), | 
|  | SYSERR, errno); | 
|  | alloc_mem(-1); | 
|  | exit(E_SETUP); | 
|  | } | 
|  | } | 
|  |  | 
|  | /*printf("alloc_fd: new file %s flags %#o fd %d\n", file, oflags, fd);*/ | 
|  |  | 
|  | /* | 
|  | * If we get here, fd is our open descriptor.  If free_slot is NULL, | 
|  | * we need to grow the cache, otherwise free_slot is the slot that | 
|  | * should hold the fd info. | 
|  | */ | 
|  |  | 
|  | if (free_slot == NULL) { | 
|  | cache = (struct fd_cache *)realloc(cache, sizeof(struct fd_cache) * (FD_ALLOC_INCR + cache_size)); | 
|  | if (cache == NULL) { | 
|  | doio_fprintf(stderr, "Could not malloc() space for fd chace"); | 
|  | alloc_mem(-1); | 
|  | exit(E_SETUP); | 
|  | } | 
|  |  | 
|  | cache_size += FD_ALLOC_INCR; | 
|  |  | 
|  | for (cp = &cache[cache_size-FD_ALLOC_INCR]; | 
|  | cp < &cache[cache_size]; cp++) { | 
|  | cp->c_fd = -1; | 
|  | } | 
|  |  | 
|  | free_slot = &cache[cache_size - FD_ALLOC_INCR]; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * finally, fill in the cache slot info | 
|  | */ | 
|  |  | 
|  | free_slot->c_fd = fd; | 
|  | free_slot->c_oflags = oflags; | 
|  | strcpy(free_slot->c_file, file); | 
|  | free_slot->c_rtc = Reqno; | 
|  |  | 
|  | if (oflags & O_DIRECT) { | 
|  | char *dio_env; | 
|  |  | 
|  | if (xfsctl(file, fd, XFS_IOC_DIOINFO, &finfo) == -1) { | 
|  | finfo.d_mem = 1; | 
|  | finfo.d_miniosz = 1; | 
|  | finfo.d_maxiosz = 1; | 
|  | } | 
|  |  | 
|  | dio_env = getenv("XFS_DIO_MIN"); | 
|  | if (dio_env) | 
|  | finfo.d_mem = finfo.d_miniosz = atoi(dio_env); | 
|  |  | 
|  | } else { | 
|  | finfo.d_mem = 1; | 
|  | finfo.d_miniosz = 1; | 
|  | finfo.d_maxiosz = 1; | 
|  | } | 
|  |  | 
|  | free_slot->c_memalign = finfo.d_mem; | 
|  | free_slot->c_miniosz = finfo.d_miniosz; | 
|  | free_slot->c_maxiosz = finfo.d_maxiosz; | 
|  | free_slot->c_memaddr = NULL; | 
|  | free_slot->c_memlen = 0; | 
|  |  | 
|  | return free_slot; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * | 
|  | *			Signal Handling Section | 
|  | * | 
|  | * | 
|  | */ | 
|  |  | 
|  | void | 
|  | cleanup_handler() | 
|  | { | 
|  | havesigint=1; /* in case there's a followup signal */ | 
|  | alloc_mem(-1); | 
|  | exit(0); | 
|  | } | 
|  |  | 
|  | void | 
|  | die_handler(sig) | 
|  | int sig; | 
|  | { | 
|  | doio_fprintf(stderr, "terminating on signal %d\n", sig); | 
|  | alloc_mem(-1); | 
|  | exit(1); | 
|  | } | 
|  |  | 
|  | void | 
|  | sigbus_handler(sig) | 
|  | int sig; | 
|  | { | 
|  | /* See sigbus_handler() in the 'ifdef sgi' case for details.  Here, | 
|  | we don't have the siginfo stuff so the guess is weaker but we'll | 
|  | do it anyway. | 
|  | */ | 
|  |  | 
|  | if( active_mmap_rw && havesigint ) | 
|  | cleanup_handler(); | 
|  | else | 
|  | die_handler(sig); | 
|  | } | 
|  |  | 
|  | void | 
|  | noop_handler(sig) | 
|  | int sig; | 
|  | { | 
|  | return; | 
|  | } | 
|  |  | 
|  |  | 
|  | /* | 
|  | * SIGINT handler for the parent (original doio) process.  It simply sends | 
|  | * a SIGINT to all of the doio children.  Since they're all in the same | 
|  | * pgrp, this can be done with a single kill(). | 
|  | */ | 
|  |  | 
|  | void | 
|  | sigint_handler() | 
|  | { | 
|  | int	i; | 
|  |  | 
|  | for (i = 0; i < Nchildren; i++) { | 
|  | if (Children[i] != -1) { | 
|  | kill(Children[i], SIGINT); | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Signal handler used to inform a process when async io completes.  Referenced | 
|  | * in do_read() and do_write().  Note that the signal handler is not | 
|  | * re-registered. | 
|  | */ | 
|  |  | 
|  | void | 
|  | aio_handler(sig) | 
|  | int	sig; | 
|  | { | 
|  | int		i; | 
|  | struct aio_info	*aiop; | 
|  |  | 
|  | for (i = 0; i < sizeof(Aio_Info) / sizeof(Aio_Info[0]); i++) { | 
|  | aiop = &Aio_Info[i]; | 
|  |  | 
|  | if (aiop->strategy == A_SIGNAL && aiop->sig == sig) { | 
|  | aiop->signalled++; | 
|  |  | 
|  | if (aio_done(aiop)) { | 
|  | aiop->done++; | 
|  | } | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * dump info on all open aio slots | 
|  | */ | 
|  | void | 
|  | dump_aio() | 
|  | { | 
|  | int		i, count; | 
|  |  | 
|  | count=0; | 
|  | for (i = 0; i < sizeof(Aio_Info) / sizeof(Aio_Info[0]); i++) { | 
|  | if (Aio_Info[i].busy) { | 
|  | count++; | 
|  | fprintf(stderr, | 
|  | "Aio_Info[%03d] id=%d fd=%d signal=%d signaled=%d\n", | 
|  | i, Aio_Info[i].id, | 
|  | Aio_Info[i].fd, | 
|  | Aio_Info[i].sig, | 
|  | Aio_Info[i].signalled); | 
|  | fprintf(stderr, "\tstrategy=%s\n", | 
|  | format_strat(Aio_Info[i].strategy)); | 
|  | } | 
|  | } | 
|  | fprintf(stderr, "%d active async i/os\n", count); | 
|  | } | 
|  |  | 
|  | struct aio_info * | 
|  | aio_slot(aio_id) | 
|  | int	aio_id; | 
|  | { | 
|  | int		i; | 
|  | static int	id = 1; | 
|  | struct aio_info	*aiop; | 
|  |  | 
|  | aiop = NULL; | 
|  |  | 
|  | for (i = 0; i < sizeof(Aio_Info) / sizeof(Aio_Info[0]); i++) { | 
|  | if (aio_id == -1) { | 
|  | if (! Aio_Info[i].busy) { | 
|  | aiop = &Aio_Info[i]; | 
|  | aiop->busy = 1; | 
|  | aiop->id = id++; | 
|  | break; | 
|  | } | 
|  | } else { | 
|  | if (Aio_Info[i].busy && Aio_Info[i].id == aio_id) { | 
|  | aiop = &Aio_Info[i]; | 
|  | break; | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | if( aiop == NULL ){ | 
|  | doio_fprintf(stderr,"aio_slot(%d) not found.  Request %d\n", | 
|  | aio_id, Reqno); | 
|  | dump_aio(); | 
|  | alloc_mem(-1); | 
|  | exit(E_INTERNAL); | 
|  | } | 
|  |  | 
|  | return aiop; | 
|  | } | 
|  |  | 
|  | int | 
|  | aio_register(fd, strategy, sig) | 
|  | int		fd; | 
|  | int		strategy; | 
|  | int		sig; | 
|  | { | 
|  | struct aio_info		*aiop; | 
|  | void			aio_handler(); | 
|  | struct sigaction	sa; | 
|  |  | 
|  | aiop = aio_slot(-1); | 
|  |  | 
|  | aiop->fd = fd; | 
|  | aiop->strategy = strategy; | 
|  | aiop->done = 0; | 
|  |  | 
|  | if (strategy == A_SIGNAL) { | 
|  | aiop->sig = sig; | 
|  | aiop->signalled = 0; | 
|  |  | 
|  | sa.sa_handler = aio_handler; | 
|  | sa.sa_flags = 0; | 
|  | sigemptyset(&sa.sa_mask); | 
|  |  | 
|  | sigaction(sig, &sa, &aiop->osa); | 
|  | } else { | 
|  | aiop->sig = -1; | 
|  | aiop->signalled = 0; | 
|  | } | 
|  |  | 
|  | return aiop->id; | 
|  | } | 
|  |  | 
|  | int | 
|  | aio_unregister(aio_id) | 
|  | int	aio_id; | 
|  | { | 
|  | struct aio_info	*aiop; | 
|  |  | 
|  | aiop = aio_slot(aio_id); | 
|  |  | 
|  | if (aiop->strategy == A_SIGNAL) { | 
|  | sigaction(aiop->sig, &aiop->osa, NULL); | 
|  | } | 
|  |  | 
|  | aiop->busy = 0; | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | #ifndef linux | 
|  | int | 
|  | aio_wait(aio_id) | 
|  | int	aio_id; | 
|  | { | 
|  | #ifdef RECALL_SIZEOF | 
|  | long		mask[RECALL_SIZEOF]; | 
|  | #endif | 
|  | sigset_t	sigset; | 
|  | struct aio_info	*aiop; | 
|  |  | 
|  | /*printf("aio_wait: errno %d return %d\n", aiop->aio_errno, aiop->aio_ret);*/ | 
|  |  | 
|  | return 0; | 
|  | } | 
|  | #endif /* !linux */ | 
|  |  | 
|  | /* | 
|  | * Format specified time into HH:MM:SS format.  t is the time to format | 
|  | * in seconds (as returned from time(2)). | 
|  | */ | 
|  |  | 
|  | char * | 
|  | hms(t) | 
|  | time_t	t; | 
|  | { | 
|  | static char	ascii_time[9]; | 
|  | struct tm	*ltime; | 
|  |  | 
|  | ltime = localtime(&t); | 
|  | strftime(ascii_time, sizeof(ascii_time), "%H:%M:%S", ltime); | 
|  |  | 
|  | return ascii_time; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Simple routine to check if an async io request has completed. | 
|  | */ | 
|  |  | 
|  | int | 
|  | aio_done(struct aio_info *ainfo) | 
|  | { | 
|  | return -1;   /* invalid */ | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Routine to handle upanic() - it first attempts to set the panic flag.  If | 
|  | * the flag cannot be set, an error message is issued.  A call to upanic | 
|  | * with PA_PANIC is then done unconditionally, in case the panic flag was set | 
|  | * from outside the program (as with the panic(8) program). | 
|  | * | 
|  | * Note - we only execute the upanic code if -U was used, and the passed in | 
|  | * mask is set in the Upanic_Conditions bitmask. | 
|  | */ | 
|  |  | 
|  | void | 
|  | doio_upanic(mask) | 
|  | int	mask; | 
|  | { | 
|  | if (U_opt == 0 || (mask & Upanic_Conditions) == 0) { | 
|  | return; | 
|  | } | 
|  | doio_fprintf(stderr, "WARNING - upanic() failed\n"); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Parse cmdline options/arguments and set appropriate global variables. | 
|  | * If the cmdline is valid, return 0 to caller.  Otherwise exit with a status | 
|  | * of 1. | 
|  | */ | 
|  |  | 
|  | int | 
|  | parse_cmdline(argc, argv, opts) | 
|  | int 	argc; | 
|  | char	**argv; | 
|  | char	*opts; | 
|  | { | 
|  | int	    	c; | 
|  | char    	cc, *cp, *tok = NULL; | 
|  | extern int	opterr; | 
|  | extern int	optind; | 
|  | extern char	*optarg; | 
|  | struct smap	*s; | 
|  | char		*memargs[NMEMALLOC]; | 
|  | int		nmemargs, ma; | 
|  | void		parse_memalloc(char *arg); | 
|  | void		parse_delay(char *arg); | 
|  | void		dump_memalloc(); | 
|  |  | 
|  | if (*argv[0] == '-') { | 
|  | argv[0]++; | 
|  | Execd = 1; | 
|  | } | 
|  |  | 
|  | if ((Prog = strrchr(argv[0], '/')) == NULL) { | 
|  | Prog = argv[0]; | 
|  | } else { | 
|  | Prog++; | 
|  | } | 
|  |  | 
|  | opterr = 0; | 
|  | while ((c = getopt(argc, argv, opts)) != EOF) { | 
|  | switch ((char)c) { | 
|  | case 'a': | 
|  | a_opt++; | 
|  | break; | 
|  |  | 
|  | case 'C': | 
|  | C_opt++; | 
|  | for(s=checkmap; s->string != NULL; s++) | 
|  | if(!strcmp(s->string, optarg)) | 
|  | break; | 
|  | if (s->string == NULL) { | 
|  | fprintf(stderr, | 
|  | "%s%s:  Illegal -C arg (%s).  Must be one of: ", | 
|  | Prog, TagName, tok); | 
|  |  | 
|  | for (s = checkmap; s->string != NULL; s++) | 
|  | fprintf(stderr, "%s ", s->string); | 
|  | fprintf(stderr, "\n"); | 
|  | exit(1); | 
|  | } | 
|  |  | 
|  | switch(s->value) { | 
|  | case C_DEFAULT: | 
|  | Data_Fill = doio_pat_fill; | 
|  | Data_Check = doio_pat_check; | 
|  | break; | 
|  | default: | 
|  | fprintf(stderr, | 
|  | "%s%s:  Unrecognised -C arg '%s' %d", | 
|  | Prog, TagName, s->string, s->value); | 
|  | exit(1); | 
|  | } | 
|  | break; | 
|  |  | 
|  | case 'd':	/* delay between i/o ops */ | 
|  | parse_delay(optarg); | 
|  | break; | 
|  |  | 
|  | case 'e': | 
|  | if (Npes > 1 && Nprocs > 1) { | 
|  | fprintf(stderr, "%s%s:  Warning - Program is a multi-pe application - exec option is ignored.\n", Prog, TagName); | 
|  | } | 
|  | e_opt++; | 
|  | break; | 
|  |  | 
|  | case 'h': | 
|  | help(stdout); | 
|  | exit(0); | 
|  | break; | 
|  |  | 
|  | case 'k': | 
|  | k_opt++; | 
|  | break; | 
|  |  | 
|  | case 'm': | 
|  | Message_Interval = strtol(optarg, &cp, 10); | 
|  | if (*cp != '\0' || Message_Interval < 0) { | 
|  | fprintf(stderr, "%s%s:  Illegal -m arg (%s):  Must be an integer >= 0\n", Prog, TagName, optarg); | 
|  | exit(1); | 
|  | } | 
|  | m_opt++; | 
|  | break; | 
|  |  | 
|  | case 'M':	/* memory allocation types */ | 
|  | nmemargs = string_to_tokens(optarg, memargs, 32, ","); | 
|  | for(ma=0; ma < nmemargs; ma++) { | 
|  | parse_memalloc(memargs[ma]); | 
|  | } | 
|  | /*dump_memalloc();*/ | 
|  | M_opt++; | 
|  | break; | 
|  |  | 
|  | case 'N': | 
|  | sprintf( TagName, "(%.37s)", optarg ); | 
|  | break; | 
|  |  | 
|  | case 'n': | 
|  | Nprocs = strtol(optarg, &cp, 10); | 
|  | if (*cp != '\0' || Nprocs < 1) { | 
|  | fprintf(stderr, | 
|  | "%s%s:  Illegal -n arg (%s):  Must be integer > 0\n", | 
|  | Prog, TagName, optarg); | 
|  | exit(E_USAGE); | 
|  | } | 
|  |  | 
|  | if (Npes > 1 && Nprocs > 1) { | 
|  | fprintf(stderr, "%s%s:  Program has been built as a multi-pe app.  -n1 is the only nprocs value allowed\n", Prog, TagName); | 
|  | exit(E_SETUP); | 
|  | } | 
|  | n_opt++; | 
|  | break; | 
|  |  | 
|  | case 'r': | 
|  | Release_Interval = strtol(optarg, &cp, 10); | 
|  | if (*cp != '\0' || Release_Interval < 0) { | 
|  | fprintf(stderr, | 
|  | "%s%s:  Illegal -r arg (%s):  Must be integer >= 0\n", | 
|  | Prog, TagName, optarg); | 
|  | exit(E_USAGE); | 
|  | } | 
|  |  | 
|  | r_opt++; | 
|  | break; | 
|  |  | 
|  | case 'w': | 
|  | Write_Log = optarg; | 
|  | w_opt++; | 
|  | break; | 
|  |  | 
|  | case 'v': | 
|  | v_opt++; | 
|  | break; | 
|  |  | 
|  | case 'V': | 
|  | if (strcasecmp(optarg, "sync") == 0) { | 
|  | Validation_Flags = O_SYNC; | 
|  | } else if (strcasecmp(optarg, "buffered") == 0) { | 
|  | Validation_Flags = 0; | 
|  | } else if (strcasecmp(optarg, "direct") == 0) { | 
|  | Validation_Flags = O_DIRECT; | 
|  | } else { | 
|  | if (sscanf(optarg, "%i%c", &Validation_Flags, &cc) != 1) { | 
|  | fprintf(stderr, "%s:  Invalid -V argument (%s) - must be a decimal, hex, or octal\n", Prog, optarg); | 
|  | fprintf(stderr, "    number, or one of the following strings:  'sync',\n"); | 
|  | fprintf(stderr, "    'buffered', 'parallel', 'ldraw', or 'raw'\n"); | 
|  | exit(E_USAGE); | 
|  | } | 
|  | } | 
|  | V_opt++; | 
|  | break; | 
|  | case 'U': | 
|  | tok = strtok(optarg, ","); | 
|  | while (tok != NULL) { | 
|  | for (s = Upanic_Args; s->string != NULL; s++) | 
|  | if (strcmp(s->string, tok) == 0) | 
|  | break; | 
|  |  | 
|  | if (s->string == NULL) { | 
|  | fprintf(stderr, | 
|  | "%s%s:  Illegal -U arg (%s).  Must be one of: ", | 
|  | Prog, TagName, tok); | 
|  |  | 
|  | for (s = Upanic_Args; s->string != NULL; s++) | 
|  | fprintf(stderr, "%s ", s->string); | 
|  |  | 
|  | fprintf(stderr, "\n"); | 
|  |  | 
|  | exit(1); | 
|  | } | 
|  |  | 
|  | Upanic_Conditions |= s->value; | 
|  | tok = strtok(NULL, ","); | 
|  | } | 
|  |  | 
|  | U_opt++; | 
|  | break; | 
|  |  | 
|  | case '?': | 
|  | usage(stderr); | 
|  | exit(E_USAGE); | 
|  | break; | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Supply defaults | 
|  | */ | 
|  |  | 
|  | if (! C_opt) { | 
|  | Data_Fill = doio_pat_fill; | 
|  | Data_Check = doio_pat_check; | 
|  | } | 
|  |  | 
|  | if (! U_opt) | 
|  | Upanic_Conditions = 0; | 
|  |  | 
|  | if (! n_opt) | 
|  | Nprocs = 1; | 
|  |  | 
|  | if (! r_opt) | 
|  | Release_Interval = DEF_RELEASE_INTERVAL; | 
|  |  | 
|  | if (! M_opt) { | 
|  | Memalloc[Nmemalloc].memtype = MEM_DATA; | 
|  | Memalloc[Nmemalloc].flags = 0; | 
|  | Memalloc[Nmemalloc].name = NULL; | 
|  | Memalloc[Nmemalloc].space = NULL; | 
|  | Nmemalloc++; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Initialize input stream | 
|  | */ | 
|  |  | 
|  | if (argc == optind) { | 
|  | Infile = NULL; | 
|  | } else { | 
|  | Infile = argv[optind++]; | 
|  | } | 
|  |  | 
|  | if (argc != optind) { | 
|  | usage(stderr); | 
|  | exit(E_USAGE); | 
|  | } | 
|  |  | 
|  | return 0; | 
|  | } | 
|  |  | 
|  |  | 
|  |  | 
|  | /* | 
|  | * Parse memory allocation types | 
|  | * | 
|  | * Types are: | 
|  | *  Data | 
|  | *  T3E-shmem:blksize[:nblks] | 
|  | *  SysV-shmem:shmid:blksize:nblks | 
|  | *	if shmid is "private", use IPC_PRIVATE | 
|  | *	and nblks is not required | 
|  | * | 
|  | *  mmap:flags:filename:blksize[:nblks] | 
|  | *   flags are one of: | 
|  | *	p - private (MAP_PRIVATE) | 
|  | *	a - private, MAP_AUTORESRV | 
|  | *	l - local (MAP_LOCAL) | 
|  | *	s - shared (nblks required) | 
|  | * | 
|  | *   plus any of: | 
|  | *	f - fixed address (MAP_FIXED) | 
|  | *	A - use an address without MAP_FIXED | 
|  | *	a - autogrow (map once at startup) | 
|  | * | 
|  | *  mmap:flags:devzero | 
|  | *	mmap /dev/zero  (shared not allowd) | 
|  | *	maps the first 4096 bytes of /dev/zero | 
|  | * | 
|  | * - put a directory at the beginning of the shared | 
|  | *   regions saying what pid has what region. | 
|  | *	DIRMAGIC | 
|  | *	BLKSIZE | 
|  | *	NBLKS | 
|  | *	nblks worth of directories - 1 int pids | 
|  | */ | 
|  | void | 
|  | parse_memalloc(char *arg) | 
|  | { | 
|  | char		*allocargs[NMEMALLOC]; | 
|  | int		nalloc; | 
|  | struct memalloc	*M; | 
|  |  | 
|  | if(Nmemalloc >= NMEMALLOC) { | 
|  | doio_fprintf(stderr, "Error - too many memory types (%d).\n", | 
|  | Nmemalloc); | 
|  | return; | 
|  | } | 
|  |  | 
|  | M = &Memalloc[Nmemalloc]; | 
|  |  | 
|  | nalloc = string_to_tokens(arg, allocargs, 32, ":"); | 
|  | if(!strcmp(allocargs[0], "data")) { | 
|  | M->memtype = MEM_DATA; | 
|  | M->flags = 0; | 
|  | M->name = NULL; | 
|  | M->space = NULL; | 
|  | Nmemalloc++; | 
|  | if(nalloc >= 2) { | 
|  | if(strchr(allocargs[1], 'p')) | 
|  | M->flags |= MEMF_MPIN; | 
|  | } | 
|  | } else if(!strcmp(allocargs[0], "mmap")) { | 
|  | /* mmap:flags:filename[:size] */ | 
|  | M->memtype = MEM_MMAP; | 
|  | M->flags = 0; | 
|  | M->space = NULL; | 
|  | if(nalloc >= 1) { | 
|  | if(strchr(allocargs[1], 'p')) | 
|  | M->flags |= MEMF_PRIVATE; | 
|  | if(strchr(allocargs[1], 'a')) | 
|  | M->flags |= MEMF_AUTORESRV; | 
|  | if(strchr(allocargs[1], 'l')) | 
|  | M->flags |= MEMF_LOCAL; | 
|  | if(strchr(allocargs[1], 's')) | 
|  | M->flags |= MEMF_SHARED; | 
|  |  | 
|  | if(strchr(allocargs[1], 'f')) | 
|  | M->flags |= MEMF_FIXADDR; | 
|  | if(strchr(allocargs[1], 'A')) | 
|  | M->flags |= MEMF_ADDR; | 
|  | if(strchr(allocargs[1], 'G')) | 
|  | M->flags |= MEMF_AUTOGROW; | 
|  |  | 
|  | if(strchr(allocargs[1], 'U')) | 
|  | M->flags |= MEMF_FILE; | 
|  | } else { | 
|  | M->flags |= MEMF_PRIVATE; | 
|  | } | 
|  |  | 
|  | if(nalloc > 2) { | 
|  | if(!strcmp(allocargs[2], "devzero")) { | 
|  | M->name = "/dev/zero"; | 
|  | if(M->flags & | 
|  | ((MEMF_PRIVATE|MEMF_LOCAL) == 0)) | 
|  | M->flags |= MEMF_PRIVATE; | 
|  | } else { | 
|  | M->name = allocargs[2]; | 
|  | } | 
|  | } else { | 
|  | M->name = "/dev/zero"; | 
|  | if(M->flags & | 
|  | ((MEMF_PRIVATE|MEMF_LOCAL) == 0)) | 
|  | M->flags |= MEMF_PRIVATE; | 
|  | } | 
|  | Nmemalloc++; | 
|  |  | 
|  | } else if(!strcmp(allocargs[0], "shmem")) { | 
|  | /* shmem:shmid:size */ | 
|  | M->memtype = MEM_SHMEM; | 
|  | M->flags = 0; | 
|  | M->space = NULL; | 
|  | if(nalloc >= 2) { | 
|  | M->name = allocargs[1]; | 
|  | } else { | 
|  | M->name = NULL; | 
|  | } | 
|  | if(nalloc >= 3) { | 
|  | sscanf(allocargs[2], "%i", &M->nblks); | 
|  | } else { | 
|  | M->nblks = 0; | 
|  | } | 
|  | if(nalloc >= 4) { | 
|  | if(strchr(allocargs[3], 'p')) | 
|  | M->flags |= MEMF_MPIN; | 
|  | } | 
|  |  | 
|  | Nmemalloc++; | 
|  | } else { | 
|  | doio_fprintf(stderr, "Error - unknown memory type '%s'.\n", | 
|  | allocargs[0]); | 
|  | exit(1); | 
|  | } | 
|  | } | 
|  |  | 
|  | void | 
|  | dump_memalloc() | 
|  | { | 
|  | int	ma; | 
|  | char	*mt; | 
|  |  | 
|  | if(Nmemalloc == 0) { | 
|  | printf("No memory allocation strategies devined\n"); | 
|  | return; | 
|  | } | 
|  |  | 
|  | for(ma=0; ma < Nmemalloc; ma++) { | 
|  | switch(Memalloc[ma].memtype) { | 
|  | case MEM_DATA:	mt = "data";	break; | 
|  | case MEM_SHMEM:	mt = "shmem";	break; | 
|  | case MEM_MMAP:	mt = "mmap";	break; | 
|  | default:	mt = "unknown";	break; | 
|  | } | 
|  | printf("mstrat[%d] = %d %s\n", ma, Memalloc[ma].memtype, mt); | 
|  | printf("\tflags=%#o name='%s' nblks=%d\n", | 
|  | Memalloc[ma].flags, | 
|  | Memalloc[ma].name, | 
|  | Memalloc[ma].nblks); | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * -d <op>:<time> - doio inter-operation delay | 
|  | *	currently this permits ONE type of delay between operations. | 
|  | */ | 
|  |  | 
|  | void | 
|  | parse_delay(char *arg) | 
|  | { | 
|  | char		*delayargs[NMEMALLOC]; | 
|  | int		ndelay; | 
|  | struct smap	*s; | 
|  |  | 
|  | ndelay = string_to_tokens(arg, delayargs, 32, ":"); | 
|  | if(ndelay < 2) { | 
|  | doio_fprintf(stderr, | 
|  | "Illegal delay arg (%s). Must be operation:time\n", arg); | 
|  | exit(1); | 
|  | } | 
|  | for(s=delaymap; s->string != NULL; s++) | 
|  | if(!strcmp(s->string, delayargs[0])) | 
|  | break; | 
|  | if (s->string == NULL) { | 
|  | fprintf(stderr, | 
|  | "Illegal Delay arg (%s).  Must be one of: ", arg); | 
|  |  | 
|  | for (s = delaymap; s->string != NULL; s++) | 
|  | fprintf(stderr, "%s ", s->string); | 
|  | fprintf(stderr, "\n"); | 
|  | exit(1); | 
|  | } | 
|  |  | 
|  | delayop = s->value; | 
|  |  | 
|  | sscanf(delayargs[1], "%i", &delaytime); | 
|  |  | 
|  | if(ndelay > 2) { | 
|  | fprintf(stderr, | 
|  | "Warning: extra delay arguments ignored.\n"); | 
|  | } | 
|  | } | 
|  |  | 
|  |  | 
|  | /* | 
|  | * Usage clause - obvious | 
|  | */ | 
|  |  | 
|  | int | 
|  | usage(stream) | 
|  | FILE	*stream; | 
|  | { | 
|  | /* | 
|  | * Only do this if we are on vpe 0, to avoid seeing it from every | 
|  | * process in the application. | 
|  | */ | 
|  |  | 
|  | if (Npes > 1 && Vpe != 0) { | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | fprintf(stream, "usage%s:  %s [-aekv] [-m message_interval] [-n nprocs] [-r release_interval] [-w write_log] [-V validation_ftype] [-U upanic_cond] [infile]\n", TagName, Prog); | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | void | 
|  | help(stream) | 
|  | FILE	*stream; | 
|  | { | 
|  | /* | 
|  | * Only the app running on vpe 0 gets to issue help - this prevents | 
|  | * everybody in the application from doing this. | 
|  | */ | 
|  |  | 
|  | if (Npes > 1 && Vpe != 0) { | 
|  | return; | 
|  | } | 
|  |  | 
|  | usage(stream); | 
|  | fprintf(stream, "\n"); | 
|  | fprintf(stream, "\t-a                   abort - kill all doio processes on data compare\n"); | 
|  | fprintf(stream, "\t                     errors.  Normally only the erroring process exits\n"); | 
|  | fprintf(stream, "\t-C data-pattern-type \n"); | 
|  | fprintf(stream, "\t                     Available data patterns are:\n"); | 
|  | fprintf(stream, "\t                     default - repeating pattern\n"); | 
|  | fprintf(stream, "\t-d Operation:Time    Inter-operation delay.\n"); | 
|  | fprintf(stream, "\t                     Operations are:\n"); | 
|  | fprintf(stream, "\t                         select:time (1 second=1000000)\n"); | 
|  | fprintf(stream, "\t                         sleep:time (1 second=1)\n"); | 
|  | fprintf(stream, "\t                         alarm:time (1 second=1)\n"); | 
|  | fprintf(stream, "\t-e                   Re-exec children before entering the main\n"); | 
|  | fprintf(stream, "\t                     loop.  This is useful for spreading\n"); | 
|  | fprintf(stream, "\t                     procs around on multi-pe systems.\n"); | 
|  | fprintf(stream, "\t-k                   Lock file regions during writes using fcntl()\n"); | 
|  | fprintf(stream, "\t-v                   Verify writes - this is done by doing a buffered\n"); | 
|  | fprintf(stream, "\t                     read() of the data if file io was done, or\n"); | 
|  | fprintf(stream, "\t                     an ssread()of the data if sds io was done\n"); | 
|  | fprintf(stream, "\t-M                   Data buffer allocation method\n"); | 
|  | fprintf(stream, "\t                     alloc-type[,type]\n"); | 
|  | fprintf(stream, "\t			    data\n"); | 
|  | fprintf(stream, "\t			    shmem:shmid:size\n"); | 
|  | fprintf(stream, "\t			    mmap:flags:filename\n"); | 
|  | fprintf(stream, "\t			        p - private\n"); | 
|  | fprintf(stream, "\t			        s - shared (shared file must exist\n"), | 
|  | fprintf(stream, "\t			            and have needed length)\n"); | 
|  | fprintf(stream, "\t			        f - fixed address (not used)\n"); | 
|  | fprintf(stream, "\t			        a - specify address (not used)\n"); | 
|  | fprintf(stream, "\t			        U - Unlink file when done\n"); | 
|  | fprintf(stream, "\t			        The default flag is private\n"); | 
|  | fprintf(stream, "\n"); | 
|  | fprintf(stream, "\t-m message_interval  Generate a message every 'message_interval'\n"); | 
|  | fprintf(stream, "\t                     requests.  An interval of 0 suppresses\n"); | 
|  | fprintf(stream, "\t                     messages.  The default is 0.\n"); | 
|  | fprintf(stream, "\t-N tagname           Tag name, for Monster.\n"); | 
|  | fprintf(stream, "\t-n nprocs            # of processes to start up\n"); | 
|  | fprintf(stream, "\t-r release_interval  Release all memory and close\n"); | 
|  | fprintf(stream, "\t                     files every release_interval operations.\n"); | 
|  | fprintf(stream, "\t                     By default procs never release memory\n"); | 
|  | fprintf(stream, "\t                     or close fds unless they have to.\n"); | 
|  | fprintf(stream, "\t-V validation_ftype  The type of file descriptor to use for doing data\n"); | 
|  | fprintf(stream, "\t                     validation.  validation_ftype may be an octal,\n"); | 
|  | fprintf(stream, "\t                     hex, or decimal number representing the open()\n"); | 
|  | fprintf(stream, "\t                     flags, or may be one of the following strings:\n"); | 
|  | fprintf(stream, "\t                     'buffered' - validate using bufferd read\n"); | 
|  | fprintf(stream, "\t                     'sync'     - validate using O_SYNC read\n"); | 
|  | fprintf(stream, "\t                     'direct    - validate using O_DIRECT read'\n"); | 
|  | fprintf(stream, "\t                     By default, 'parallel'\n"); | 
|  | fprintf(stream, "\t                     is used if the write was done with O_PARALLEL\n"); | 
|  | fprintf(stream, "\t                     or 'buffered' for all other writes.\n"); | 
|  | fprintf(stream, "\t-w write_log         File to log file writes to.  The doio_check\n"); | 
|  | fprintf(stream, "\t                     program can reconstruct datafiles using the\n"); | 
|  | fprintf(stream, "\t                     write_log, and detect if a file is corrupt\n"); | 
|  | fprintf(stream, "\t                     after all procs have exited.\n"); | 
|  | fprintf(stream, "\t-U upanic_cond       Comma separated list of conditions that will\n"); | 
|  | fprintf(stream, "\t                     cause a call to upanic(PA_PANIC).\n"); | 
|  | fprintf(stream, "\t                     'corruption' -> upanic on bad data comparisons\n"); | 
|  | fprintf(stream, "\t                     'iosw'     ---> upanic on unexpected async iosw\n"); | 
|  | fprintf(stream, "\t                     'rval'     ---> upanic on unexpected syscall rvals\n"); | 
|  | fprintf(stream, "\t                     'all'      ---> all of the above\n"); | 
|  | fprintf(stream, "\n"); | 
|  | fprintf(stream, "\tinfile               Input stream - default is stdin - must be a list\n"); | 
|  | fprintf(stream, "\t                     of io_req structures (see doio.h).  Currently\n"); | 
|  | fprintf(stream, "\t                     only the iogen program generates the proper\n"); | 
|  | fprintf(stream, "\t                     format\n"); | 
|  | } | 
|  |  |