X-Git-Url: https://git.whamcloud.com/?p=fs%2Flustre-release.git;a=blobdiff_plain;f=lustre%2Futils%2Fliblustreapi.c;h=7a315c7e06fa4ee4650899d48825f623af56b452;hp=e4b782870c3ce9a5bd7e3ba5c6f9451cdd8a2383;hb=6523ee9dadfe075a42668a29652a8b4fe0e7dff4;hpb=77670a3bde8dba845fc60830d040b5f45c8dc066 diff --git a/lustre/utils/liblustreapi.c b/lustre/utils/liblustreapi.c index e4b7828..7a315c7 100644 --- a/lustre/utils/liblustreapi.c +++ b/lustre/utils/liblustreapi.c @@ -1,31 +1,49 @@ /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*- * vim:expandtab:shiftwidth=8:tabstop=8: * - * Copyright (C) 2002 Cluster File Systems, Inc. - * Author: Peter J. Braam - * Author: Phil Schwan - * Author: Robert Read + * 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 + * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf * + * 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/liblustreapi.c + * + * Author: Peter J. Braam + * Author: Phil Schwan + * Author: Robert Read */ - /* for O_DIRECTORY */ +#ifndef _GNU_SOURCE #define _GNU_SOURCE +#endif #include #include @@ -39,602 +57,3005 @@ #include #include #include -#include +#include +#include +#include +#include +#ifdef HAVE_LINUX_UNISTD_H #include +#else +#include +#endif +#include #include -#include -#include -#include -#include +#include +#include +#include +#include +#include + +static unsigned llapi_dir_filetype_table[] = { + [DT_UNKNOWN]= 0, + [DT_FIFO]= S_IFIFO, + [DT_CHR] = S_IFCHR, + [DT_DIR] = S_IFDIR, + [DT_BLK] = S_IFBLK, + [DT_REG] = S_IFREG, + [DT_LNK] = S_IFLNK, + [DT_SOCK]= S_IFSOCK, +#if defined(DT_DOOR) && defined(S_IFDOOR) + [DT_DOOR]= S_IFDOOR, +#endif +}; -#include +#if defined(DT_DOOR) && defined(S_IFDOOR) +static const int DT_MAX = DT_DOOR; +#else +static const int DT_MAX = DT_SOCK; +#endif + +static unsigned llapi_filetype_dir_table[] = { + [0]= DT_UNKNOWN, + [S_IFIFO]= DT_FIFO, + [S_IFCHR] = DT_CHR, + [S_IFDIR] = DT_DIR, + [S_IFBLK] = DT_BLK, + [S_IFREG] = DT_REG, + [S_IFLNK] = DT_LNK, + [S_IFSOCK]= DT_SOCK, +#if defined(DT_DOOR) && defined(S_IFDOOR) + [S_IFDOOR]= DT_DOOR, +#endif +}; + +#if defined(DT_DOOR) && defined(S_IFDOOR) +static const int S_IFMAX = DT_DOOR; +#else +static const int S_IFMAX = DT_SOCK; +#endif + +/* liblustreapi message level */ +static int llapi_msg_level = LLAPI_MSG_MAX; + +void llapi_msg_set_level(int level) +{ + /* ensure level is in the good range */ + if (level < LLAPI_MSG_OFF) + llapi_msg_level = LLAPI_MSG_OFF; + else if (level > LLAPI_MSG_MAX) + llapi_msg_level = LLAPI_MSG_MAX; + else + llapi_msg_level = level; +} -static void err_msg(char *fmt, ...) +void llapi_err(int level, char *fmt, ...) { va_list args; - int tmp_errno = errno; + int tmp_errno = abs(errno); + + if ((level & LLAPI_MSG_MASK) > llapi_msg_level) + return; va_start(args, fmt); vfprintf(stderr, fmt, args); va_end(args); - fprintf(stderr, ": %s (%d)\n", strerror(tmp_errno), tmp_errno); + + if (level & LLAPI_MSG_NO_ERRNO) + fprintf(stderr, "\n"); + else + fprintf(stderr, ": %s (%d)\n", strerror(tmp_errno), tmp_errno); } -int op_create_file(char *name, long stripe_size, int stripe_offset, - int stripe_count) +#define llapi_err_noerrno(level, fmt, a...) \ + llapi_err((level) | LLAPI_MSG_NO_ERRNO, fmt, ## a) + +void llapi_printf(int level, char *fmt, ...) { - struct lov_user_md lum = { 0 }; - int fd, rc = 0; + va_list args; - /* Initialize IOCTL striping pattern structure */ - lum.lmm_magic = LOV_USER_MAGIC; - lum.lmm_stripe_size = stripe_size; - lum.lmm_stripe_offset = stripe_offset; - lum.lmm_stripe_count = stripe_count; + if ((level & LLAPI_MSG_MASK) > llapi_msg_level) + return; - fd = open(name, O_CREAT | O_RDWR | O_LOV_DELAY_CREATE, 0644); - if (errno == EISDIR) - fd = open(name, O_DIRECTORY | O_RDONLY); + va_start(args, fmt); + vfprintf(stdout, fmt, args); + va_end(args); +} - if (fd < 0) { - err_msg("unable to open '%s'",name); - rc = -errno; - return rc; +/** + * size_units is unchanged if no specifier used + */ +int parse_size(char *optarg, unsigned long long *size, + unsigned long long *size_units, int bytes_spec) +{ + char *end; + + *size = strtoull(optarg, &end, 0); + + if (*end != '\0') { + if ((*end == 'b') && *(end+1) == '\0' && + (*size & (~0ULL << (64 - 9))) == 0 && + !bytes_spec) { + *size <<= 9; + *size_units = 1 << 9; + } else if ((*end == 'b') && *(end+1) == '\0' && + bytes_spec) { + *size_units = 1; + } else if ((*end == 'k' || *end == 'K') && + *(end+1) == '\0' && (*size & + (~0ULL << (64 - 10))) == 0) { + *size <<= 10; + *size_units = 1 << 10; + } else if ((*end == 'm' || *end == 'M') && + *(end+1) == '\0' && (*size & + (~0ULL << (64 - 20))) == 0) { + *size <<= 20; + *size_units = 1 << 20; + } else if ((*end == 'g' || *end == 'G') && + *(end+1) == '\0' && (*size & + (~0ULL << (64 - 30))) == 0) { + *size <<= 30; + *size_units = 1 << 30; + } else if ((*end == 't' || *end == 'T') && + *(end+1) == '\0' && (*size & + (~0ULL << (64 - 40))) == 0) { + *size <<= 40; + *size_units = 1ULL << 40; + } else if ((*end == 'p' || *end == 'P') && + *(end+1) == '\0' && (*size & + (~0ULL << (64 - 50))) == 0) { + *size <<= 50; + *size_units = 1ULL << 50; + } else if ((*end == 'e' || *end == 'E') && + *(end+1) == '\0' && (*size & + (~0ULL << (64 - 60))) == 0) { + *size <<= 60; + *size_units = 1ULL << 60; + } else { + return -1; + } } - if (ioctl(fd, LL_IOC_LOV_SETSTRIPE, &lum)) { - char *errmsg = "stripe already set"; - if (errno != EEXIST && errno != EALREADY) - errmsg = strerror(errno); + return 0; +} - fprintf(stderr, "error on ioctl for '%s' (%d): %s\n", - name, fd, errmsg); - rc = -errno; +int llapi_stripe_limit_check(unsigned long long stripe_size, int stripe_offset, + int stripe_count, int stripe_pattern) +{ + int page_size; + + /* 64 KB is the largest common page size I'm aware of (on ia64), but + * check the local page size just in case. */ + page_size = LOV_MIN_STRIPE_SIZE; + if (getpagesize() > page_size) { + page_size = getpagesize(); + llapi_err_noerrno(LLAPI_MSG_WARN, + "warning: your page size (%u) is " + "larger than expected (%u)", page_size, + LOV_MIN_STRIPE_SIZE); } - if (close(fd) < 0) { - err_msg("error on close for '%s' (%d)", name, fd); - if (rc == 0) - rc = -errno; + if (stripe_size < 0 || (stripe_size & (LOV_MIN_STRIPE_SIZE - 1))) { + llapi_err(LLAPI_MSG_ERROR, "error: bad stripe_size %lu, " + "must be an even multiple of %d bytes", + stripe_size, page_size); + return -EINVAL; } - return rc; + if (stripe_offset < -1 || stripe_offset > MAX_OBD_DEVICES) { + errno = -EINVAL; + llapi_err(LLAPI_MSG_ERROR, "error: bad stripe offset %d", + stripe_offset); + return -EINVAL; + } + if (stripe_count < -1 || stripe_count > LOV_MAX_STRIPE_COUNT) { + errno = -EINVAL; + llapi_err(LLAPI_MSG_ERROR, "error: bad stripe count %d", + stripe_count); + return -EINVAL; + } + if (stripe_size >= (1ULL << 32)){ + errno = -EINVAL; + llapi_err(LLAPI_MSG_ERROR, "warning: stripe size larger than 4G" + " is not currently supported and would wrap"); + return -EINVAL; + } + return 0; } -struct find_param { - int recursive; - int verbose; - int quiet; - struct obd_uuid *obduuid; - struct obd_ioctl_data data; - struct lov_desc desc; - int uuidslen; - char *buf; - int buflen; - struct obd_uuid *uuids; - struct lov_user_md *lum; - int got_uuids; - int obdindex; - int max_ost_count; -}; - -/* XXX Max obds per lov currently hardcoded to 1000 in lov/lov_obd.c */ -#define MAX_LOV_UUID_COUNT 1000 -#define OBD_NOT_FOUND (-1) +static int poolpath(char *fsname, char *pathname, char *pool_pathname); -static int prepare_find(struct find_param *param) +int llapi_file_open_pool(const char *name, int flags, int mode, + unsigned long long stripe_size, int stripe_offset, + int stripe_count, int stripe_pattern, char *pool_name) { - int datalen, desclen; - int cfglen, lumlen; - int max_ost_count = MAX_LOV_UUID_COUNT; - - datalen = size_round(sizeof(param->data)); - desclen = size_round(sizeof(param->desc)); - param->uuidslen = size_round(max_ost_count * sizeof(*param->uuids)); - cfglen = datalen + desclen + param->uuidslen; - lumlen = lov_mds_md_size(max_ost_count); - if (cfglen > lumlen) - param->buflen = cfglen; - else - param->buflen = lumlen; + struct lov_user_md_v3 lum = { 0 }; + int fd, rc = 0; + int isdir = 0; + char fsname[MAX_OBD_NAME + 1], *ptr; - /* XXX max ioctl buffer size currently hardcoded to 8192 */ - if (param->buflen > 8192) { - int nuuids, remaining; + fd = open(name, flags | O_LOV_DELAY_CREATE, mode); + if (fd < 0 && errno == EISDIR) { + fd = open(name, O_DIRECTORY | O_RDONLY); + isdir++; + } - param->buflen = 8192; - nuuids = (param->buflen - datalen - desclen) / - sizeof(*param->uuids); - param->uuidslen = size_round(nuuids * sizeof(*param->uuids)); - remaining = nuuids * sizeof(*param->uuids); - if (param->uuidslen > remaining) - nuuids--; - max_ost_count = nuuids; - while ((lumlen=lov_mds_md_size(max_ost_count)) > param->buflen) - --max_ost_count; + if (fd < 0) { + rc = -errno; + llapi_err(LLAPI_MSG_ERROR, "unable to open '%s'", name); + return rc; + } - cfglen = datalen + desclen + param->uuidslen; + if ((rc = llapi_stripe_limit_check(stripe_size, stripe_offset, + stripe_count, stripe_pattern)) != 0){ + errno = rc; + goto out; } - if ((param->buf = malloc(param->buflen)) == NULL) { - err_msg("unable to allocate %d bytes of memory for ioctl's", - param->buflen); - return ENOMEM; + /* Initialize IOCTL striping pattern structure */ + lum.lmm_magic = LOV_USER_MAGIC_V3; + lum.lmm_pattern = stripe_pattern; + lum.lmm_stripe_size = stripe_size; + lum.lmm_stripe_count = stripe_count; + lum.lmm_stripe_offset = stripe_offset; + + /* in case user give the full pool name ., skip + * the fsname */ + if (pool_name != NULL) { + ptr = strchr(pool_name, '.'); + if (ptr != NULL) { + strncpy(fsname, pool_name, ptr - pool_name); + *ptr = '\0'; + /* if fsname matches a filesystem skip it + * if not keep the poolname as is */ + if (poolpath(fsname, NULL, NULL) == 0) + pool_name = ptr + 1; + } + strncpy(lum.lmm_pool_name, pool_name, LOV_MAXPOOLNAME); + } else { + /* If no pool is specified at all, use V1 request */ + lum.lmm_magic = LOV_USER_MAGIC_V1; } - param->lum = (struct lov_user_md *)param->buf; - param->uuids = (struct obd_uuid *)param->buf; - param->got_uuids = 0; - param->obdindex = OBD_NOT_FOUND; - param->max_ost_count = max_ost_count; + if (ioctl(fd, LL_IOC_LOV_SETSTRIPE, &lum)) { + char *errmsg = "stripe already set"; + rc = -errno; + if (errno != EEXIST && errno != EALREADY) + errmsg = strerror(errno); - return 0; + llapi_err_noerrno(LLAPI_MSG_ERROR, + "error on ioctl "LPX64" for '%s' (%d): %s", + (__u64)LL_IOC_LOV_SETSTRIPE, name, fd,errmsg); + } +out: + if (rc) { + close(fd); + fd = rc; + } + + return fd; } -static void cleanup_find(struct find_param *param) +int llapi_file_open(const char *name, int flags, int mode, + unsigned long long stripe_size, int stripe_offset, + int stripe_count, int stripe_pattern) { - if (param->obduuid) - free(param->obduuid); - if (param->buf) - free(param->buf); + return llapi_file_open_pool(name, flags, mode, stripe_size, + stripe_offset, stripe_count, + stripe_pattern, NULL); } -static int get_obd_uuids(DIR *dir, char *dname, struct find_param *param) +int llapi_file_create(const char *name, unsigned long long stripe_size, + int stripe_offset, int stripe_count, int stripe_pattern) { - int obdcount; - struct obd_uuid *uuidp; - int rc, i; - - param->got_uuids = 1; - memset(¶m->data, 0, sizeof(param->data)); - param->data.ioc_inllen1 = sizeof(struct lov_desc); - param->data.ioc_inlbuf1 = (char *)¶m->desc; - param->data.ioc_inllen2 = param->uuidslen; - param->data.ioc_inlbuf2 = (char *)param->uuids; - - memset(¶m->desc, 0, sizeof(struct lov_desc)); - param->desc.ld_tgt_count = param->max_ost_count; - - if (obd_ioctl_pack(¶m->data, ¶m->buf, param->buflen)) { - fprintf(stderr, "internal buffer error from %s\n", dname); - return (param->obduuid ? EINVAL : 0); - } + int fd; - rc = ioctl(dirfd(dir), OBD_IOC_LOV_GET_CONFIG, param->buf); - if (rc) { - err_msg("error getting LOV config from %s", dname); - return (param->obduuid ? errno : 0); - } + fd = llapi_file_open_pool(name, O_CREAT | O_WRONLY, 0644, stripe_size, + stripe_offset, stripe_count, stripe_pattern, + NULL); + if (fd < 0) + return fd; - if (obd_ioctl_unpack(¶m->data, param->buf, param->buflen)) { - err_msg("invalid reply from ioctl from %s", dname); - return (param->obduuid ? EINVAL : 0); - } + close(fd); + return 0; +} - obdcount = param->desc.ld_tgt_count; - if (obdcount == 0) - return 0; +int llapi_file_create_pool(const char *name, unsigned long long stripe_size, + int stripe_offset, int stripe_count, + int stripe_pattern, char *pool_name) +{ + int fd; - if (param->obduuid) { - for (i = 0, uuidp = param->uuids; i < obdcount; i++, uuidp++) { - if (strncmp(param->obduuid->uuid, uuidp->uuid, - sizeof(*uuidp)) == 0) { - param->obdindex = i; - break; - } - } - if (param->obdindex == OBD_NOT_FOUND) { - printf("unknown obduuid: %s\n", param->obduuid->uuid); - return EINVAL; - } - } else if (!param->quiet) { - printf("OBDS:\n"); - for (i = 0, uuidp = param->uuids; i < obdcount; i++, uuidp++) - printf("%4d: %s\n", i, uuidp->uuid); - } + fd = llapi_file_open_pool(name, O_CREAT | O_WRONLY, 0644, stripe_size, + stripe_offset, stripe_count, stripe_pattern, + pool_name); + if (fd < 0) + return fd; + close(fd); return 0; } -void lov_dump_user_lmm_v1(struct lov_user_md_v1 *lum, char *dname, char *fname, - int obdindex, int quiet, int header, int body) +/* + * Find the fsname, the full path, and/or an open fd. + * Either the fsname or path must not be NULL + */ +#define WANT_PATH 0x1 +#define WANT_FSNAME 0x2 +#define WANT_FD 0x4 +#define WANT_INDEX 0x8 +#define WANT_ERROR 0x10 +static int get_root_path(int want, char *fsname, int *outfd, char *path, + int index) { - int i, obdstripe = 0; - - if (obdindex != OBD_NOT_FOUND) { - for (i = 0; i < lum->lmm_stripe_count; i++) { - if (obdindex == lum->lmm_objects[i].l_ost_idx) { - printf("%s/%s\n", dname, fname); - obdstripe = 1; - break; - } - } - } else if (!quiet) { - printf("%s/%s\n", dname, fname); - obdstripe = 1; + struct mntent mnt; + char buf[PATH_MAX], mntdir[PATH_MAX]; + char *name = NULL, *ptr; + FILE *fp; + int idx = 0, len = 0, mntlen, fd; + int rc = -ENODEV; + + /* get the mount point */ + fp = setmntent(MOUNTED, "r"); + if (fp == NULL) { + llapi_err(LLAPI_MSG_ERROR, + "setmntent(%s) failed: %s:", MOUNTED, + strerror (errno)); + return -EIO; } + while (1) { + if (getmntent_r(fp, &mnt, buf, sizeof(buf)) == NULL) + break; - /* if it's a directory */ - if (*fname == '\0') { - if (header && (obdstripe == 1)) { - printf("count: %d, size: %d, offset: %d\n\n", - lum->lmm_stripe_count, lum->lmm_stripe_size, - (short int)lum->lmm_stripe_offset); - } - return; - } + if (!llapi_is_lustre_mnt(&mnt)) + continue; - if (header && (obdstripe == 1)) { - printf("lmm_magic: 0x%08X\n", lum->lmm_magic); - printf("lmm_object_gr: "LPX64"\n", lum->lmm_object_gr); - printf("lmm_object_id: "LPX64"\n", lum->lmm_object_id); - printf("lmm_stripe_count: %u\n", (int)lum->lmm_stripe_count); - printf("lmm_stripe_size: %u\n", lum->lmm_stripe_size); - printf("lmm_stripe_pattern: %x\n", lum->lmm_pattern); - } + mntlen = strlen(mnt.mnt_dir); + ptr = strrchr(mnt.mnt_fsname, '/'); + if (!ptr && !len) { + rc = -EINVAL; + break; + } + ptr++; - if (body) { - long long oid; + if ((want & WANT_INDEX) && (idx++ != index)) + continue; - if ((!quiet) && (obdstripe == 1)) - printf("\tobdidx\t\t objid\t\tobjid\t\t group\n"); + /* Check the fsname for a match */ + if (!(want & WANT_FSNAME) && fsname != NULL && + (strlen(fsname) > 0) && (strcmp(ptr, fsname) != 0)) + continue; - for (i = 0; i < lum->lmm_stripe_count; i++) { - int idx = lum->lmm_objects[i].l_ost_idx; - oid = lum->lmm_objects[i].l_object_id; - if ((obdindex == OBD_NOT_FOUND) || (obdindex == idx)) - printf("\t%6u\t%14llu\t%#13llx\t%14lld%s\n", - idx, oid, oid, - (long long)lum->lmm_objects[i].l_object_gr, - obdindex == idx ? " *" : ""); + /* If the path isn't set return the first one we find */ + if (path == NULL || strlen(path) == 0) { + strcpy(mntdir, mnt.mnt_dir); + name = ptr; + rc = 0; + break; + /* Otherwise find the longest matching path */ + } else if ((strlen(path) >= mntlen) && (mntlen >= len) && + (strncmp(mnt.mnt_dir, path, mntlen) == 0)) { + strcpy(mntdir, mnt.mnt_dir); + mntlen = len; + name = ptr; + rc = 0; } - printf("\n"); } + endmntent(fp); + + /* Found it */ + if (rc == 0) { + if ((want & WANT_FSNAME) && fsname != NULL) + strcpy(fsname, name); + if ((want & WANT_PATH) && path != NULL) + strcpy(path, mntdir); + if (want & WANT_FD) { + fd = open(mntdir, O_RDONLY | O_DIRECTORY | O_NONBLOCK); + if (fd < 0) { + perror("open"); + rc = -errno; + } else { + *outfd = fd; + } + } + } else if (want & WANT_ERROR) + llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO, + "can't find fs root for '%s': %d", + (want & WANT_PATH) ? fsname : path, rc); + return rc; } -void lov_dump_user_lmm(struct find_param *param, char *dname, char *fname) +/* + * search lustre mounts + * + * Calling this function will return to the user the mount point, mntdir, and + * the file system name, fsname, if the user passed a buffer to this routine. + * + * The user inputs are pathname and index. If the pathname is supplied then + * the value of the index will be ignored. The pathname will return data if + * the pathname is located on a lustre mount. Index is used to pick which + * mount point you want in the case of multiple mounted lustre file systems. + * See function lfs_osts in lfs.c for a example of the index use. + */ +int llapi_search_mounts(const char *pathname, int index, char *mntdir, + char *fsname) { - switch(*(__u32 *)param->lum) { /* lum->lmm_magic */ - case LOV_USER_MAGIC_V1: - lov_dump_user_lmm_v1(param->lum, dname, fname, param->obdindex, - param->quiet, param->verbose, - (param->verbose || !param->obduuid)); - break; - default: - printf("unknown lmm_magic: 0x%08X\n", *(__u32 *)param->lum); - return; - } + int want = WANT_PATH, idx = -1; + + if (!pathname) { + want |= WANT_INDEX; + idx = index; + } else + strcpy(mntdir, pathname); + + if (fsname) + want |= WANT_FSNAME; + return get_root_path(want, fsname, NULL, mntdir, idx); } -int get_file_stripe(char *path, struct lov_user_md *lum) +int llapi_search_fsname(const char *pathname, char *fsname) { - char *dname, *fname; - int fd, rc = 0; - - fname = strrchr(path, '/'); + return get_root_path(WANT_FSNAME | WANT_ERROR, fsname, NULL, + (char *)pathname, -1); +} - /* It should be a file (or other non-directory) */ - if (fname == NULL) { - dname = (char *)malloc(2); - if (dname == NULL) - return ENOMEM; - strcpy(dname, "."); - fname = path; - } else { - dname = (char *)malloc(fname - path + 1); - if (dname == NULL) - return ENOMEM; - strncpy(dname, path, fname - path); - dname[fname - path + 1] = '\0'; - fname++; - } +/* return the first file matching this pattern */ +static int first_match(char *pattern, char *buffer) +{ + glob_t glob_info; - if ((fd = open(dname, O_RDONLY)) == -1) { - free(dname); - return errno; - } + if (glob(pattern, GLOB_BRACE, NULL, &glob_info)) + return -ENOENT; - strncpy((char *)lum, fname, sizeof(*lum)); - if (ioctl(fd, IOC_MDC_GETSTRIPE, (void *)lum) == -1) { - close(fd); - free(dname); - return errno; + if (glob_info.gl_pathc < 1) { + globfree(&glob_info); + return -ENOENT; } - if (close(fd) == -1) - rc = errno; - - free(dname); + strcpy(buffer, glob_info.gl_pathv[0]); - return rc; + globfree(&glob_info); + return 0; } -static int process_file(DIR *dir, char *dname, char *fname, - struct find_param *param) +/* + * find the pool directory path under /proc + * (can be also used to test if a fsname is known) + */ +static int poolpath(char *fsname, char *pathname, char *pool_pathname) { - int rc; + int rc = 0; + char pattern[PATH_MAX + 1]; + char buffer[PATH_MAX]; - strncpy((char *)param->lum, fname, param->buflen); + if (fsname == NULL) { + rc = llapi_search_fsname(pathname, buffer); + if (rc != 0) + return rc; + fsname = buffer; + strcpy(pathname, fsname); + } - rc = ioctl(dirfd(dir), IOC_MDC_GETSTRIPE, (void *)param->lum); - if (rc) { - if (errno == ENODATA) { - if (!param->obduuid && !param->quiet) - fprintf(stderr, - "%s/%s has no stripe info\n", - dname, fname); - rc = 0; - } else if (errno == EISDIR) { - fprintf(stderr, "process_file on directory %s/%s!\n", - dname, fname); - /* add fname to directory list; */ - rc = errno; - } else { - err_msg("IOC_MDC_GETSTRIPE ioctl failed"); - rc = errno; - } + snprintf(pattern, PATH_MAX, "/proc/fs/lustre/lov/%s-*/pools", fsname); + rc = first_match(pattern, buffer); + if (rc) return rc; - } - lov_dump_user_lmm(param, dname, fname); + /* in fsname test mode, pool_pathname is NULL */ + if (pool_pathname != NULL) + strcpy(pool_pathname, buffer); return 0; } - -static int process_dir(DIR *dir, char *dname, struct find_param *param) +/** + * Get the list of pool members. + * \param poolname string of format . + * \param members caller-allocated array of char* + * \param list_size size of the members array + * \param buffer caller-allocated buffer for storing OST names + * \param buffer_size size of the buffer + * + * \return number of members retrieved for this pool + * \retval -error failure + */ +int llapi_get_poolmembers(const char *poolname, char **members, + int list_size, char *buffer, int buffer_size) { - struct dirent64 *dirp; - DIR *subdir; - char path[1024]; - int rc; - - if (!param->got_uuids) { - rc = get_obd_uuids(dir, dname, param); - if (rc) - return rc; + char fsname[PATH_MAX + 1]; + char *pool, *tmp; + char pathname[PATH_MAX + 1]; + char path[PATH_MAX + 1]; + char buf[1024]; + FILE *fd; + int rc = 0; + int nb_entries = 0; + int used = 0; + + /* name is FSNAME.POOLNAME */ + if (strlen(poolname) > PATH_MAX) + return -EOVERFLOW; + strcpy(fsname, poolname); + pool = strchr(fsname, '.'); + if (pool == NULL) + return -EINVAL; + + *pool = '\0'; + pool++; + + rc = poolpath(fsname, NULL, pathname); + if (rc != 0) { + errno = -rc; + llapi_err(LLAPI_MSG_ERROR, "Lustre filesystem '%s' not found", + fsname); + return rc; } - /* retrieve dir's stripe info */ - strncpy((char *)param->lum, dname, param->buflen); - rc = ioctl(dirfd(dir), LL_IOC_LOV_GETSTRIPE, (void *)param->lum); - if (rc) { - if (errno == ENODATA) { - if (!param->obduuid && !param->quiet) - printf("%s/%s has no stripe info\n", - dname, ""); - rc = 0; - } else { - err_msg("IOC_MDC_GETSTRIPE ioctl failed"); - return errno; - } - } else { - lov_dump_user_lmm(param, dname, ""); + llapi_printf(LLAPI_MSG_NORMAL, "Pool: %s.%s\n", fsname, pool); + sprintf(path, "%s/%s", pathname, pool); + if ((fd = fopen(path, "r")) == NULL) { + llapi_err(LLAPI_MSG_ERROR, "Cannot open %s", path); + return -EINVAL; } - /* Handle the contents of the directory */ - while ((dirp = readdir64(dir)) != NULL) { - if (!strcmp(dirp->d_name, ".") || !strcmp(dirp->d_name, "..")) - continue; - - switch (dirp->d_type) { - case DT_UNKNOWN: - err_msg("\"%s\" is UNKNOWN type %d", dirp->d_name, - dirp->d_type); - /* If we cared we could stat the file to determine - * type and continue on here, but we don't since we - * know d_type should be valid for lustre and this - * tool only makes sense for lustre filesystems. */ - return EINVAL; - break; - case DT_DIR: - if (!param->recursive) - break; - strcpy(path, dname); - strcat(path, "/"); - strcat(path, dirp->d_name); - subdir = opendir(path); - if (subdir == NULL) { - err_msg("\"%.40s\" opendir failed", path); - return errno; - } - rc = process_dir(subdir, path, param); - closedir(subdir); - if (rc) - return rc; + rc = 0; + while (fgets(buf, sizeof(buf), fd) != NULL) { + if (nb_entries >= list_size) { + rc = -EOVERFLOW; break; - case DT_REG: - rc = process_file(dir, dname, dirp->d_name, param); - if (rc) - return rc; - break; - default: + } + /* remove '\n' */ + if ((tmp = strchr(buf, '\n')) != NULL) + *tmp='\0'; + if (used + strlen(buf) + 1 > buffer_size) { + rc = -EOVERFLOW; break; } + + strcpy(buffer + used, buf); + members[nb_entries] = buffer + used; + used += strlen(buf) + 1; + nb_entries++; + rc = nb_entries; } - return 0; + fclose(fd); + return rc; } -static int process_path(char *path, struct find_param *param) +/** + * Get the list of pools in a filesystem. + * \param name filesystem name or path + * \param poollist caller-allocated array of char* + * \param list_size size of the poollist array + * \param buffer caller-allocated buffer for storing pool names + * \param buffer_size size of the buffer + * + * \return number of pools retrieved for this filesystem + * \retval -error failure + */ +int llapi_get_poollist(const char *name, char **poollist, int list_size, + char *buffer, int buffer_size) { - char *fname, *dname; + char fsname[PATH_MAX + 1], rname[PATH_MAX + 1], pathname[PATH_MAX + 1]; + char *ptr; DIR *dir; + struct dirent pool; + struct dirent *cookie = NULL; int rc = 0; + unsigned int nb_entries = 0; + unsigned int used = 0; + unsigned int i; + + /* initilize output array */ + for (i = 0; i < list_size; i++) + poollist[i] = NULL; + + /* is name a pathname ? */ + ptr = strchr(name, '/'); + if (ptr != NULL) { + /* only absolute pathname is supported */ + if (*name != '/') + return -EINVAL; + if (!realpath(name, rname)) { + rc = -errno; + llapi_err(LLAPI_MSG_ERROR, "invalid path '%s'", name); + return rc; + } - fname = strrchr(path, '/'); - if (fname != NULL && fname[1] == '\0') { - /* Trailing '/', it must be a dir */ - *fname = '\0'; - dir = opendir(path); - if (dir == NULL) { - err_msg("\"%.40s\" opendir failed", path); - rc = errno; - } else { - rc = process_dir(dir, path, param); - closedir(dir); + rc = poolpath(NULL, rname, pathname); + if (rc != 0) { + errno = -rc; + llapi_err(LLAPI_MSG_ERROR, "'%s' is not" + " a Lustre filesystem", name); + return rc; } - } else if ((dir = opendir(path)) != NULL) { - /* No trailing '/', but it is still a dir */ - rc = process_dir(dir, path, param); - closedir(dir); + strcpy(fsname, rname); } else { - /* It must be a file (or other non-directory) */ - if (fname == NULL) { - dname = "."; - fname = path; - } else { - *fname = '\0'; - fname++; - dname = path; - } - dir = opendir(dname); - if (dir == NULL) { - err_msg("\"%.40s\" opendir failed", dname); - rc = errno; - } else { - if (!param->got_uuids) - rc = get_obd_uuids(dir, dname, param); - if (rc == 0) - rc = process_file(dir, dname, fname, param); - closedir(dir); - } + /* name is FSNAME */ + strcpy(fsname, name); + rc = poolpath(fsname, NULL, pathname); + } + if (rc != 0) { + errno = -rc; + llapi_err(LLAPI_MSG_ERROR, "Lustre filesystem '%s' not found", + name); + return rc; } - return rc; -} + llapi_printf(LLAPI_MSG_NORMAL, "Pools from %s:\n", fsname); + if ((dir = opendir(pathname)) == NULL) { + llapi_err(LLAPI_MSG_ERROR, "Could not open pool list for '%s'", + name); + return -errno; + } + while(1) { + rc = readdir_r(dir, &pool, &cookie); -int op_find(char *path, struct obd_uuid *obduuid, int recursive, - int verbose, int quiet) -{ - struct find_param param; - int ret = 0; + if (rc != 0) { + llapi_err(LLAPI_MSG_ERROR, + "Error reading pool list for '%s'", name); + return -errno; + } else if ((rc == 0) && (cookie == NULL)) + /* end of directory */ + break; - memset(¶m, 0, sizeof(param)); - param.recursive = recursive; - param.verbose = verbose; - param.quiet = quiet; - if (obduuid) { - param.obduuid = malloc(sizeof(*obduuid)); - if (param.obduuid == NULL) { - ret = ENOMEM; - goto out; - } - memcpy(param.obduuid, obduuid, sizeof(*obduuid)); - } + /* ignore . and .. */ + if (!strcmp(pool.d_name, ".") || !strcmp(pool.d_name, "..")) + continue; - ret = prepare_find(¶m); - if (ret) - goto out; + /* check output bounds */ + if (nb_entries >= list_size) + return -EOVERFLOW; - process_path(path, ¶m); -out: - cleanup_find(¶m); - return ret; -} + /* +2 for '.' and final '\0' */ + if (used + strlen(pool.d_name) + strlen(fsname) + 2 + > buffer_size) + return -EOVERFLOW; -#define MAX_STRING_SIZE 128 -#define DEVICES_LIST "/proc/fs/lustre/devices" + sprintf(buffer + used, "%s.%s", fsname, pool.d_name); + poollist[nb_entries] = buffer + used; + used += strlen(pool.d_name) + strlen(fsname) + 2; + nb_entries++; + } + + closedir(dir); + return nb_entries; +} -int op_check(int type_num, char **obd_type, char *dir) +/* wrapper for lfs.c and obd.c */ +int llapi_poollist(const char *name) { - int rc = 0; - int i; + /* list of pool names (assume that pool count is smaller + than OST count) */ + char *list[FIND_MAX_OSTS]; + char *buffer; + /* fsname-OST0000_UUID < 32 char, 1 per OST */ + int bufsize = FIND_MAX_OSTS * 32; + int i, nb; + + buffer = malloc(bufsize); + if (buffer == NULL) + return -ENOMEM; + + if ((name[0] == '/') || (strchr(name, '.') == NULL)) + /* name is a path or fsname */ + nb = llapi_get_poollist(name, list, FIND_MAX_OSTS, buffer, + bufsize); + else + /* name is a pool name (.) */ + nb = llapi_get_poolmembers(name, list, FIND_MAX_OSTS, buffer, + bufsize); - char buf[MAX_STRING_SIZE]; + for (i = 0; i < nb; i++) + llapi_printf(LLAPI_MSG_NORMAL, "%s\n", list[i]); + + free(buffer); + return (nb < 0 ? nb : 0); +} + + +typedef int (semantic_func_t)(char *path, DIR *parent, DIR *d, + void *data, cfs_dirent_t *de); + +#define MAX_LOV_UUID_COUNT max(LOV_MAX_STRIPE_COUNT, 1000) +#define OBD_NOT_FOUND (-1) + +static int common_param_init(struct find_param *param) +{ + param->lumlen = lov_mds_md_size(MAX_LOV_UUID_COUNT, LOV_MAGIC_V3); + if ((param->lmd = malloc(sizeof(lstat_t) + param->lumlen)) == NULL) { + llapi_err(LLAPI_MSG_ERROR, + "error: allocation of %d bytes for ioctl", + sizeof(lstat_t) + param->lumlen); + return -ENOMEM; + } + + param->got_uuids = 0; + param->obdindexes = NULL; + param->obdindex = OBD_NOT_FOUND; + return 0; +} + +static void find_param_fini(struct find_param *param) +{ + if (param->obdindexes) + free(param->obdindexes); + + if (param->lmd) + free(param->lmd); +} + +int llapi_file_fget_lov_uuid(int fd, struct obd_uuid *lov_name) +{ + int rc = ioctl(fd, OBD_IOC_GETNAME, lov_name); + if (rc) { + rc = errno; + llapi_err(LLAPI_MSG_ERROR, "error: can't get lov name."); + } + return rc; +} + +int llapi_file_get_lov_uuid(const char *path, struct obd_uuid *lov_uuid) +{ + int fd, rc; + + fd = open(path, O_RDONLY); + if (fd < 0) { + rc = errno; + llapi_err(LLAPI_MSG_ERROR, "error opening %s", path); + return rc; + } + + rc = llapi_file_fget_lov_uuid(fd, lov_uuid); + + close(fd); + + return rc; +} + +/* + * If uuidp is NULL, return the number of available obd uuids. + * If uuidp is non-NULL, then it will return the uuids of the obds. If + * there are more OSTs then allocated to uuidp, then an error is returned with + * the ost_count set to number of available obd uuids. + */ +int llapi_lov_get_uuids(int fd, struct obd_uuid *uuidp, int *ost_count) +{ + struct obd_uuid lov_name; + char buf[1024]; + FILE *fp; + int rc = 0, index = 0; + + /* Get the lov name */ + rc = llapi_file_fget_lov_uuid(fd, &lov_name); + if (rc) + return rc; + + /* Now get the ost uuids from /proc */ + snprintf(buf, sizeof(buf), "/proc/fs/lustre/lov/%s/target_obd", + lov_name.uuid); + fp = fopen(buf, "r"); + if (fp == NULL) { + rc = errno; + llapi_err(LLAPI_MSG_ERROR, "error: opening '%s'", buf); + return rc; + } + + while (fgets(buf, sizeof(buf), fp) != NULL) { + if (uuidp && (index < *ost_count)) { + if (sscanf(buf, "%d: %s", &index, uuidp[index].uuid) <2) + break; + } + index++; + } + + fclose(fp); + + if (uuidp && (index >= *ost_count)) + return -EOVERFLOW; + + *ost_count = index; + return rc; +} + +int llapi_get_obd_count(char *mnt, int *count, int is_mdt) +{ + DIR *root; + int rc; + + root = opendir(mnt); + if (!root) { + llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt); + return -1; + } + + *count = is_mdt; + rc = ioctl(dirfd(root), LL_IOC_GETOBDCOUNT, count); + + closedir(root); + return rc; +} + +/* Here, param->obduuid points to a single obduuid, the index of which is + * returned in param->obdindex */ +static int setup_obd_uuid(DIR *dir, char *dname, struct find_param *param) +{ + struct obd_uuid lov_uuid; + char uuid[sizeof(struct obd_uuid)]; + char buf[1024]; + FILE *fp; + int rc = 0, index; + + /* Get the lov name */ + rc = llapi_file_fget_lov_uuid(dirfd(dir), &lov_uuid); + if (rc) { + if (errno != ENOTTY) { + rc = errno; + llapi_err(LLAPI_MSG_ERROR, + "error: can't get lov name: %s", dname); + } else { + rc = 0; + } + return rc; + } + + param->got_uuids = 1; + + /* Now get the ost uuids from /proc */ + snprintf(buf, sizeof(buf), "/proc/fs/lustre/lov/%s/target_obd", + lov_uuid.uuid); + fp = fopen(buf, "r"); + if (fp == NULL) { + rc = errno; + llapi_err(LLAPI_MSG_ERROR, "error: opening '%s'", buf); + return rc; + } + + if (!param->obduuid && !param->quiet && !param->obds_printed) + llapi_printf(LLAPI_MSG_NORMAL, "OBDS:\n"); + + while (fgets(buf, sizeof(buf), fp) != NULL) { + if (sscanf(buf, "%d: %s", &index, uuid) < 2) + break; + + if (param->obduuid) { + if (strncmp(param->obduuid->uuid, uuid, + sizeof(uuid)) == 0) { + param->obdindex = index; + break; + } + } else if (!param->quiet && !param->obds_printed) { + /* Print everything */ + llapi_printf(LLAPI_MSG_NORMAL, "%s", buf); + } + } + param->obds_printed = 1; + + fclose(fp); + + if (!param->quiet && param->obduuid && + (param->obdindex == OBD_NOT_FOUND)) { + llapi_err_noerrno(LLAPI_MSG_ERROR, + "error: %s: unknown obduuid: %s", + __FUNCTION__, param->obduuid->uuid); + //rc = EINVAL; + } + + return (rc); +} + +/* In this case, param->obduuid will be an array of obduuids and + * obd index for all these obduuids will be returned in + * param->obdindexes */ +static int setup_obd_indexes(DIR *dir, struct find_param *param) +{ + struct obd_uuid *uuids = NULL; + int obdcount = INIT_ALLOC_NUM_OSTS; + int ret, obd_valid = 0, obdnum, i; + + uuids = (struct obd_uuid *)malloc(INIT_ALLOC_NUM_OSTS * + sizeof(struct obd_uuid)); + if (uuids == NULL) + return -ENOMEM; + +retry_get_uuids: + ret = llapi_lov_get_uuids(dirfd(dir), uuids, + &obdcount); + if (ret) { + struct obd_uuid *uuids_temp; + + if (ret == -EOVERFLOW) { + uuids_temp = realloc(uuids, obdcount * + sizeof(struct obd_uuid)); + if (uuids_temp != NULL) + goto retry_get_uuids; + else + ret = -ENOMEM; + } + + llapi_err(LLAPI_MSG_ERROR, "get ost uuid failed"); + return ret; + } + + param->obdindexes = malloc(param->num_obds * sizeof(param->obdindex)); + if (param->obdindexes == NULL) + return -ENOMEM; + + for (obdnum = 0; obdnum < param->num_obds; obdnum++) { + for (i = 0; i <= obdcount; i++) { + if (strcmp((char *)¶m->obduuid[obdnum].uuid, + (char *)&uuids[i]) == 0) { + param->obdindexes[obdnum] = i; + obd_valid++; + break; + } + } + if (i == obdcount) + param->obdindexes[obdnum] = OBD_NOT_FOUND; + } + + if (obd_valid == 0) + param->obdindex = OBD_NOT_FOUND; + else + param->obdindex = obd_valid; + + param->got_uuids = 1; + + return 0; +} + +static void lov_dump_user_lmm_header(struct lov_user_md *lum, char *path, + int is_dir, int verbose, int quiet, + char *pool_name) +{ + char *prefix = is_dir ? "" : "lmm_"; + char nl = is_dir ? ' ' : '\n'; + + if (verbose && path) + llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path); + + if ((verbose & VERBOSE_DETAIL) && !is_dir) { + llapi_printf(LLAPI_MSG_NORMAL, "lmm_magic: 0x%08X\n", + lum->lmm_magic); + llapi_printf(LLAPI_MSG_NORMAL, "lmm_object_gr: "LPX64"\n", + lum->lmm_object_gr); + llapi_printf(LLAPI_MSG_NORMAL, "lmm_object_id: "LPX64"\n", + lum->lmm_object_id); + } + + if (verbose & VERBOSE_COUNT) { + if (!quiet) + llapi_printf(LLAPI_MSG_NORMAL, "%sstripe_count: ", + prefix); + llapi_printf(LLAPI_MSG_NORMAL, "%u%c", + (int)lum->lmm_stripe_count, nl); + } + + if (verbose & VERBOSE_SIZE) { + if (!quiet) + llapi_printf(LLAPI_MSG_NORMAL, "%sstripe_size: ", + prefix); + llapi_printf(LLAPI_MSG_NORMAL, "%u%c", lum->lmm_stripe_size, + nl); + } + + if ((verbose & VERBOSE_DETAIL) && !is_dir) { + llapi_printf(LLAPI_MSG_NORMAL, "lmm_stripe_pattern: %x%c", + lum->lmm_pattern, nl); + } + + if (verbose & VERBOSE_OFFSET) { + if (!quiet) + llapi_printf(LLAPI_MSG_NORMAL, "%sstripe_offset: ", + prefix); + llapi_printf(LLAPI_MSG_NORMAL, "%u%c", + lum->lmm_objects[0].l_ost_idx, nl); + } + + if ((verbose & VERBOSE_POOL) && (pool_name != NULL)) + llapi_printf(LLAPI_MSG_NORMAL, "pool: %s%c", pool_name, nl); + + if (is_dir) + llapi_printf(LLAPI_MSG_NORMAL, "\n"); +} + +void lov_dump_user_lmm_v1v3(struct lov_user_md *lum, char *pool_name, + struct lov_user_ost_data_v1 *objects, + char *path, int is_dir, + int obdindex, int quiet, int header, int body) +{ + int i, obdstripe = 0; + + if (obdindex != OBD_NOT_FOUND) { + for (i = 0; !is_dir && i < lum->lmm_stripe_count; i++) { + if (obdindex == objects[i].l_ost_idx) { + llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path); + obdstripe = 1; + break; + } + } + } else { + if (!quiet) + llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path); + obdstripe = 1; + } + + /* if it's a directory */ + if (is_dir) { + if (obdstripe == 1) { + if (lum->lmm_object_gr == LOV_OBJECT_GROUP_DEFAULT) { + llapi_printf(LLAPI_MSG_NORMAL, "(Default) "); + lum->lmm_object_gr = LOV_OBJECT_GROUP_CLEAR; + } + /* maintain original behavior */ + if (!header) + header |= VERBOSE_ALL; + lov_dump_user_lmm_header(lum, path, is_dir, header, + quiet, pool_name); + } + return; + } + + if (header && (obdstripe == 1)) + lov_dump_user_lmm_header(lum, NULL, is_dir, header, quiet, + pool_name); + + if (body) { + if ((!quiet) && (obdstripe == 1)) + llapi_printf(LLAPI_MSG_NORMAL, + "\tobdidx\t\t objid\t\tobjid\t\t group\n"); + + for (i = 0; i < lum->lmm_stripe_count; i++) { + int idx = objects[i].l_ost_idx; + long long oid = objects[i].l_object_id; + long long gr = objects[i].l_object_gr; + if ((obdindex == OBD_NOT_FOUND) || (obdindex == idx)) + llapi_printf(LLAPI_MSG_NORMAL, + "\t%6u\t%14llu\t%#13llx\t%14llu%s\n", + idx, oid, oid, gr, + obdindex == idx ? " *" : ""); + } + llapi_printf(LLAPI_MSG_NORMAL, "\n"); + } +} + +void lov_dump_user_lmm_join(struct lov_user_md_v1 *lum, char *path, + int is_dir, int obdindex, int quiet, + int header, int body) +{ + struct lov_user_md_join *lumj = (struct lov_user_md_join *)lum; + int i, obdstripe = 0; + + if (obdindex != OBD_NOT_FOUND) { + for (i = 0; i < lumj->lmm_stripe_count; i++) { + if (obdindex == lumj->lmm_objects[i].l_ost_idx) { + llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path); + obdstripe = 1; + break; + } + } + } else { + if (!quiet) + llapi_printf(LLAPI_MSG_NORMAL, "%s\n", path); + obdstripe = 1; + } + + if (header && obdstripe == 1) { + lov_dump_user_lmm_header(lum, NULL, 0, header, quiet, NULL); + + llapi_printf(LLAPI_MSG_NORMAL, "lmm_extent_count: %x\n", + lumj->lmm_extent_count); + } + + if (body) { + unsigned long long start = -1, end = 0; + if (!quiet && obdstripe == 1) + llapi_printf(LLAPI_MSG_NORMAL, + "joined\tobdidx\t\t objid\t\tobjid\t\t " + "group\t\tstart\t\tend\n"); + for (i = 0; i < lumj->lmm_stripe_count; i++) { + int idx = lumj->lmm_objects[i].l_ost_idx; + long long oid = lumj->lmm_objects[i].l_object_id; + long long gr = lumj->lmm_objects[i].l_object_gr; + if (obdindex == OBD_NOT_FOUND || obdindex == idx) + llapi_printf(LLAPI_MSG_NORMAL, + "\t%6u\t%14llu\t%#13llx\t%14llu%s", + idx, oid, oid, gr, + obdindex == idx ? " *" : ""); + if (start != lumj->lmm_objects[i].l_extent_start || + end != lumj->lmm_objects[i].l_extent_end) { + start = lumj->lmm_objects[i].l_extent_start; + llapi_printf(LLAPI_MSG_NORMAL,"\t%14llu",start); + end = lumj->lmm_objects[i].l_extent_end; + if (end == (unsigned long long)-1) + llapi_printf(LLAPI_MSG_NORMAL, + "\t\tEOF\n"); + else + llapi_printf(LLAPI_MSG_NORMAL, + "\t\t%llu\n", end); + } else { + llapi_printf(LLAPI_MSG_NORMAL, "\t\t\t\t\n"); + } + } + llapi_printf(LLAPI_MSG_NORMAL, "\n"); + } +} + +void llapi_lov_dump_user_lmm(struct find_param *param, + char *path, int is_dir) +{ + switch(*(__u32 *)¶m->lmd->lmd_lmm) { /* lum->lmm_magic */ + case LOV_USER_MAGIC_V1: + lov_dump_user_lmm_v1v3(¶m->lmd->lmd_lmm, NULL, + param->lmd->lmd_lmm.lmm_objects, + path, is_dir, + param->obdindex, param->quiet, + param->verbose, + (param->verbose || !param->obduuid)); + break; + case LOV_USER_MAGIC_JOIN: + lov_dump_user_lmm_join(¶m->lmd->lmd_lmm, path, is_dir, + param->obdindex, param->quiet, + param->verbose, + (param->verbose || !param->obduuid)); + break; + case LOV_USER_MAGIC_V3: { + char pool_name[LOV_MAXPOOLNAME + 1]; + struct lov_user_ost_data_v1 *objects; + struct lov_user_md_v3 *lmmv3 = (void *)¶m->lmd->lmd_lmm; + + strncpy(pool_name, lmmv3->lmm_pool_name, LOV_MAXPOOLNAME); + pool_name[LOV_MAXPOOLNAME] = '\0'; + objects = lmmv3->lmm_objects; + lov_dump_user_lmm_v1v3(¶m->lmd->lmd_lmm, pool_name, + objects, path, is_dir, + param->obdindex, param->quiet, + param->verbose, + (param->verbose || !param->obduuid)); + break; + } + default: + llapi_printf(LLAPI_MSG_NORMAL, "unknown lmm_magic: %#x " + "(expecting one of %#x %#x %#x)\n", + *(__u32 *)¶m->lmd->lmd_lmm, + LOV_USER_MAGIC_V1, LOV_USER_MAGIC_JOIN, + LOV_USER_MAGIC_V3); + return; + } +} + +int llapi_file_get_stripe(const char *path, struct lov_user_md *lum) +{ + const char *fname; + char *dname; + int fd, rc = 0; + + fname = strrchr(path, '/'); + + /* It should be a file (or other non-directory) */ + if (fname == NULL) { + dname = (char *)malloc(2); + if (dname == NULL) + return ENOMEM; + strcpy(dname, "."); + fname = (char *)path; + } else { + dname = (char *)malloc(fname - path + 1); + if (dname == NULL) + return ENOMEM; + strncpy(dname, path, fname - path); + dname[fname - path] = '\0'; + fname++; + } + + if ((fd = open(dname, O_RDONLY)) == -1) { + rc = errno; + free(dname); + return rc; + } + + strcpy((char *)lum, fname); + if (ioctl(fd, IOC_MDC_GETFILESTRIPE, (void *)lum) == -1) + rc = errno; + + if (close(fd) == -1 && rc == 0) + rc = errno; + + free(dname); + + return rc; +} + +int llapi_file_lookup(int dirfd, const char *name) +{ + struct obd_ioctl_data data = { 0 }; + char rawbuf[8192]; + char *buf = rawbuf; + int rc; + + if (dirfd < 0 || name == NULL) + return -EINVAL; + + data.ioc_version = OBD_IOCTL_VERSION; + data.ioc_len = sizeof(data); + data.ioc_inlbuf1 = (char *)name; + data.ioc_inllen1 = strlen(name) + 1; + + rc = obd_ioctl_pack(&data, &buf, sizeof(rawbuf)); + if (rc) { + llapi_err(LLAPI_MSG_ERROR, + "error: IOC_MDC_LOOKUP pack failed for '%s': rc %d", + name, rc); + return rc; + } + + return ioctl(dirfd, IOC_MDC_LOOKUP, buf); +} + +int llapi_mds_getfileinfo(char *path, DIR *parent, + struct lov_user_mds_data *lmd) +{ + lstat_t *st = &lmd->lmd_st; + char *fname = strrchr(path, '/'); + int ret = 0; + + if (parent == NULL) + return -EINVAL; + + fname = (fname == NULL ? path : fname + 1); + /* retrieve needed file info */ + strncpy((char *)lmd, fname, + lov_mds_md_size(MAX_LOV_UUID_COUNT, LOV_MAGIC)); + ret = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO, (void *)lmd); + + if (ret) { + if (errno == ENOTTY) { + /* ioctl is not supported, it is not a lustre fs. + * Do the regular lstat(2) instead. */ + ret = lstat_f(path, st); + if (ret) { + llapi_err(LLAPI_MSG_ERROR, + "error: %s: lstat failed for %s", + __FUNCTION__, path); + return ret; + } + } else if (errno == ENOENT) { + llapi_err(LLAPI_MSG_WARN, + "warning: %s: %s does not exist", + __FUNCTION__, path); + return -ENOENT; + } else { + llapi_err(LLAPI_MSG_ERROR, + "error: %s: IOC_MDC_GETFILEINFO failed for %s", + __FUNCTION__, path); + return ret; + } + } + + return 0; +} + +static DIR *opendir_parent(char *path) +{ + DIR *parent; + char *fname; + char c; + + fname = strrchr(path, '/'); + if (fname == NULL) + return opendir("."); + + c = fname[1]; + fname[1] = '\0'; + parent = opendir(path); + fname[1] = c; + return parent; +} + +static int llapi_semantic_traverse(char *path, int size, DIR *parent, + semantic_func_t sem_init, + semantic_func_t sem_fini, void *data, + cfs_dirent_t *de) +{ + cfs_dirent_t *dent; + int len, ret; + DIR *d, *p = NULL; + + ret = 0; + len = strlen(path); + + d = opendir(path); + if (!d && errno != ENOTDIR) { + llapi_err(LLAPI_MSG_ERROR, "%s: Failed to open '%s'", + __FUNCTION__, path); + return -EINVAL; + } else if (!d && !parent) { + /* ENOTDIR. Open the parent dir. */ + p = opendir_parent(path); + if (!p) + GOTO(out, ret = -EINVAL); + } + + if (sem_init && (ret = sem_init(path, parent ?: p, d, data, de))) + goto err; + + if (!d) + GOTO(out, ret = 0); + + while ((dent = readdir64(d)) != NULL) { + ((struct find_param *)data)->have_fileinfo = 0; + + if (!strcmp(dent->d_name, ".") || !strcmp(dent->d_name, "..")) + continue; + + /* Don't traverse .lustre directory */ + if (!(strcmp(dent->d_name, dot_lustre_name))) + continue; + + path[len] = 0; + if ((len + dent->d_reclen + 2) > size) { + llapi_err(LLAPI_MSG_ERROR, + "error: %s: string buffer is too small", + __FUNCTION__); + break; + } + strcat(path, "/"); + strcat(path, dent->d_name); + + if (dent->d_type == DT_UNKNOWN) { + lstat_t *st = &((struct find_param *)data)->lmd->lmd_st; + + ret = llapi_mds_getfileinfo(path, d, + ((struct find_param *)data)->lmd); + if (ret == 0) { + ((struct find_param *)data)->have_fileinfo = 1; + dent->d_type = + llapi_filetype_dir_table[st->st_mode & + S_IFMT]; + } + if (ret == -ENOENT) + continue; + } + + switch (dent->d_type) { + case DT_UNKNOWN: + llapi_err(LLAPI_MSG_ERROR, + "error: %s: '%s' is UNKNOWN type %d", + __FUNCTION__, dent->d_name, dent->d_type); + break; + case DT_DIR: + ret = llapi_semantic_traverse(path, size, d, sem_init, + sem_fini, data, dent); + if (ret < 0) + goto out; + break; + default: + ret = 0; + if (sem_init) { + ret = sem_init(path, d, NULL, data, dent); + if (ret < 0) + goto out; + } + if (sem_fini && ret == 0) + sem_fini(path, d, NULL, data, dent); + } + } + +out: + path[len] = 0; + + if (sem_fini) + sem_fini(path, parent, d, data, de); +err: + if (d) + closedir(d); + if (p) + closedir(p); + return ret; +} + +/* Check if the value matches 1 of the given criteria (e.g. --atime +/-N). + * @mds indicates if this is MDS timestamps and there are attributes on OSTs. + * + * The result is -1 if it does not match, 0 if not yet clear, 1 if matches. + * The table below gives the answers for the specified parameters (value and + * sign), 1st column is the answer for the MDS value, the 2nd is for the OST: + * -------------------------------------- + * 1 | file > limit; sign > 0 | -1 / -1 | + * 2 | file = limit; sign > 0 | ? / 1 | + * 3 | file < limit; sign > 0 | ? / 1 | + * 4 | file > limit; sign = 0 | -1 / -1 | + * 5 | file = limit; sign = 0 | ? / 1 | <- (see the Note below) + * 6 | file < limit; sign = 0 | ? / -1 | + * 7 | file > limit; sign < 0 | 1 / 1 | + * 8 | file = limit; sign < 0 | ? / -1 | + * 9 | file < limit; sign < 0 | ? / -1 | + * -------------------------------------- + * Note: 5th actually means that the value is within the interval + * (limit - margin, limit]. */ +static int find_value_cmp(unsigned int file, unsigned int limit, int sign, + unsigned long long margin, int mds) +{ + if (sign > 0) { + if (file < limit) + return mds ? 0 : 1; + } + + if (sign == 0) { + if (file <= limit && file + margin > limit) + return mds ? 0 : 1; + if (file + margin <= limit) + return mds ? 0 : -1; + } + + if (sign < 0) { + if (file > limit) + return 1; + if (mds) + return 0; + } + + return -1; +} + +/* Check if the file time matches all the given criteria (e.g. --atime +/-N). + * Return -1 or 1 if file timestamp does not or does match the given criteria + * correspondingly. Return 0 if the MDS time is being checked and there are + * attributes on OSTs and it is not yet clear if the timespamp matches. + * + * If 0 is returned, we need to do another RPC to the OSTs to obtain the + * updated timestamps. */ +static int find_time_check(lstat_t *st, struct find_param *param, int mds) +{ + int ret; + int rc = 0; + + /* Check if file is accepted. */ + if (param->atime) { + ret = find_value_cmp(st->st_atime, param->atime, + param->asign, 24 * 60 * 60, mds); + if (ret < 0) + return ret; + rc = ret; + } + + if (param->mtime) { + ret = find_value_cmp(st->st_mtime, param->mtime, + param->msign, 24 * 60 * 60, mds); + if (ret < 0) + return ret; + + /* If the previous check matches, but this one is not yet clear, + * we should return 0 to do an RPC on OSTs. */ + if (rc == 1) + rc = ret; + } + + if (param->ctime) { + ret = find_value_cmp(st->st_ctime, param->ctime, + param->csign, 24 * 60 * 60, mds); + if (ret < 0) + return ret; + + /* If the previous check matches, but this one is not yet clear, + * we should return 0 to do an RPC on OSTs. */ + if (rc == 1) + rc = ret; + } + + return rc; +} + +static int cb_find_init(char *path, DIR *parent, DIR *dir, + void *data, cfs_dirent_t *de) +{ + struct find_param *param = (struct find_param *)data; + int decision = 1; /* 1 is accepted; -1 is rejected. */ + lstat_t *st = ¶m->lmd->lmd_st; + int lustre_fs = 1; + int checked_type = 0; + int ret = 0; + + LASSERT(parent != NULL || dir != NULL); + + param->lmd->lmd_lmm.lmm_stripe_count = 0; + + /* If a regular expression is presented, make the initial decision */ + if (param->pattern != NULL) { + char *fname = strrchr(path, '/'); + fname = (fname == NULL ? path : fname + 1); + ret = fnmatch(param->pattern, fname, 0); + if ((ret == FNM_NOMATCH && !param->exclude_pattern) || + (ret == 0 && param->exclude_pattern)) + goto decided; + } + + /* See if we can check the file type from the dirent. */ + if (param->type && de != NULL && de->d_type != DT_UNKNOWN && + de->d_type < DT_MAX) { + checked_type = 1; + if (llapi_dir_filetype_table[de->d_type] == param->type) { + if (param->exclude_type) + goto decided; + } else { + if (!param->exclude_type) + goto decided; + } + } + + + /* If a time or OST should be checked, the decision is not taken yet. */ + if (param->atime || param->ctime || param->mtime || param->obduuid || + param->size) + decision = 0; + + ret = 0; + /* Request MDS for the stat info. */ + if (param->have_fileinfo == 0) { + if (dir) { + /* retrieve needed file info */ + ret = ioctl(dirfd(dir), LL_IOC_MDC_GETINFO, + (void *)param->lmd); + } else { + char *fname = strrchr(path, '/'); + fname = (fname == NULL ? path : fname + 1); + + /* retrieve needed file info */ + strncpy((char *)param->lmd, fname, param->lumlen); + ret = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO, + (void *)param->lmd); + } + } + + if (ret) { + if (errno == ENOTTY) { + /* ioctl is not supported, it is not a lustre fs. + * Do the regular lstat(2) instead. */ + lustre_fs = 0; + ret = lstat_f(path, st); + if (ret) { + llapi_err(LLAPI_MSG_ERROR, + "error: %s: lstat failed for %s", + __FUNCTION__, path); + return ret; + } + } else if (errno == ENOENT) { + llapi_err(LLAPI_MSG_WARN, + "warning: %s: %s does not exist", + __FUNCTION__, path); + goto decided; + } else { + llapi_err(LLAPI_MSG_ERROR,"error: %s: %s failed for %s", + __FUNCTION__, dir ? "LL_IOC_MDC_GETINFO" : + "IOC_MDC_GETFILEINFO", path); + return ret; + } + } + + if (param->type && !checked_type) { + if ((st->st_mode & S_IFMT) == param->type) { + if (param->exclude_type) + goto decided; + } else { + if (!param->exclude_type) + goto decided; + } + } + + /* Prepare odb. */ + if (param->obduuid) { + if (lustre_fs && param->got_uuids && + param->st_dev != st->st_dev) { + /* A lustre/lustre mount point is crossed. */ + param->got_uuids = 0; + param->obds_printed = 0; + param->obdindex = OBD_NOT_FOUND; + } + + if (lustre_fs && !param->got_uuids) { + ret = setup_obd_indexes(dir ? dir : parent, param); + if (ret) + return ret; + + param->st_dev = st->st_dev; + } else if (!lustre_fs && param->got_uuids) { + /* A lustre/non-lustre mount point is crossed. */ + param->got_uuids = 0; + param->obdindex = OBD_NOT_FOUND; + } + } + + /* If an OBD UUID is specified but no one matches, skip this file. */ + if (param->obduuid && param->obdindex == OBD_NOT_FOUND) + goto decided; + + /* If a OST UUID is given, and some OST matches, check it here. */ + if (param->obdindex != OBD_NOT_FOUND) { + if (!S_ISREG(st->st_mode)) + goto decided; + + /* Only those files should be accepted, which have a + * stripe on the specified OST. */ + if (!param->lmd->lmd_lmm.lmm_stripe_count) { + goto decided; + } else { + int i, j; + struct lov_user_ost_data_v1 *lmm_objects; + + if (param->lmd->lmd_lmm.lmm_magic == + LOV_USER_MAGIC_V3) { + struct lov_user_md_v3 *lmmv3 = + (void *)¶m->lmd->lmd_lmm; + + lmm_objects = lmmv3->lmm_objects; + } else { + lmm_objects = param->lmd->lmd_lmm.lmm_objects; + } + + for (i = 0; + i < param->lmd->lmd_lmm.lmm_stripe_count; i++) { + for (j = 0; j < param->num_obds; j++) { + if (param->obdindexes[j] == + lmm_objects[i].l_ost_idx) + goto obd_matches; + } + } + + if (i == param->lmd->lmd_lmm.lmm_stripe_count) + goto decided; + } + } + + if (param->check_uid) { + if (st->st_uid == param->uid) { + if (param->exclude_uid) + goto decided; + } else { + if (!param->exclude_uid) + goto decided; + } + } + + if (param->check_gid) { + if (st->st_gid == param->gid) { + if (param->exclude_gid) + goto decided; + } else { + if (!param->exclude_gid) + goto decided; + } + } + + if (param->check_pool) { + struct lov_user_md_v3 *lmmv3 = (void *)¶m->lmd->lmd_lmm; + + /* empty requested pool is taken as no pool search => V1 */ + if (((param->lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V1) && + (param->poolname[0] == '\0')) || + ((param->lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V3) && + (strncmp(lmmv3->lmm_pool_name, + param->poolname, LOV_MAXPOOLNAME) == 0)) || + ((param->lmd->lmd_lmm.lmm_magic == LOV_USER_MAGIC_V3) && + (strcmp(param->poolname, "*") == 0))) { + if (param->exclude_pool) + goto decided; + } else { + if (!param->exclude_pool) + goto decided; + } + } + + /* Check the time on mds. */ + if (!decision) { + int for_mds; + + for_mds = lustre_fs ? (S_ISREG(st->st_mode) && + param->lmd->lmd_lmm.lmm_stripe_count) + : 0; + decision = find_time_check(st, param, for_mds); + } + +obd_matches: + /* If file still fits the request, ask ost for updated info. + The regular stat is almost of the same speed as some new + 'glimpse-size-ioctl'. */ + if (!decision && S_ISREG(st->st_mode) && + (param->lmd->lmd_lmm.lmm_stripe_count || param->size)) { + if (param->obdindex != OBD_NOT_FOUND) { + /* Check whether the obd is active or not, if it is + * not active, just print the object affected by this + * failed ost + * */ + struct obd_statfs stat_buf; + struct obd_uuid uuid_buf; + + memset(&stat_buf, 0, sizeof(struct obd_statfs)); + memset(&uuid_buf, 0, sizeof(struct obd_uuid)); + ret = llapi_obd_statfs(path, LL_STATFS_LOV, + param->obdindex, &stat_buf, + &uuid_buf); + if (ret) { + if (ret == -ENODATA || ret == -ENODEV + || ret == -EIO) + errno = EIO; + llapi_printf(LLAPI_MSG_NORMAL, + "obd_uuid: %s failed %s ", + param->obduuid->uuid, + strerror(errno)); + goto print_path; + } + } + if (dir) { + ret = ioctl(dirfd(dir), IOC_LOV_GETINFO, + (void *)param->lmd); + } else if (parent) { + ret = ioctl(dirfd(parent), IOC_LOV_GETINFO, + (void *)param->lmd); + } + + if (ret) { + if (errno == ENOENT) { + llapi_err(LLAPI_MSG_ERROR, + "warning: %s: %s does not exist", + __FUNCTION__, path); + goto decided; + } else { + llapi_err(LLAPI_MSG_ERROR, + "%s: IOC_LOV_GETINFO on %s failed", + __FUNCTION__, path); + return ret; + } + } + + /* Check the time on osc. */ + decision = find_time_check(st, param, 0); + if (decision == -1) + goto decided; + } + + if (param->size) + decision = find_value_cmp(st->st_size, param->size, + param->size_sign, param->size_units, + 0); + +print_path: + if (decision != -1) { + llapi_printf(LLAPI_MSG_NORMAL, "%s", path); + if (param->zeroend) + llapi_printf(LLAPI_MSG_NORMAL, "%c", '\0'); + else + llapi_printf(LLAPI_MSG_NORMAL, "\n"); + } + +decided: + /* Do not get down anymore? */ + if (param->depth == param->maxdepth) + return 1; + + param->depth++; + return 0; +} + +static int cb_common_fini(char *path, DIR *parent, DIR *d, void *data, + cfs_dirent_t *de) +{ + struct find_param *param = (struct find_param *)data; + param->depth--; + return 0; +} + +int llapi_find(char *path, struct find_param *param) +{ + char *buf; + int ret, len = strlen(path); + + if (len > PATH_MAX) { + llapi_err(LLAPI_MSG_ERROR, "%s: Path name '%s' is too long", + __FUNCTION__, path); + return -EINVAL; + } + + buf = (char *)malloc(PATH_MAX + 1); + if (!buf) + return -ENOMEM; + + ret = common_param_init(param); + if (ret) { + free(buf); + return ret; + } + + param->depth = 0; + + strncpy(buf, path, PATH_MAX + 1); + ret = llapi_semantic_traverse(buf, PATH_MAX + 1, NULL, cb_find_init, + cb_common_fini, param, NULL); + + find_param_fini(param); + free(buf); + return ret < 0 ? ret : 0; +} + +static int cb_getstripe(char *path, DIR *parent, DIR *d, void *data, + cfs_dirent_t *de) +{ + struct find_param *param = (struct find_param *)data; + int ret = 0; + + LASSERT(parent != NULL || d != NULL); + + /* Prepare odb. */ + if (!param->got_uuids) { + ret = setup_obd_uuid(d ? d : parent, path, param); + if (ret) + return ret; + } + + if (d) { + ret = ioctl(dirfd(d), LL_IOC_LOV_GETSTRIPE, + (void *)¶m->lmd->lmd_lmm); + } else if (parent) { + char *fname = strrchr(path, '/'); + fname = (fname == NULL ? path : fname + 1); + + strncpy((char *)¶m->lmd->lmd_lmm, fname, param->lumlen); + ret = ioctl(dirfd(parent), IOC_MDC_GETFILESTRIPE, + (void *)¶m->lmd->lmd_lmm); + } + + if (ret) { + if (errno == ENODATA) { + if (!param->obduuid && !param->quiet) + llapi_printf(LLAPI_MSG_NORMAL, + "%s has no stripe info\n", path); + goto out; + } else if (errno == ENOTTY) { + llapi_err(LLAPI_MSG_ERROR, + "%s: '%s' not on a Lustre fs?", + __FUNCTION__, path); + } else if (errno == ENOENT) { + llapi_err(LLAPI_MSG_WARN, + "warning: %s: %s does not exist", + __FUNCTION__, path); + goto out; + } else { + llapi_err(LLAPI_MSG_ERROR, + "error: %s: %s failed for %s", + __FUNCTION__, d ? "LL_IOC_LOV_GETSTRIPE" : + "IOC_MDC_GETFILESTRIPE", path); + } + + return ret; + } + + llapi_lov_dump_user_lmm(param, path, d ? 1 : 0); +out: + /* Do not get down anymore? */ + if (param->depth == param->maxdepth) + return 1; + + param->depth++; + return 0; +} + +int llapi_getstripe(char *path, struct find_param *param) +{ + char *buf; + int ret = 0, len = strlen(path); + + if (len > PATH_MAX) { + llapi_err(LLAPI_MSG_ERROR, + "%s: Path name '%s' is too long", + __FUNCTION__, path); + return -EINVAL; + } + + buf = (char *)malloc(PATH_MAX + 1); + if (!buf) + return -ENOMEM; + + ret = common_param_init(param); + if (ret) { + free(buf); + return ret; + } + + param->depth = 0; + + strncpy(buf, path, PATH_MAX + 1); + ret = llapi_semantic_traverse(buf, PATH_MAX + 1, NULL, cb_getstripe, + cb_common_fini, param, NULL); + find_param_fini(param); + free(buf); + return ret < 0 ? ret : 0; +} + +int llapi_obd_statfs(char *path, __u32 type, __u32 index, + struct obd_statfs *stat_buf, + struct obd_uuid *uuid_buf) +{ + int fd; + char raw[OBD_MAX_IOCTL_BUFFER] = {'\0'}; + char *rawbuf = raw; + struct obd_ioctl_data data = { 0 }; + int rc = 0; + + data.ioc_inlbuf1 = (char *)&type; + data.ioc_inllen1 = sizeof(__u32); + data.ioc_inlbuf2 = (char *)&index; + data.ioc_inllen2 = sizeof(__u32); + data.ioc_pbuf1 = (char *)stat_buf; + data.ioc_plen1 = sizeof(struct obd_statfs); + data.ioc_pbuf2 = (char *)uuid_buf; + data.ioc_plen2 = sizeof(struct obd_uuid); + + if ((rc = obd_ioctl_pack(&data, &rawbuf, sizeof(raw))) != 0) { + llapi_err(LLAPI_MSG_ERROR, + "llapi_obd_statfs: error packing ioctl data"); + return rc; + } + + fd = open(path, O_RDONLY); + if (errno == EISDIR) + fd = open(path, O_DIRECTORY | O_RDONLY); + + if (fd < 0) { + rc = errno ? -errno : -EBADF; + llapi_err(LLAPI_MSG_ERROR, "error: %s: opening '%s'", + __FUNCTION__, path); + return rc; + } + rc = ioctl(fd, IOC_OBD_STATFS, (void *)rawbuf); + if (rc) + rc = errno ? -errno : -EINVAL; + + close(fd); + return rc; +} + +#define MAX_STRING_SIZE 128 +#define DEVICES_LIST "/proc/fs/lustre/devices" + +int llapi_ping(char *obd_type, char *obd_name) +{ + char path[MAX_STRING_SIZE]; + char buf[1]; + int rc, fd; + + snprintf(path, MAX_STRING_SIZE, "/proc/fs/lustre/%s/%s/ping", + obd_type, obd_name); + + fd = open(path, O_WRONLY); + if (fd < 0) { + rc = errno; + llapi_err(LLAPI_MSG_ERROR, "error opening %s", path); + return rc; + } + + rc = write(fd, buf, 1); + close(fd); + + if (rc == 1) + return 0; + return rc; +} + +int llapi_target_iterate(int type_num, char **obd_type,void *args,llapi_cb_t cb) +{ + char buf[MAX_STRING_SIZE]; FILE *fp = fopen(DEVICES_LIST, "r"); - + int i, rc = 0; + if (fp == NULL) { - fprintf(stderr, "error: %s could not open file " - DEVICES_LIST " .\n", strerror(rc = errno)); + rc = errno; + llapi_err(LLAPI_MSG_ERROR, "error: opening "DEVICES_LIST); + return rc; + } + + while (fgets(buf, sizeof(buf), fp) != NULL) { + char *obd_type_name = NULL; + char *obd_name = NULL; + char *obd_uuid = NULL; + char *bufp = buf; + struct obd_ioctl_data datal = { 0, }; + struct obd_statfs osfs_buffer; + + while(bufp[0] == ' ') + ++bufp; + + for(i = 0; i < 3; i++) { + obd_type_name = strsep(&bufp, " "); + } + obd_name = strsep(&bufp, " "); + obd_uuid = strsep(&bufp, " "); + + memset(&osfs_buffer, 0, sizeof (osfs_buffer)); + + datal.ioc_pbuf1 = (char *)&osfs_buffer; + datal.ioc_plen1 = sizeof(osfs_buffer); + + for (i = 0; i < type_num; i++) { + if (strcmp(obd_type_name, obd_type[i]) != 0) + continue; + + cb(obd_type_name, obd_name, obd_uuid, args); + } + } + fclose(fp); + return rc; +} + +static void do_target_check(char *obd_type_name, char *obd_name, + char *obd_uuid, void *args) +{ + int rc; + + rc = llapi_ping(obd_type_name, obd_name); + if (rc == ENOTCONN) { + llapi_printf(LLAPI_MSG_NORMAL, "%s inactive.\n", obd_name); + } else if (rc) { + llapi_err(LLAPI_MSG_ERROR, "error: check '%s'", obd_name); + } else { + llapi_printf(LLAPI_MSG_NORMAL, "%s active.\n", obd_name); + } +} + +int llapi_target_check(int type_num, char **obd_type, char *dir) +{ + return llapi_target_iterate(type_num, obd_type, NULL, do_target_check); +} + +#undef MAX_STRING_SIZE + +int llapi_catinfo(char *dir, char *keyword, char *node_name) +{ + char raw[OBD_MAX_IOCTL_BUFFER]; + char out[LLOG_CHUNK_SIZE]; + char *buf = raw; + struct obd_ioctl_data data = { 0 }; + char key[30]; + DIR *root; + int rc; + + sprintf(key, "%s", keyword); + memset(raw, 0, sizeof(raw)); + memset(out, 0, sizeof(out)); + data.ioc_inlbuf1 = key; + data.ioc_inllen1 = strlen(key) + 1; + if (node_name) { + data.ioc_inlbuf2 = node_name; + data.ioc_inllen2 = strlen(node_name) + 1; + } + data.ioc_pbuf1 = out; + data.ioc_plen1 = sizeof(out); + rc = obd_ioctl_pack(&data, &buf, sizeof(raw)); + if (rc) + return rc; + + root = opendir(dir); + if (root == NULL) { + rc = errno; + llapi_err(LLAPI_MSG_ERROR, "open %s failed", dir); + return rc; + } + + rc = ioctl(dirfd(root), OBD_IOC_LLOG_CATINFO, buf); + if (rc) + llapi_err(LLAPI_MSG_ERROR, "ioctl OBD_IOC_CATINFO failed"); + else + llapi_printf(LLAPI_MSG_NORMAL, "%s", data.ioc_pbuf1); + + closedir(root); + return rc; +} + +/* Is this a lustre fs? */ +int llapi_is_lustre_mnttype(const char *type) +{ + return (strcmp(type, "lustre") == 0 || strcmp(type,"lustre_lite") == 0); +} + +/* Is this a lustre client fs? */ +int llapi_is_lustre_mnt(struct mntent *mnt) +{ + return (llapi_is_lustre_mnttype(mnt->mnt_type) && + strstr(mnt->mnt_fsname, ":/") != NULL); +} + +int llapi_quotacheck(char *mnt, int check_type) +{ + DIR *root; + int rc; + + root = opendir(mnt); + if (!root) { + llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt); + return -1; + } + + rc = ioctl(dirfd(root), LL_IOC_QUOTACHECK, check_type); + + closedir(root); + return rc; +} + +int llapi_poll_quotacheck(char *mnt, struct if_quotacheck *qchk) +{ + DIR *root; + int poll_intvl = 2; + int rc; + + root = opendir(mnt); + if (!root) { + llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt); + return -1; + } + + while (1) { + rc = ioctl(dirfd(root), LL_IOC_POLL_QUOTACHECK, qchk); + if (!rc) + break; + sleep(poll_intvl); + if (poll_intvl < 30) + poll_intvl *= 2; + } + + closedir(root); + return rc; +} + +int llapi_quotactl(char *mnt, struct if_quotactl *qctl) +{ + DIR *root; + int rc; + + root = opendir(mnt); + if (!root) { + llapi_err(LLAPI_MSG_ERROR, "open %s failed", mnt); + return -1; + } + + rc = ioctl(dirfd(root), LL_IOC_QUOTACTL, qctl); + + closedir(root); + return rc; +} + +static int cb_quotachown(char *path, DIR *parent, DIR *d, void *data, + cfs_dirent_t *de) +{ + struct find_param *param = (struct find_param *)data; + lstat_t *st; + int rc; + + LASSERT(parent != NULL || d != NULL); + + if (d) { + rc = ioctl(dirfd(d), LL_IOC_MDC_GETINFO, + (void *)param->lmd); + } else if (parent) { + char *fname = strrchr(path, '/'); + fname = (fname == NULL ? path : fname + 1); + + strncpy((char *)param->lmd, fname, param->lumlen); + rc = ioctl(dirfd(parent), IOC_MDC_GETFILEINFO, + (void *)param->lmd); + } else { + return 0; + } + + if (rc) { + if (errno == ENODATA) { + if (!param->obduuid && !param->quiet) + llapi_err(LLAPI_MSG_ERROR, + "%s has no stripe info", path); + rc = 0; + } else if (errno == ENOENT) { + llapi_err(LLAPI_MSG_ERROR, + "warning: %s: %s does not exist", + __FUNCTION__, path); + rc = 0; + } else if (errno != EISDIR) { + rc = errno; + llapi_err(LLAPI_MSG_ERROR, "%s ioctl failed for %s.", + d ? "LL_IOC_MDC_GETINFO" : + "IOC_MDC_GETFILEINFO", path); + } return rc; } - while (fgets(buf, sizeof(buf), fp) != NULL) { - char *obd_type_name = NULL; - char *obd_name = NULL; + st = ¶m->lmd->lmd_st; + + /* libc chown() will do extra check, and if the real owner is + * the same as the ones to set, it won't fall into kernel, so + * invoke syscall directly. */ + rc = syscall(SYS_chown, path, -1, -1); + if (rc) + llapi_err(LLAPI_MSG_ERROR,"error: chown %s (%u,%u)", path); + + rc = chmod(path, st->st_mode); + if (rc) + llapi_err(LLAPI_MSG_ERROR, "error: chmod %s (%hu)", + path, st->st_mode); + + return rc; +} + +int llapi_quotachown(char *path, int flag) +{ + struct find_param param; + char *buf; + int ret = 0, len = strlen(path); + + if (len > PATH_MAX) { + llapi_err(LLAPI_MSG_ERROR, "%s: Path name '%s' is too long", + __FUNCTION__, path); + return -EINVAL; + } + + buf = (char *)malloc(PATH_MAX + 1); + if (!buf) + return -ENOMEM; + + memset(¶m, 0, sizeof(param)); + param.recursive = 1; + param.verbose = 0; + param.quiet = 1; + + ret = common_param_init(¶m); + if (ret) + goto out; + + strncpy(buf, path, PATH_MAX + 1); + ret = llapi_semantic_traverse(buf, PATH_MAX + 1, NULL, cb_quotachown, + NULL, ¶m, NULL); +out: + find_param_fini(¶m); + free(buf); + return ret; +} + +#include +#include +#include +#include +#include +#include + +static int rmtacl_notify(int ops) +{ + FILE *fp; + struct mntent *mnt; + int found = 0, fd, rc; + + fp = setmntent(MOUNTED, "r"); + if (fp == NULL) { + perror("setmntent"); + return -1; + } + + while (1) { + mnt = getmntent(fp); + if (!mnt) + break; + + if (!llapi_is_lustre_mnt(mnt)) + continue; + + fd = open(mnt->mnt_dir, O_RDONLY | O_DIRECTORY); + if (fd < 0) { + perror("open"); + return -1; + } + + rc = ioctl(fd, LL_IOC_RMTACL, ops); + if (rc < 0) { + perror("ioctl"); + return -1; + } + + found++; + } + endmntent(fp); + return found; +} + +static char *next_token(char *p, int div) +{ + if (p == NULL) + return NULL; - char rawbuf[OBD_MAX_IOCTL_BUFFER]; - char *bufl = rawbuf; - char *bufp = buf; - int max = sizeof(rawbuf); - struct obd_ioctl_data datal; - struct obd_statfs osfs_buffer; - - while(bufp[0] == ' ') bufp += 1; - for(i = 0; i < 3; i++) { - obd_type_name = strsep(&bufp, " "); + if (div) + while (*p && *p != ':' && !isspace(*p)) + p++; + else + while (*p == ':' || isspace(*p)) + p++; + + return *p ? p : NULL; +} + +static int rmtacl_name2id(char *name, int is_user) +{ + if (is_user) { + struct passwd *pw; + + if ((pw = getpwnam(name)) == NULL) + return INVALID_ID; + else + return (int)(pw->pw_uid); + } else { + struct group *gr; + + if ((gr = getgrnam(name)) == NULL) + return INVALID_ID; + else + return (int)(gr->gr_gid); + } +} + +static int isodigit(int c) +{ + return (c >= '0' && c <= '7') ? 1 : 0; +} + +/* + * Whether the name is just digits string (uid/gid) already or not. + * Return value: + * 1: str is id + * 0: str is not id + */ +static int str_is_id(char *str) +{ + if (str == NULL) + return 0; + + if (*str == '0') { + str++; + if (*str == 'x' || *str == 'X') { /* for Hex. */ + if (!isxdigit(*(++str))) + return 0; + + while (isxdigit(*(++str))); + } else if (isodigit(*str)) { /* for Oct. */ + while (isodigit(*(++str))); } - obd_name = strsep(&bufp, " "); + } else if (isdigit(*str)) { /* for Dec. */ + while (isdigit(*(++str))); + } - memset (&osfs_buffer, 0, sizeof (osfs_buffer)); + return (*str == 0) ? 1 : 0; +} - memset(bufl, 0, sizeof(rawbuf)); - datal.ioc_pbuf1 = (char *)&osfs_buffer; - datal.ioc_plen1 = sizeof (osfs_buffer); +typedef struct { + char *name; + int length; + int is_user; + int next_token; +} rmtacl_name_t; + +#define RMTACL_OPTNAME(name) name, sizeof(name) - 1 + +static rmtacl_name_t rmtacl_namelist[] = { + { RMTACL_OPTNAME("user:"), 1, 0 }, + { RMTACL_OPTNAME("group:"), 0, 0 }, + { RMTACL_OPTNAME("default:user:"), 1, 0 }, + { RMTACL_OPTNAME("default:group:"), 0, 0 }, + /* for --tabular option */ + { RMTACL_OPTNAME("user"), 1, 1 }, + { RMTACL_OPTNAME("group"), 0, 1 }, + { 0 } +}; + +static int rgetfacl_output(char *str) +{ + char *start = NULL, *end = NULL; + int is_user = 0, n, id; + char c; + rmtacl_name_t *rn; + + if (str == NULL) + return -1; + + for (rn = rmtacl_namelist; rn->name; rn++) { + if(strncmp(str, rn->name, rn->length) == 0) { + if (!rn->next_token) + start = str + rn->length; + else + start = next_token(str + rn->length, 0); + is_user = rn->is_user; + break; + } + } + + end = next_token(start, 1); + if (end == NULL || start == end) { + n = printf("%s", str); + return n; + } - for (i=0;i 16 + */ +static int get_mdtname(char *name, char *format, char *buf) +{ + char suffix[]="-MDT0000"; + int len = strlen(name); + + if ((len > 5) && (strncmp(name + len - 5, "_UUID", 5) == 0)) { + name[len - 5] = '\0'; + len -= 5; } - data.ioc_pbuf1 = out; - data.ioc_plen1 = sizeof(out); - rc = obd_ioctl_pack(&data, &buf, sizeof(raw)); - if (rc) + + if (len > 8) { + if ((len <= 16) && strncmp(name + len - 8, "-MDT", 4) == 0) { + suffix[0] = '\0'; + } else { + /* Not enough room to add suffix */ + llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO, + "MDT name too long |%s|", name); + return -EINVAL; + } + } + + return sprintf(buf, format, name, suffix); +} + +/****** Changelog API ********/ +#define CHANGELOG_PRIV_MAGIC 0xCA8E1080 +struct changelog_private { + int magic; + int fd; + int flags; +}; + +/** Start reading from a changelog + * @param priv Opaque private control structure + * @param flags Start flags (e.g. follow) + * @param device Report changes recorded on this MDT + * @param startrec Report changes beginning with this record number + */ +int llapi_changelog_start(void **priv, int flags, const char *device, + long long startrec) +{ + struct changelog_private *cp; + char path[256]; + char mdtname[20]; + int rc, fd; + + if (device[0] == '/') + rc = llapi_search_fsname(device, mdtname); + else + strncpy(mdtname, device, sizeof(mdtname)); + + /* Use either the mdd changelog (preferred) or a client mdc changelog */ + if (get_mdtname(mdtname, + "/proc/fs/lustre/md[cd]/%s%s{,-mdc-*}/changelog", + path) < 0) + return -EINVAL; + rc = first_match(path, path); + if (rc) return rc; - - root = opendir(dir); - if (root == NULL) { - err_msg("open %s failed", dir); - return errno; + + if ((fd = open(path, O_RDONLY)) < 0) { + llapi_err(LLAPI_MSG_ERROR, "error: can't open |%s|\n", path); + return -errno; + } + + rc = lseek(fd, (off_t)startrec, SEEK_SET); + if (rc < 0) { + llapi_err(LLAPI_MSG_ERROR, "can't seek rc=%d\n", rc); + return -errno; } - rc = ioctl(dirfd(root), OBD_IOC_LLOG_CATINFO, buf); + cp = malloc(sizeof(*cp)); + if (cp == NULL) { + close(fd); + return -ENOMEM; + } + + cp->magic = CHANGELOG_PRIV_MAGIC; + cp->fd = fd; + cp->flags = flags; + *priv = cp; + + return 0; +} + +/** Finish reading from a changelog */ +int llapi_changelog_fini(void **priv) +{ + struct changelog_private *cp = (struct changelog_private *)*priv; + + if (!cp || (cp->magic != CHANGELOG_PRIV_MAGIC)) + return -EINVAL; + + close(cp->fd); + free(cp); + *priv = NULL; + return 0; +} + +static int pollwait(int fd) { + struct pollfd pfds[1]; + int rc; + + pfds[0].fd = fd; + pfds[0].events = POLLIN; + rc = poll(pfds, 1, -1); + return rc < 0 ? -errno : rc; +} + +/** Read the next changelog entry + * @param priv Opaque private control structure + * @param rech Changelog record handle; record will be allocated here + * @return 0 valid message received; rec is set + * <0 error code + * 1 EOF + */ +int llapi_changelog_recv(void *priv, struct changelog_rec **rech) +{ + struct changelog_private *cp = (struct changelog_private *)priv; + struct changelog_rec rec, *recp; + int rc = 0; + + if (!cp || (cp->magic != CHANGELOG_PRIV_MAGIC)) + return -EINVAL; + if (rech == NULL) + return -EINVAL; + +readrec: + /* Read in the rec to get the namelen */ + rc = read(cp->fd, &rec, sizeof(rec)); + if (rc < 0) + return -errno; + if (rc == 0) { + if (cp->flags && CHANGELOG_FLAG_FOLLOW) { + rc = pollwait(cp->fd); + if (rc < 0) + return rc; + goto readrec; + } + return 1; + } + + recp = malloc(sizeof(rec) + rec.cr_namelen); + if (recp == NULL) + return -ENOMEM; + memcpy(recp, &rec, sizeof(rec)); + rc = read(cp->fd, recp->cr_name, rec.cr_namelen); + if (rc < 0) { + free(recp); + llapi_err(LLAPI_MSG_ERROR, "Can't read entire filename"); + return -errno; + } + + *rech = recp; + return 0; +} + +/** Release the changelog record when done with it. */ +int llapi_changelog_free(struct changelog_rec **rech) +{ + if (*rech) + free(*rech); + *rech = NULL; + return 0; +} + +int llapi_changelog_clear(const char *mdtname, const char *idstr, + long long endrec) +{ + struct ioc_changelog_clear data; + char fsname[17]; + char *ptr; + int id, fd, index, rc; + + if (endrec < 0) { + llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO, + "can't purge negative records\n"); + return -EINVAL; + } + + id = strtol(idstr + strlen(CHANGELOG_USER_PREFIX), NULL, 10); + if ((id == 0) || (strncmp(idstr, CHANGELOG_USER_PREFIX, + strlen(CHANGELOG_USER_PREFIX)) != 0)) { + llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO, + "expecting id of the form '"CHANGELOG_USER_PREFIX + "'; got '%s'\n", idstr); + return -EINVAL; + } + + /* Take path, fsname, or MDTNAME. Assume MDT0000 in the former cases */ + if (mdtname[0] == '/') { + index = 0; + fd = open(mdtname, O_RDONLY | O_DIRECTORY | O_NONBLOCK); + rc = fd < 0 ? -errno : 0; + } else { + if (get_mdtname((char *)mdtname, "%s%s", fsname) < 0) + return -EINVAL; + ptr = fsname + strlen(fsname) - 8; + *ptr = '\0'; + index = strtol(ptr + 4, NULL, 10); + rc = get_root_path(WANT_FD | WANT_ERROR, fsname, &fd, NULL, -1); + } + if (rc < 0) { + llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO, + "Can't open %s: %d\n", mdtname, rc); + return rc; + } + + data.icc_mdtindex = index; + data.icc_id = id; + data.icc_recno = endrec; + rc = ioctl(fd, OBD_IOC_CHANGELOG_CLEAR, &data); if (rc) - err_msg("ioctl OBD_IOC_CATINFO failed"); - else - fprintf(stdout, "%s", data.ioc_pbuf1); - - closedir(root); + llapi_err(LLAPI_MSG_ERROR, "ioctl err %d", rc); + + close(fd); + return rc; +} + +int llapi_fid2path(const char *device, const char *fidstr, char *buf, + int buflen, long long *recno, int *linkno) +{ + char path[PATH_MAX]; + struct lu_fid fid; + struct getinfo_fid2path *gf; + int fd, rc; + + while (*fidstr == '[') + fidstr++; + + sscanf(fidstr, SFID, RFID(&fid)); + if (!fid_is_sane(&fid)) { + llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO, + "bad FID format [%s], should be "DFID"\n", + fidstr, (__u64)1, 2, 0); + return -EINVAL; + } + + /* Take path or fsname */ + if (device[0] == '/') { + strcpy(path, device); + } else { + rc = get_root_path(WANT_PATH | WANT_ERROR, (char *)device, + NULL, path, -1); + if (rc < 0) + return rc; + } + sprintf(path, "%s/%s/fid/%s", path, dot_lustre_name, fidstr); + fd = open(path, O_RDONLY | O_NONBLOCK); + if (fd < 0) + return -errno; + + gf = malloc(sizeof(*gf) + buflen); + gf->gf_fid = fid; + gf->gf_recno = *recno; + gf->gf_linkno = *linkno; + gf->gf_pathlen = buflen; + rc = ioctl(fd, OBD_IOC_FID2PATH, gf); + if (rc) { + llapi_err(LLAPI_MSG_ERROR, "ioctl err %d", rc); + } else { + memcpy(buf, gf->gf_path, gf->gf_pathlen); + *recno = gf->gf_recno; + *linkno = gf->gf_linkno; + } + + free(gf); + close(fd); + return rc; +} + +static int path2fid_from_lma(const char *path, lustre_fid *fid) +{ + char buf[512]; + struct lustre_mdt_attrs *lma; + int rc; + + rc = lgetxattr(path, XATTR_NAME_LMA, buf, sizeof(buf)); + if (rc < 0) + return -errno; + lma = (struct lustre_mdt_attrs *)buf; + fid_be_to_cpu(fid, &lma->lma_self_fid); + return 0; +} + +int llapi_path2fid(const char *path, lustre_fid *fid) +{ + int fd, rc; + + memset(fid, 0, sizeof(*fid)); + fd = open(path, O_RDONLY | O_NONBLOCK | O_NOFOLLOW); + if (fd < 0) { + if (errno == ELOOP) /* symbolic link */ + return path2fid_from_lma(path, fid); + return -errno; + } + + rc = ioctl(fd, LL_IOC_PATH2FID, fid) < 0 ? -errno : 0; + if (rc == -EINVAL) /* char special device */ + rc = path2fid_from_lma(path, fid); + + close(fd); + return rc; +} + +/****** HSM Copytool API ********/ +#define CT_PRIV_MAGIC 0xC0BE2001 +struct copytool_private { + int magic; + lustre_netlink lnl; + int archive_num_count; + int archive_nums[0]; +}; + +#include + +/** Register a copytool + * @param priv Opaque private control structure + * @param flags Open flags, currently unused (e.g. O_NONBLOCK) + * @param archive_nums Which archive numbers this copytool is responsible for + */ +int llapi_copytool_start(void **priv, int flags, int archive_num_count, + int *archive_nums) +{ + struct copytool_private *ct; + int rc; + + if (archive_num_count > 0 && archive_nums == NULL) { + llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO, + "NULL archive numbers"); + return -EINVAL; + } + + ct = malloc(sizeof(*ct) + + archive_num_count * sizeof(ct->archive_nums[0])); + if (ct == NULL) + return -ENOMEM; + + ct->magic = CT_PRIV_MAGIC; + ct->archive_num_count = archive_num_count; + if (ct->archive_num_count > 0) + memcpy(ct->archive_nums, archive_nums, archive_num_count * + sizeof(ct->archive_nums[0])); + + rc = libcfs_ulnl_start(&ct->lnl, LNL_GRP_HSM); + if (rc < 0) + goto out_err; + + *priv = ct; + return 0; + +out_err: + free(ct); + return rc; +} + +/** Deregister a copytool */ +int llapi_copytool_fini(void **priv) +{ + struct copytool_private *ct = (struct copytool_private *)*priv; + + if (!ct || (ct->magic != CT_PRIV_MAGIC)) + return -EINVAL; + + libcfs_ulnl_stop(&ct->lnl); + free(ct); + *priv = NULL; + return 0; +} + +/** Wait for the next hsm_action_list + * @param priv Opaque private control structure + * @param halh Action list handle, will be allocated here + * @param msgsize Number of bytes in the message, will be set here + * @return 0 valid message received; halh and msgsize are set + * <0 error code + */ +int llapi_copytool_recv(void *priv, struct hsm_action_list **halh, int *msgsize) +{ + struct copytool_private *ct = (struct copytool_private *)priv; + struct lnl_hdr *lnlh; + struct hsm_action_list *hal; + int rc = 0; + + if (!ct || (ct->magic != CT_PRIV_MAGIC)) + return -EINVAL; + if (halh == NULL || msgsize == NULL) + return -EINVAL; + + rc = libcfs_ulnl_msg_get(&ct->lnl, HAL_MAXSIZE, + LNL_TRANSPORT_HSM, &lnlh); + if (rc < 0) + return rc; + + /* Handle generic messages */ + if (lnlh->lnl_transport == LNL_TRANSPORT_GENERIC && + lnlh->lnl_msgtype == LNL_MSG_SHUTDOWN) { + rc = -ESHUTDOWN; + goto out_free; + } + + if (lnlh->lnl_transport != LNL_TRANSPORT_HSM || + lnlh->lnl_msgtype != HMT_ACTION_LIST) { + llapi_err(LLAPI_MSG_ERROR | LLAPI_MSG_NO_ERRNO, + "Unknown HSM message type %d:%d\n", + lnlh->lnl_transport, lnlh->lnl_msgtype); + rc = -EPROTO; + goto out_free; + } + + /* Our message is an hsm_action_list */ + + hal = (struct hsm_action_list *)(lnlh + 1); + + /* Check that we have registered for this archive # */ + for (rc = 0; rc < ct->archive_num_count; rc++) { + if (hal->hal_archive_num == ct->archive_nums[rc]) + break; + } + if (rc >= ct->archive_num_count) { + CDEBUG(D_INFO, "This copytool does not service archive #%d, " + "ignoring this request.\n", hal->hal_archive_num); + rc = 0; + goto out_free; + } + + *halh = hal; + *msgsize = lnlh->lnl_msglen - sizeof(*lnlh); + return 0; + +out_free: + libcfs_ulnl_msg_free(&lnlh); + *halh = NULL; + *msgsize = 0; return rc; } - + +/** Release the action list when done with it. */ +int llapi_copytool_free(struct hsm_action_list **hal) +{ + if (*hal) { + struct lnl_hdr *lnlh = (struct lnl_hdr *)*hal - 1; + libcfs_ulnl_msg_free(&lnlh); + } + *hal = NULL; + return 0; +} + + +