X-Git-Url: https://git.whamcloud.com/?a=blobdiff_plain;ds=sidebyside;f=lustre%2Futils%2Floadgen.c;h=e49c887d935d436f867ef0e4ea1456b905a053f7;hb=adde80ff;hp=2e57ccd88b95b1d0ffb654596cf74f2e989aaa1f;hpb=bac5e4db725c00141ff70ab1effdbb1d0fbee9e6;p=fs%2Flustre-release.git diff --git a/lustre/utils/loadgen.c b/lustre/utils/loadgen.c index 2e57ccd..e49c887 100644 --- a/lustre/utils/loadgen.c +++ b/lustre/utils/loadgen.c @@ -1,30 +1,46 @@ /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*- * vim:expandtab:shiftwidth=8:tabstop=8: * - * Copyright (C) 2006 Cluster File Systems, Inc. - * Author: Nathan Rutman + * GPL HEADER START * - * This file is part of Lustre, http://www.lustre.org. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * - * Lustre is free software; you can redistribute it and/or - * modify it under the terms of version 2 of the GNU General Public - * License as published by the Free Software Foundation. + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 2 only, + * as published by the Free Software Foundation. * - * Lustre is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License version 2 for more details (a copy is included + * in the LICENSE file that accompanied this code). * - * You should have received a copy of the GNU General Public License - * along with Lustre; if not, write to the Free Software - * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + * You should have received a copy of the GNU General Public License + * version 2 along with this program; If not, see [sun.com URL with a + * copy of GPLv2]. + * + * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, + * CA 95054 USA or visit www.sun.com if you need additional information or + * have any questions. + * + * GPL HEADER END + */ +/* + * Copyright 2008 Sun Microsystems, Inc. All rights reserved + * Use is subject to license terms. + */ +/* + * This file is part of Lustre, http://www.lustre.org/ + * Lustre is a trademark of Sun Microsystems, Inc. + * + * lustre/utils/loadgen.c * - * loadgen.c * See how many local OSCs we can start whaling on a OST * We're doing direct ioctls instead of going though a system() call to lctl * to avoid the bash overhead. * Adds an osc / echo client pair in each thread and starts echo transactions. * + * Author: Nathan Rutman */ #include @@ -42,7 +58,7 @@ #include #include -#include "parser.h" +#include #include "obdctl.h" static char cmdname[512]; @@ -51,6 +67,8 @@ char nid[64] = ""; static int live_threads = 0; static int sig_received = 0; static int o_verbose = 4; /* 0-5 */ +static int my_oss = 0; +static int my_ecs = 0; static int jt_quit(int argc, char **argv) { Parser_quit(argc, argv); @@ -60,7 +78,7 @@ static int jt_quit(int argc, char **argv) { static int loadgen_usage(int argc, char **argv) { if (argc == 1) { - fprintf(stderr, + fprintf(stderr, "This is a test program used to simulate large numbers of\n" "clients. The echo obds are used, so the obdecho module must\n" "be loaded.\n" @@ -76,7 +94,9 @@ static int loadgen_usage(int argc, char **argv) static int loadgen_verbose(int argc, char **argv); static int loadgen_target(int argc, char **argv); +static int loadgen_start_echosrv(int argc, char **argv); static int loadgen_start_clients(int argc, char **argv); +static int loadgen_wait(int argc, char **argv); static int loadgen_write(int argc, char **argv); command_t cmdlist[] = { @@ -85,12 +105,13 @@ command_t cmdlist[] = { "usage: device []"}, {"dl", jt_obd_list, 0, "show all devices\n" "usage: dl"}, - {"start", loadgen_start_clients, 0, - "set up echo clients\n" + {"echosrv", loadgen_start_echosrv, 0, "start an echo server\n"}, + {"start", loadgen_start_clients, 0, "set up echo clients\n" "usage: start_clients "}, - {"verbose", loadgen_verbose, 0, - "set verbosity level 0-5\n" + {"verbose", loadgen_verbose, 0, "set verbosity level 0-5\n" "usage: verbose "}, + {"wait", loadgen_wait, 0, + "wait for all threads to finish\n"}, {"write", loadgen_write, 0, "start a test_brw write test on X clients for Y iterations\n" "usage: write []"}, @@ -111,7 +132,7 @@ command_t cmdlist[] = { struct command_t { int c_flags; - int c_count; + int c_rpt; int c_delay; }; @@ -145,20 +166,26 @@ static struct kid_t *push_kid(int tnum) int trigger_count = 0; int waiting_count = 0; int timer_on = 0; +int all_done = 1; struct timeval trigger_start; -struct command_t *trigger_cmd = NULL; +struct command_t trigger_cmd; pthread_mutex_t m_trigger = PTHREAD_MUTEX_INITIALIZER; pthread_cond_t cv_trigger = PTHREAD_COND_INITIALIZER; -__u64 write_bytes; +unsigned long long write_bytes; pthread_mutex_t m_count = PTHREAD_MUTEX_INITIALIZER; -static void trigger(struct command_t *cmd, int count) +static void trigger(int command, int threads, int repeat, int delay) { pthread_mutex_lock(&m_trigger); - trigger_cmd = cmd; - trigger_count = count; + trigger_cmd.c_flags = command; + trigger_cmd.c_rpt = repeat; + trigger_cmd.c_delay = delay; + trigger_count = threads; + if (o_verbose > 4) + printf("trigger %d cmd c=%d f=%x\n", trigger_count, + trigger_cmd.c_rpt, trigger_cmd.c_flags); gettimeofday(&trigger_start, NULL); timer_on = 1; pthread_mutex_lock(&m_count); @@ -176,12 +203,10 @@ static __inline__ void stop_all(int unused) static void kill_kids(void) { - struct command_t cmd; struct kid_t *tmp = kid_list; stop_all(SIGINT); - cmd.c_flags = C_STOP; - trigger(&cmd, 0); + trigger(C_STOP, 0, 0, 0); while(tmp) { pthread_kill(tmp->k_pthread, SIGTERM); tmp = tmp->k_next; @@ -239,7 +264,7 @@ static int write_proc(char *proc_path, char *value) return rc; } -static int read_proc(char *proc_path, long long *value) +static int read_proc(char *proc_path, unsigned long long *value) { int fd, rc; char buf[50]; @@ -253,9 +278,13 @@ static int read_proc(char *proc_path, long long *value) rc = read(fd, buf, sizeof(buf)); close(fd); + if (errno == EOPNOTSUPP) { + /* probably an echo server */ + return rc; + } if (rc <= 0) { fprintf(stderr, "read('%s') failed: %s (%d)\n", - proc_path, strerror(errno), rc); + proc_path, strerror(errno), errno); return rc; } *value = strtoull(buf, NULL, 10); @@ -326,7 +355,7 @@ static int cleanup(char *obdname, int quiet) return rc; } -static int setup(char *oname, char *ename, int *dev) +static int echocli_setup(char *oname, char *ename, int *dev) { char *args[5]; char proc_path[50]; @@ -448,6 +477,7 @@ static int obj_create(struct kid_t *kid) data.ioc_dev = kid->k_dev; data.ioc_obdo1.o_mode = 0100644; data.ioc_obdo1.o_id = 0; + data.ioc_obdo1.o_gr = 2; data.ioc_obdo1.o_uid = 0; data.ioc_obdo1.o_gid = 0; data.ioc_obdo1.o_valid = OBD_MD_FLTYPE | OBD_MD_FLMODE | @@ -461,7 +491,7 @@ static int obj_create(struct kid_t *kid) } if (!(data.ioc_obdo1.o_valid & OBD_MD_FLID)) { - fprintf(stderr,"%d: create oid not valid "LPX64"\n", + fprintf(stderr, "%d: create oid not valid "LPX64"\n", kid->k_id, data.ioc_obdo1.o_valid); return rc; } @@ -481,7 +511,7 @@ static int obj_delete(struct kid_t *kid) int rc; if (o_verbose > 4) - printf("%d: del "LPU64"\n", kid->k_id, kid->k_objid); + printf("%d: del "LPX64"\n", kid->k_id, kid->k_objid); memset(&data, 0, sizeof(data)); data.ioc_dev = kid->k_dev; @@ -491,7 +521,7 @@ static int obj_delete(struct kid_t *kid) rc = obj_ioctl(OBD_IOC_DESTROY, &data, 1); if (rc) - fprintf(stderr, "%s-%d: can't destroy obj "LPU64" (%d)\n", + fprintf(stderr, "%s-%d: can't destroy obj "LPX64" (%d)\n", cmdname, kid->k_id, kid->k_objid, rc); kid->k_objid = 0; @@ -580,7 +610,7 @@ static int do_work(struct kid_t *kid) if (!(kid->k_cmd.c_flags & C_CREATE_EVERY)) rc = obj_create(kid); - for (iter = 0; iter < kid->k_cmd.c_count; iter++) { + for (iter = 0; iter < kid->k_cmd.c_rpt; iter++) { if (rc || sig_received) break; @@ -631,7 +661,6 @@ static void report_perf() (write_bytes >> 20) / diff); pthread_mutex_unlock(&m_count); } - timer_on = 0; } static void *run_one_child(void *threadvp) @@ -646,7 +675,7 @@ static void *run_one_child(void *threadvp) sprintf(oname, "o%.5d", thread); sprintf(ename, "e%.5d", thread); - rc = setup(oname, ename, &dev); + rc = echocli_setup(oname, ename, &dev); if (rc) { fprintf(stderr, "%s: can't setup '%s/%s' (%d)\n", cmdname, oname, ename, rc); @@ -663,11 +692,14 @@ static void *run_one_child(void *threadvp) while(!(rc || sig_received)) { pthread_mutex_lock(&m_trigger); waiting_count++; - if ((waiting_count == live_threads) && timer_on) + if ((waiting_count == live_threads) && timer_on) { report_perf(); - + timer_on = 0; + all_done = 1; + } pthread_cond_wait(&cv_trigger, &m_trigger); waiting_count--; + all_done = 0; /* First trigger_count threads will do the work, the rest will block again */ @@ -675,9 +707,9 @@ static void *run_one_child(void *threadvp) if (o_verbose > 4) printf("%d: trigger %d cmd %x\n", kid->k_id, trigger_count, - trigger_cmd->c_flags); + trigger_cmd.c_flags); trigger_count--; - kid->k_cmd = *trigger_cmd; + memcpy(&kid->k_cmd, &trigger_cmd, sizeof(trigger_cmd)); pthread_mutex_unlock(&m_trigger); rc = do_work(kid); } else { @@ -714,14 +746,14 @@ static int loadgen_start_clients(int argc, char **argv) return CMD_HELP; if (!target[0]) { - fprintf(stderr,"%s: target OST is not defined, use 'device' " + fprintf(stderr, "%s: target OST is not defined, use 'device' " "command\n", cmdname); return -EINVAL; } rc = pthread_attr_init(&attr); if (rc) { - fprintf(stderr,"%s: pthread_attr_init:(%d) %s\n", + fprintf(stderr, "%s: pthread_attr_init:(%d) %s\n", cmdname, rc, strerror(errno)); return -errno; } @@ -825,7 +857,6 @@ static int loadgen_verbose(int argc, char **argv) static int loadgen_write(int argc, char **argv) { - struct command_t cmd; int threads; if (argc < 3 || argc > 4) @@ -837,11 +868,99 @@ static int loadgen_write(int argc, char **argv) threads, live_threads); return -EOVERFLOW; } - cmd.c_flags = C_WRITE; - cmd.c_count = atoi(argv[2]); - if (argc == 4) - cmd.c_delay = atoi(argv[3]); - trigger(&cmd, threads); + trigger(C_WRITE, threads, atoi(argv[2]), + (argc == 4) ? atoi(argv[3]) : 0); + return 0; +} + +char ecsname[] = "echosrv"; +static int loadgen_stop_echosrv(int argc, char **argv) +{ + int verbose = (argc != 9); + if (my_oss) { + char name[]="OSS"; + cleanup(name, verbose); + my_oss = 0; + } + if (my_ecs || (argc == 9)) { + cleanup(ecsname, verbose); + my_ecs = 0; + } + return 0; +} + +static int loadgen_start_echosrv(int argc, char **argv) +{ + char *args[5]; + int rc; + + pthread_mutex_lock(&m_config); + + args[0] = cmdname; + + /* attach obdecho echosrv echosrv_UUID */ + args[1] = "obdecho"; + args[2] = args[3] = ecsname; + rc = jt_lcfg_attach(4, args); + if (rc) { + fprintf(stderr, "%s: can't attach echo server (%d)\n", + cmdname, rc); + /* Assume we want e.g. an old one cleaned anyhow. */ + goto clean; + } + my_ecs = 1; + + /* setup */ + rc = jt_lcfg_setup(1, args); + if (rc) { + fprintf(stderr, "%s: can't setup echo server (%d)\n", + cmdname, rc); + goto clean; + } + + /* Create an OSS to handle the communications */ + /* attach ost OSS OSS_UUID */ + args[1] = "ost"; + args[2] = args[3] = "OSS"; + + rc = jt_lcfg_attach(4, args); + if (rc == EEXIST) { + /* Already set up for somebody else, that's fine. */ + printf("OSS already set up, no problem.\n"); + pthread_mutex_unlock(&m_config); + return 0; + } + if (rc) { + fprintf(stderr, "%s: can't attach OSS (%d)\n", + cmdname, rc); + goto clean; + } + my_oss = 1; + + /* setup */ + rc = jt_lcfg_setup(1, args); + if (rc) { + fprintf(stderr, "%s: can't setup OSS (%d)\n", + cmdname, rc); + goto clean; + } + + pthread_mutex_unlock(&m_config); + return rc; + +clean: + pthread_mutex_unlock(&m_config); + loadgen_stop_echosrv(9, argv); + return rc; +} + +static int loadgen_wait(int argc, char **argv) +{ + /* Give scripts a chance to start some threads */ + sleep(1); + while (!all_done) { + sleep(1); + } return 0; } @@ -877,9 +996,13 @@ static int loadgen_init(int argc, char **argv) static int loadgen_exit() { int rc; + printf("stopping %d children\n", live_threads); kill_kids(); rc = wait_for_threads(); + + loadgen_stop_echosrv(0, NULL); + return rc; } @@ -889,6 +1012,8 @@ static int loadgen_main(int argc, char **argv) int rc; setlinebuf(stdout); + /* without this threaded errors cause segfault */ + setlinebuf(stderr); if ((rc = ptl_initialize(argc, argv)) < 0) exit(rc); @@ -926,4 +1051,3 @@ int main (int argc, char **argv) return rc; } #endif -