lustre/tests/copytool.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 only,
  10  * as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License version 2 for more details (a copy is included
  16  * in the LICENSE file that accompanied this code).
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * version 2 along with this program; If not, see
  20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  21  *
  22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  23  * CA 95054 USA or visit www.sun.com if you need additional information or
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright  2009 Sun Microsystems, Inc. All rights reserved
  30  * Use is subject to license terms.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  *
  36  * Author: Nathan Rutman <nathan.rutman@sun.com>
  37  *
  38  */
  39
  40 /* HSM copytool example program.
  41  * The copytool acts on action requests from Lustre to copy files to and from
  42  * an HSM archive system.
  43  *
  44  * Note: under Linux, until llapi_copytool_fini is called (or the program is
  45  * killed), the libcfs module will be referenced and unremovable,
  46  * even after Lustre services stop.
  47  */
  48
  49 #include <stdio.h>
  50 #include <getopt.h>
  51 #include <libcfs/libcfs.h>
  52 #include <lustre/lustre_user.h>
  53 #include <lustre/liblustreapi.h>
  54
  55 int main(int argc, char **argv) {
  56         int c, test = 0;
  57         struct option long_opts[] = {
  58                 {"test", no_argument, 0, 't'},
  59                 {0, 0, 0, 0}
  60         };
  61         void *ctdata;
  62         int archive_nums[] = {1}; /* which archive numbers we care about */
  63         int rc;
  64
  65         optind = 0;
  66         while ((c = getopt_long(argc, argv, "t", long_opts, NULL)) != -1) {
  67                 switch (c) {
  68                 case 't':
  69                         test++;
  70                         break;
  71                 default:
  72                         fprintf(stderr, "error: %s: option '%s' unrecognized\n",
  73                                 argv[0], argv[optind - 1]);
  74                         return EINVAL;
  75                 }
  76         }
  77
  78         rc = llapi_copytool_start(&ctdata, 0, ARRAY_SIZE(archive_nums),
  79                                   archive_nums);
  80         if (rc < 0) {
  81                 fprintf(stderr, "Can't start copytool interface: %s\n",
  82                         strerror(-rc));
  83                 return -rc;
  84         }
  85
  86         if (test)
  87                 return -llapi_copytool_fini(&ctdata);
  88
  89         printf("Waiting for message from kernel (pid=%d)\n", getpid());
  90
  91         while(1) {
  92                 struct hsm_action_list *hal;
  93                 struct hsm_action_item *hai;
  94                 int msgsize, i = 0;
  95
  96                 rc = llapi_copytool_recv(ctdata, &hal, &msgsize);
  97                 if (rc == -ESHUTDOWN) {
  98                         fprintf(stderr, "shutting down");
  99                         break;
 100                 }
 101                 if (rc < 0) {
 102                         fprintf(stderr, "Message receive: %s", strerror(-rc));
 103                         break;
 104                 }
 105                 if (msgsize == 0)
 106                         continue; /* msg not for us */
 107
 108                 printf("Copytool fs=%s archive#=%d item_count=%d\n",
 109                        hal->hal_fsname, hal->hal_archive_num, hal->hal_count);
 110
 111                 hai = hai_zero(hal);
 112                 while (++i <= hal->hal_count) {
 113                         printf("Item %d: action %d reclen %d\n", i,
 114                                hai->hai_action, hai->hai_len);
 115                         printf(" "DFID" gid="LPU64" cookie="LPU64"\n",
 116                                PFID(&hai->hai_fid), hai->hai_gid,
 117                                hai->hai_cookie);
 118                         hai = hai_next(hai);
 119                 }
 120
 121                 llapi_copytool_free(&hal);
 122         }
 123
 124         llapi_copytool_fini(&ctdata);
 125
 126         return -rc;
 127 }
 128
 129
 130