| /* |
| * Copyright (c) 2003-2005 Silicon Graphics, Inc. |
| * All Rights Reserved. |
| * |
| * This program is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU General Public License as |
| * published by the Free Software Foundation. |
| * |
| * This program is distributed in the hope that it would be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program; if not, write the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA |
| */ |
| |
| #include <pthread.h> |
| #include "platform_defs.h" |
| #include "command.h" |
| #include "input.h" |
| #include "init.h" |
| #include "io.h" |
| |
| char *progname; |
| int exitcode; |
| int expert; |
| int idlethread; |
| size_t pagesize; |
| struct timeval stopwatch; |
| |
| void |
| usage(void) |
| { |
| fprintf(stderr, |
| _("Usage: %s [-adfinrRstVx] [-m mode] [-p prog] [[-c|-C] cmd]... file\n"), |
| progname); |
| exit(1); |
| } |
| |
| void |
| init_cvtnum( |
| size_t *blocksize, |
| size_t *sectsize) |
| { |
| if (!file || (file->flags & IO_FOREIGN)) { |
| *blocksize = 4096; |
| *sectsize = 512; |
| } else { |
| *blocksize = file->geom.blocksize; |
| *sectsize = file->geom.sectsize; |
| } |
| } |
| |
| static void |
| init_commands(void) |
| { |
| attr_init(); |
| bmap_init(); |
| copy_range_init(); |
| cowextsize_init(); |
| encrypt_init(); |
| fadvise_init(); |
| fiemap_init(); |
| file_init(); |
| flink_init(); |
| freeze_init(); |
| fsync_init(); |
| getrusage_init(); |
| help_init(); |
| imap_init(); |
| inject_init(); |
| madvise_init(); |
| mincore_init(); |
| mmap_init(); |
| open_init(); |
| parent_init(); |
| pread_init(); |
| prealloc_init(); |
| pwrite_init(); |
| quit_init(); |
| readdir_init(); |
| reflink_init(); |
| resblks_init(); |
| seek_init(); |
| sendfile_init(); |
| shutdown_init(); |
| sync_init(); |
| sync_range_init(); |
| truncate_init(); |
| utimes_init(); |
| } |
| |
| /* |
| * This allows xfs_io commands specified on the command line to be run on every |
| * open file in the file table. Commands that should not be iterated across all |
| * open files need to specify CMD_FLAG_ONESHOT in their command flags. |
| */ |
| static int |
| filetable_iterator( |
| int index) |
| { |
| if (index >= filecount) |
| return 0; |
| file = &filetable[index++]; |
| return index; |
| } |
| |
| static int |
| init_check_command( |
| const cmdinfo_t *ct) |
| { |
| if (!file && !(ct->flags & CMD_NOFILE_OK)) { |
| fprintf(stderr, _("no files are open, try 'help open'\n")); |
| return 0; |
| } |
| if (!mapping && !(ct->flags & CMD_NOMAP_OK)) { |
| fprintf(stderr, _("no mapped regions, try 'help mmap'\n")); |
| return 0; |
| } |
| if (file && !(ct->flags & CMD_FOREIGN_OK) && |
| (file->flags & IO_FOREIGN)) { |
| fprintf(stderr, |
| _("foreign file active, %s command is for XFS filesystems only\n"), |
| ct->name); |
| return 0; |
| } |
| return 1; |
| } |
| |
| void |
| init( |
| int argc, |
| char **argv) |
| { |
| int c, flags = 0; |
| char *sp; |
| mode_t mode = 0600; |
| xfs_fsop_geom_t geometry = { 0 }; |
| |
| progname = basename(argv[0]); |
| setlocale(LC_ALL, ""); |
| bindtextdomain(PACKAGE, LOCALEDIR); |
| textdomain(PACKAGE); |
| |
| pagesize = getpagesize(); |
| gettimeofday(&stopwatch, NULL); |
| |
| while ((c = getopt(argc, argv, "ac:C:dFfim:p:nrRstTVx")) != EOF) { |
| switch (c) { |
| case 'a': |
| flags |= IO_APPEND; |
| break; |
| case 'c': |
| add_user_command(optarg); |
| break; |
| case 'C': |
| add_oneshot_user_command(optarg); |
| break; |
| case 'd': |
| flags |= IO_DIRECT; |
| break; |
| case 'F': |
| /* Ignored / deprecated now, handled automatically */ |
| break; |
| case 'f': |
| flags |= IO_CREAT; |
| break; |
| case 'i': |
| idlethread = 1; |
| break; |
| case 'm': |
| mode = strtoul(optarg, &sp, 0); |
| if (!sp || sp == optarg) { |
| fprintf(stderr, _("non-numeric mode -- %s\n"), |
| optarg); |
| exit(1); |
| } |
| break; |
| case 'n': |
| flags |= IO_NONBLOCK; |
| break; |
| case 'p': |
| progname = optarg; |
| break; |
| case 'r': |
| flags |= IO_READONLY; |
| break; |
| case 's': |
| flags |= IO_OSYNC; |
| break; |
| case 't': |
| flags |= IO_TRUNC; |
| break; |
| case 'R': |
| flags |= IO_REALTIME; |
| break; |
| case 'T': |
| flags |= IO_TMPFILE; |
| break; |
| case 'x': |
| expert = 1; |
| break; |
| case 'V': |
| printf(_("%s version %s\n"), progname, VERSION); |
| exit(0); |
| default: |
| usage(); |
| } |
| } |
| |
| while (optind < argc) { |
| if ((c = openfile(argv[optind], &geometry, flags, mode)) < 0) |
| exit(1); |
| if (!platform_test_xfs_fd(c)) |
| flags |= IO_FOREIGN; |
| if (addfile(argv[optind], c, &geometry, flags) < 0) |
| exit(1); |
| optind++; |
| } |
| |
| init_commands(); |
| add_command_iterator(filetable_iterator); |
| add_check_command(init_check_command); |
| } |
| |
| /* |
| * The purpose of this idle thread is to test io from a multi threaded process. |
| * With single threaded process, the file table is not shared and file structs |
| * are not reference counted. Spawning an idle thread can help detecting file |
| * struct reference leaks. |
| */ |
| void * |
| idle_loop(void *arg) |
| { |
| for (;;) |
| pause(); |
| } |
| |
| void |
| start_idle_thread() |
| { |
| pthread_t t; |
| |
| if (pthread_create(&t, NULL, idle_loop, NULL)) { |
| fprintf(stderr, "Error creating idle thread\n"); |
| exit(1); |
| } |
| } |
| |
| int |
| main( |
| int argc, |
| char **argv) |
| { |
| init(argc, argv); |
| if (idlethread) |
| start_idle_thread(); |
| command_loop(); |
| return exitcode; |
| } |