4 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
6 * (C) Copyright 2012 Commissariat a l'energie atomique et aux energies
9 * All rights reserved. This program and the accompanying materials
10 * are made available under the terms of the GNU Lesser General Public License
11 * (LGPL) version 2.1 or (at your discretion) any later version.
12 * (LGPL) version 2.1 accompanies this distribution, and is available at
13 * http://www.gnu.org/licenses/lgpl-2.1.html
15 * This library is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * Lesser General Public License for more details.
23 * lustre/utils/liblustreapi_hsm.c
25 * lustreapi library for hsm calls
27 * Author: Aurelien Degremont <aurelien.degremont@cea.fr>
28 * Author: JC Lafoucriere <jacques-charles.lafoucriere@cea.fr>
29 * Author: Thomas Leibovici <thomas.leibovici@cea.fr>
30 * Author: Henri Doreau <henri.doreau@cea.fr>
37 #include <sys/ioctl.h>
44 #include <sys/types.h>
46 #include <sys/syscall.h>
50 #ifdef HAVE_LINUX_UNISTD_H
51 #include <linux/unistd.h>
56 #include <libcfs/libcfs.h>
57 #include <lnet/lnetctl.h>
58 #include <lustre/lustre_idl.h>
59 #include <lustre/lustreapi.h>
60 #include "lustreapi_internal.h"
62 #define OPEN_BY_FID_PATH dot_lustre_name"/fid"
64 /****** HSM Copytool API ********/
65 #define CT_PRIV_MAGIC 0xC0BE2001
66 struct hsm_copytool_private {
72 lustre_kernelcomm kuc;
76 #define CP_PRIV_MAGIC 0x19880429
77 struct hsm_copyaction_private {
80 const struct hsm_copytool_private *ct_priv;
85 #include <libcfs/libcfs.h>
87 enum ct_progress_type {
98 CT_ARCHIVE_START = HSMA_ARCHIVE,
99 CT_ARCHIVE_RUNNING = HSMA_ARCHIVE + CT_RUNNING,
100 CT_ARCHIVE_FINISH = HSMA_ARCHIVE + CT_FINISH,
101 CT_ARCHIVE_CANCEL = HSMA_ARCHIVE + CT_CANCEL,
102 CT_ARCHIVE_ERROR = HSMA_ARCHIVE + CT_ERROR,
103 CT_RESTORE_START = HSMA_RESTORE,
104 CT_RESTORE_RUNNING = HSMA_RESTORE + CT_RUNNING,
105 CT_RESTORE_FINISH = HSMA_RESTORE + CT_FINISH,
106 CT_RESTORE_CANCEL = HSMA_RESTORE + CT_CANCEL,
107 CT_RESTORE_ERROR = HSMA_RESTORE + CT_ERROR,
108 CT_REMOVE_START = HSMA_REMOVE,
109 CT_REMOVE_RUNNING = HSMA_REMOVE + CT_RUNNING,
110 CT_REMOVE_FINISH = HSMA_REMOVE + CT_FINISH,
111 CT_REMOVE_CANCEL = HSMA_REMOVE + CT_CANCEL,
112 CT_REMOVE_ERROR = HSMA_REMOVE + CT_ERROR,
116 /* initialized in llapi_hsm_register_event_fifo() */
117 static int llapi_hsm_event_fd = -1;
118 static bool created_hsm_event_fifo;
120 static inline const char *llapi_hsm_ct_ev2str(int type)
127 case CT_ARCHIVE_START:
128 return "ARCHIVE_START";
129 case CT_ARCHIVE_RUNNING:
130 return "ARCHIVE_RUNNING";
131 case CT_ARCHIVE_FINISH:
132 return "ARCHIVE_FINISH";
133 case CT_ARCHIVE_CANCEL:
134 return "ARCHIVE_CANCEL";
135 case CT_ARCHIVE_ERROR:
136 return "ARCHIVE_ERROR";
137 case CT_RESTORE_START:
138 return "RESTORE_START";
139 case CT_RESTORE_RUNNING:
140 return "RESTORE_RUNNING";
141 case CT_RESTORE_FINISH:
142 return "RESTORE_FINISH";
143 case CT_RESTORE_CANCEL:
144 return "RESTORE_CANCEL";
145 case CT_RESTORE_ERROR:
146 return "RESTORE_ERROR";
147 case CT_REMOVE_START:
148 return "REMOVE_START";
149 case CT_REMOVE_RUNNING:
150 return "REMOVE_RUNNING";
151 case CT_REMOVE_FINISH:
152 return "REMOVE_FINISH";
153 case CT_REMOVE_CANCEL:
154 return "REMOVE_CANCEL";
155 case CT_REMOVE_ERROR:
156 return "REMOVE_ERROR";
158 llapi_err_noerrno(LLAPI_MSG_ERROR,
159 "Unknown event type: %d", type);
165 * Writes a JSON event to the monitor FIFO. Noop if no FIFO has been
168 * \param event A list of llapi_json_items comprising a
169 * single JSON-formatted event.
171 * \retval 0 on success.
172 * \retval -errno on error.
174 static int llapi_hsm_write_json_event(struct llapi_json_item_list **event)
177 char time_string[40];
178 char json_buf[PIPE_BUF];
180 time_t event_time = time(0);
181 struct tm time_components;
182 struct llapi_json_item_list *json_items;
184 /* Noop unless the event fd was initialized */
185 if (llapi_hsm_event_fd < 0)
188 if (event == NULL || *event == NULL)
193 localtime_r(&event_time, &time_components);
195 if (strftime(time_string, sizeof(time_string), "%Y-%m-%d %T %z",
196 &time_components) == 0) {
198 llapi_error(LLAPI_MSG_ERROR, rc, "strftime() failed");
202 rc = llapi_json_add_item(&json_items, "event_time", LLAPI_JSON_STRING,
205 llapi_error(LLAPI_MSG_ERROR, -rc, "error in "
206 "llapi_json_add_item()");
210 buf_file = fmemopen(json_buf, sizeof(json_buf), "w");
211 if (buf_file == NULL)
214 rc = llapi_json_write_list(event, buf_file);
222 if (write(llapi_hsm_event_fd, json_buf, strlen(json_buf)) < 0) {
223 /* Ignore write failures due to missing reader. */
232 * Hook for llapi_hsm_copytool_register and llapi_hsm_copytool_unregister
233 * to generate JSON events suitable for consumption by a copytool
234 * monitoring process.
236 * \param priv Opaque private control structure.
237 * \param event_type The type of event (register or unregister).
239 * \retval 0 on success.
240 * \retval -errno on error.
242 static int llapi_hsm_log_ct_registration(struct hsm_copytool_private **priv,
246 char agent_uuid[UUID_MAX];
247 struct hsm_copytool_private *ct;
248 struct llapi_json_item_list *json_items;
250 /* Noop unless the event fd was initialized */
251 if (llapi_hsm_event_fd < 0)
254 if (priv == NULL || *priv == NULL)
258 if (ct->magic != CT_PRIV_MAGIC)
261 if (event_type != CT_REGISTER && event_type != CT_UNREGISTER)
264 rc = llapi_json_init_list(&json_items);
268 rc = llapi_get_agent_uuid(ct->mnt, agent_uuid, sizeof(agent_uuid));
271 llapi_chomp_string(agent_uuid);
273 rc = llapi_json_add_item(&json_items, "uuid", LLAPI_JSON_STRING,
278 rc = llapi_json_add_item(&json_items, "mount_point", LLAPI_JSON_STRING,
283 rc = llapi_json_add_item(&json_items, "archive", LLAPI_JSON_INTEGER,
288 rc = llapi_json_add_item(&json_items, "event_type", LLAPI_JSON_STRING,
289 (char *)llapi_hsm_ct_ev2str(event_type));
293 rc = llapi_hsm_write_json_event(&json_items);
300 llapi_error(LLAPI_MSG_ERROR, rc, "error in "
301 "llapi_hsm_log_ct_registration()");
304 if (json_items != NULL)
305 llapi_json_destroy_list(&json_items);
311 * Given a copytool progress update, construct a JSON event suitable for
312 * consumption by a copytool monitoring process.
314 * Examples of various events generated here and written by
315 * llapi_hsm_write_json_event:
317 * Copytool registration and deregistration:
318 * {"event_time": "2014-02-26 14:58:01 -0500", "event_type": "REGISTER",
319 * "archive": 0, "mount_point": "/mnt/lustre",
320 * "uuid": "80379a60-1f8a-743f-daf2-307cde793ec2"}
321 * {"event_time": "2014-02-26 14:58:01 -0500", "event_type": "UNREGISTER",
322 * "archive": 0, "mount_point": "/mnt/lustre",
323 * "uuid": "80379a60-1f8a-743f-daf2-307cde793ec2"}
325 * An archive action, start to completion:
326 * {"event_time": "2014-02-26 14:50:13 -0500", "event_type": "ARCHIVE_START",
327 * "total_bytes": 0, "lustre_path": "d71.sanity-hsm/f71.sanity-hsm",
328 * "source_fid": "0x2000013a1:0x2:0x0", "data_fid": "0x2000013a1:0x2:0x0"}
329 * {"event_time": "2014-02-26 14:50:18 -0500", "event_type": "ARCHIVE_RUNNING",
330 * "current_bytes": 5242880, "total_bytes": 39000000,
331 * "lustre_path": "d71.sanity-hsm/f71.sanity-hsm",
332 * "source_fid": "0x2000013a1:0x2:0x0", "data_fid": "0x2000013a1:0x2:0x0"}
333 * {"event_time": "2014-02-26 14:50:50 -0500", "event_type": "ARCHIVE_FINISH",
334 * "source_fid": "0x2000013a1:0x2:0x0", "data_fid": "0x2000013a1:0x2:0x0"}
337 * {"event_time": "2014-02-26 14:50:13 -0500", "event_type": "LOGGED_MESSAGE",
339 * "message": "lhsmtool_posix[42]: copytool fs=lustre archive#=2 item_count=1"}
341 * \param hcp Opaque action handle returned by
342 * llapi_hsm_action_start.
343 * \param hai The hsm_action_item describing the request.
344 * \param progress_type The ct_progress_type describing the update.
345 * \param total The total expected bytes for the request.
346 * \param current The current copied byte count for the request.
348 * \retval 0 on success.
349 * \retval -errno on error.
351 static int llapi_hsm_log_ct_progress(struct hsm_copyaction_private **phcp,
352 const struct hsm_action_item *hai,
354 __u64 total, __u64 current)
358 long long recno = -1;
359 char lustre_path[PATH_MAX];
360 char strfid[FID_NOBRACE_LEN + 1];
361 struct hsm_copyaction_private *hcp;
362 struct llapi_json_item_list *json_items;
364 /* Noop unless the event fd was initialized */
365 if (llapi_hsm_event_fd < 0)
368 if (phcp == NULL || *phcp == NULL)
373 rc = llapi_json_init_list(&json_items);
377 snprintf(strfid, sizeof(strfid), DFID_NOBRACE, PFID(&hai->hai_dfid));
378 rc = llapi_json_add_item(&json_items, "data_fid",
379 LLAPI_JSON_STRING, strfid);
383 snprintf(strfid, sizeof(strfid), DFID_NOBRACE, PFID(&hai->hai_fid));
384 rc = llapi_json_add_item(&json_items, "source_fid",
385 LLAPI_JSON_STRING, strfid);
389 if (hcp->copy.hc_errval == ECANCELED) {
390 progress_type = CT_CANCEL;
394 if (hcp->copy.hc_errval != 0) {
395 progress_type = CT_ERROR;
397 rc = llapi_json_add_item(&json_items, "errno",
399 &hcp->copy.hc_errval);
403 rc = llapi_json_add_item(&json_items, "error",
405 strerror(hcp->copy.hc_errval));
412 /* lustre_path isn't available after a restore completes */
413 /* total_bytes isn't available after a restore or archive completes */
414 if (progress_type != CT_FINISH) {
415 rc = llapi_fid2path(hcp->ct_priv->mnt, strfid, lustre_path,
416 sizeof(lustre_path), &recno, &linkno);
420 rc = llapi_json_add_item(&json_items, "lustre_path",
421 LLAPI_JSON_STRING, lustre_path);
425 rc = llapi_json_add_item(&json_items, "total_bytes",
426 LLAPI_JSON_BIGNUM, &total);
431 if (progress_type == CT_RUNNING)
432 rc = llapi_json_add_item(&json_items, "current_bytes",
433 LLAPI_JSON_BIGNUM, ¤t);
438 rc = llapi_json_add_item(&json_items, "event_type", LLAPI_JSON_STRING,
439 (char *)llapi_hsm_ct_ev2str(hai->hai_action +
444 rc = llapi_hsm_write_json_event(&json_items);
451 llapi_error(LLAPI_MSG_ERROR, rc, "error in "
452 "llapi_hsm_log_ct_progress()");
455 if (json_items != NULL)
456 llapi_json_destroy_list(&json_items);
462 * Given a path to a FIFO, create a filehandle for nonblocking writes to it.
463 * Intended to be used for copytool monitoring processes that read an
464 * event stream from the FIFO. Events written in the absence of a reader
467 * \param path Path to monitor FIFO.
469 * \retval 0 on success.
470 * \retval -errno on error.
472 int llapi_hsm_register_event_fifo(const char *path)
477 /* Create the FIFO if necessary. */
478 if ((mkfifo(path, 0644) < 0) && (errno != EEXIST)) {
479 llapi_error(LLAPI_MSG_ERROR, errno, "mkfifo(%s) failed", path);
482 if (errno == EEXIST) {
483 if (stat(path, &statbuf) < 0) {
484 llapi_error(LLAPI_MSG_ERROR, errno, "mkfifo(%s) failed",
488 if (!S_ISFIFO(statbuf.st_mode) ||
489 ((statbuf.st_mode & 0777) != 0644)) {
490 llapi_error(LLAPI_MSG_ERROR, errno, "%s exists but is "
491 "not a pipe or has a wrong mode", path);
495 created_hsm_event_fifo = true;
498 /* Open the FIFO for read so that the subsequent open for write
499 * doesn't immediately fail. */
500 read_fd = open(path, O_RDONLY | O_NONBLOCK);
502 llapi_error(LLAPI_MSG_ERROR, errno,
503 "cannot open(%s) for read", path);
507 /* Open the FIFO for writes, but don't block on waiting
509 llapi_hsm_event_fd = open(path, O_WRONLY | O_NONBLOCK);
510 if (llapi_hsm_event_fd < 0) {
511 llapi_error(LLAPI_MSG_ERROR, errno,
512 "cannot open(%s) for write", path);
516 /* Now close the reader. An external monitoring process can
517 * now open the FIFO for reads. If no reader comes along the
518 * events are lost. NOTE: Only one reader at a time! */
521 /* Ignore SIGPIPEs -- can occur if the reader goes away. */
522 signal(SIGPIPE, SIG_IGN);
528 * Given a path to a FIFO, close its filehandle and delete the FIFO.
530 * \param path Path to monitor FIFO.
532 * \retval 0 on success.
533 * \retval -errno on error.
535 int llapi_hsm_unregister_event_fifo(const char *path)
537 /* Noop unless the event fd was initialized */
538 if (llapi_hsm_event_fd < 0)
541 if (close(llapi_hsm_event_fd) < 0)
544 if (created_hsm_event_fifo) {
546 created_hsm_event_fifo = false;
549 llapi_hsm_event_fd = -1;
555 * Custom logging callback to be used when a monitoring FIFO has been
556 * registered. Formats log entries as JSON events suitable for
557 * consumption by a copytool monitoring process.
559 * \param level The message loglevel.
560 * \param _rc The returncode associated with the message.
561 * \param fmt The message format string.
562 * \param args Arguments to be formatted by the format string.
566 void llapi_hsm_log_error(enum llapi_message_level level, int _rc,
567 const char *fmt, va_list args)
574 struct llapi_json_item_list *json_items;
576 /* Noop unless the event fd was initialized */
577 if (llapi_hsm_event_fd < 0)
580 rc = llapi_json_init_list(&json_items);
584 if ((level & LLAPI_MSG_NO_ERRNO) == 0) {
585 rc = llapi_json_add_item(&json_items, "errno",
591 rc = llapi_json_add_item(&json_items, "error",
598 va_copy(args2, args);
599 msg_len = vsnprintf(NULL, 0, fmt, args2) + 1;
602 msg = (char *) alloca(msg_len);
608 rc = vsnprintf(msg, msg_len, fmt, args);
612 rc = llapi_json_add_item(&json_items, "message",
618 rc = llapi_json_add_item(&json_items, "message",
620 "INTERNAL ERROR: message failed");
625 real_level = level & LLAPI_MSG_NO_ERRNO;
626 real_level = real_level > 0 ? level - LLAPI_MSG_NO_ERRNO : level;
628 rc = llapi_json_add_item(&json_items, "level", LLAPI_JSON_STRING,
629 (void *)llapi_msg_level2str(real_level));
633 rc = llapi_json_add_item(&json_items, "event_type", LLAPI_JSON_STRING,
638 rc = llapi_hsm_write_json_event(&json_items);
645 /* Write directly to stderr to avoid llapi_error, which now
646 * emits JSON event messages. */
647 fprintf(stderr, "\nFATAL ERROR IN llapi_hsm_log_error(): rc %d,", rc);
650 if (json_items != NULL)
651 llapi_json_destroy_list(&json_items);
656 /** Register a copytool
657 * \param[out] priv Opaque private control structure
658 * \param mnt Lustre filesystem mount point
659 * \param archive_count
660 * \param archives Which archive numbers this copytool is responsible for
661 * \param rfd_flags flags applied to read fd of pipe (e.g. O_NONBLOCK)
663 int llapi_hsm_copytool_register(struct hsm_copytool_private **priv,
664 const char *mnt, int archive_count,
665 int *archives, int rfd_flags)
667 struct hsm_copytool_private *ct;
670 if (archive_count > 0 && archives == NULL) {
671 llapi_err_noerrno(LLAPI_MSG_ERROR,
672 "NULL archive numbers");
676 ct = calloc(1, sizeof(*ct));
680 ct->magic = CT_PRIV_MAGIC;
682 ct->open_by_fid_fd = -1;
683 ct->kuc.lk_rfd = LK_NOFD;
684 ct->kuc.lk_wfd = LK_NOFD;
686 ct->mnt = strdup(mnt);
687 if (ct->mnt == NULL) {
692 ct->kuch = malloc(HAL_MAXSIZE + sizeof(*ct->kuch));
693 if (ct->kuch == NULL) {
698 ct->mnt_fd = open(ct->mnt, O_RDONLY);
699 if (ct->mnt_fd < 0) {
704 ct->open_by_fid_fd = openat(ct->mnt_fd, OPEN_BY_FID_PATH, O_RDONLY);
705 if (ct->open_by_fid_fd < 0) {
710 /* no archives specified means "match all". */
712 for (rc = 0; rc < archive_count; rc++) {
713 if (archives[rc] > 8 * sizeof(ct->archives)) {
714 llapi_err_noerrno(LLAPI_MSG_ERROR,
715 "maximum of %zu archives supported",
716 8 * sizeof(ct->archives));
719 /* in the list we have a all archive wildcard
720 * so move to all archives mode
722 if (archives[rc] == 0) {
727 ct->archives |= (1 << (archives[rc] - 1));
730 rc = libcfs_ukuc_start(&ct->kuc, KUC_GRP_HSM, rfd_flags);
734 /* Storing archive(s) in lk_data; see mdc_ioc_hsm_ct_start */
735 ct->kuc.lk_data = ct->archives;
736 rc = ioctl(ct->mnt_fd, LL_IOC_HSM_CT_START, &ct->kuc);
739 llapi_error(LLAPI_MSG_ERROR, rc,
740 "cannot start copytool on '%s'", mnt);
746 llapi_hsm_log_ct_registration(&ct, CT_REGISTER);
748 /* Only the kernel reference keeps the write side open */
749 close(ct->kuc.lk_wfd);
750 ct->kuc.lk_wfd = LK_NOFD;
758 /* cleanup the kuc channel */
759 libcfs_ukuc_stop(&ct->kuc);
762 if (!(ct->mnt_fd < 0))
765 if (!(ct->open_by_fid_fd < 0))
766 close(ct->open_by_fid_fd);
777 /** Deregister a copytool
778 * Note: under Linux, until llapi_hsm_copytool_unregister is called
779 * (or the program is killed), the libcfs module will be referenced
780 * and unremovable, even after Lustre services stop.
782 int llapi_hsm_copytool_unregister(struct hsm_copytool_private **priv)
784 struct hsm_copytool_private *ct;
786 if (priv == NULL || *priv == NULL)
790 if (ct->magic != CT_PRIV_MAGIC)
793 /* Tell the kernel to stop sending us messages */
794 ct->kuc.lk_flags = LK_FLG_STOP;
795 ioctl(ct->mnt_fd, LL_IOC_HSM_CT_START, &ct->kuc);
797 /* Shut down the kernelcomms */
798 libcfs_ukuc_stop(&ct->kuc);
800 llapi_hsm_log_ct_registration(&ct, CT_UNREGISTER);
802 close(ct->open_by_fid_fd);
812 /** Returns a file descriptor to poll/select on.
813 * \param ct Opaque private control structure
814 * \retval -EINVAL on error
815 * \retval the file descriptor for reading HSM events from the kernel
817 int llapi_hsm_copytool_get_fd(struct hsm_copytool_private *ct)
819 if (ct == NULL || ct->magic != CT_PRIV_MAGIC)
822 return libcfs_ukuc_get_rfd(&ct->kuc);
825 /** Wait for the next hsm_action_list
826 * \param ct Opaque private control structure
827 * \param halh Action list handle, will be allocated here
828 * \param msgsize Number of bytes in the message, will be set here
829 * \return 0 valid message received; halh and msgsize are set
831 * Note: The application must not call llapi_hsm_copytool_recv until it has
832 * cleared the data in ct->kuch from the previous call.
834 int llapi_hsm_copytool_recv(struct hsm_copytool_private *ct,
835 struct hsm_action_list **halh, int *msgsize)
837 struct kuc_hdr *kuch;
838 struct hsm_action_list *hal;
841 if (ct == NULL || ct->magic != CT_PRIV_MAGIC)
844 if (halh == NULL || msgsize == NULL)
850 rc = libcfs_ukuc_msg_get(&ct->kuc, (char *)kuch,
851 HAL_MAXSIZE + sizeof(*kuch),
856 /* Handle generic messages */
857 if (kuch->kuc_transport == KUC_TRANSPORT_GENERIC &&
858 kuch->kuc_msgtype == KUC_MSG_SHUTDOWN) {
863 if (kuch->kuc_transport != KUC_TRANSPORT_HSM ||
864 kuch->kuc_msgtype != HMT_ACTION_LIST) {
865 llapi_err_noerrno(LLAPI_MSG_ERROR,
866 "Unknown HSM message type %d:%d\n",
867 kuch->kuc_transport, kuch->kuc_msgtype);
872 if (kuch->kuc_msglen < sizeof(*kuch) + sizeof(*hal)) {
873 llapi_err_noerrno(LLAPI_MSG_ERROR, "Short HSM message %d",
879 /* Our message is a hsm_action_list. Use pointer math to skip
880 * kuch_hdr and point directly to the message payload.
882 hal = (struct hsm_action_list *)(kuch + 1);
884 /* Check that we have registered for this archive #
885 * if 0 registered, we serve any archive */
887 ((1 << (hal->hal_archive_id - 1)) & ct->archives) == 0) {
888 llapi_err_noerrno(LLAPI_MSG_INFO,
889 "This copytool does not service archive #%d,"
890 " ignoring this request."
891 " Mask of served archive is 0x%.8X",
892 hal->hal_archive_id, ct->archives);
898 *msgsize = kuch->kuc_msglen - sizeof(*kuch);
907 /** Get parent path from mount point and fid.
909 * \param mnt Filesystem root path.
910 * \param fid Object FID.
911 * \param parent Destination buffer.
912 * \param parent_len Destination buffer size.
913 * \return 0 on success.
915 static int fid_parent(const char *mnt, const lustre_fid *fid, char *parent,
920 long long recno = -1;
922 char strfid[FID_NOBRACE_LEN + 1];
925 snprintf(strfid, sizeof(strfid), DFID_NOBRACE, PFID(fid));
927 rc = llapi_fid2path(mnt, strfid, file, sizeof(file),
932 /* fid2path returns a relative path */
933 rc = snprintf(parent, parent_len, "%s/%s", mnt, file);
934 if (rc >= parent_len)
935 return -ENAMETOOLONG;
937 /* remove file name */
938 ptr = strrchr(parent, '/');
939 if (ptr == NULL || ptr == parent) {
949 static int ct_open_by_fid(const struct hsm_copytool_private *ct,
950 const struct lu_fid *fid, int open_flags)
952 char fid_name[FID_NOBRACE_LEN + 1];
955 snprintf(fid_name, sizeof(fid_name), DFID_NOBRACE, PFID(fid));
957 fd = openat(ct->open_by_fid_fd, fid_name, open_flags);
958 return fd < 0 ? -errno : fd;
961 static int ct_stat_by_fid(const struct hsm_copytool_private *ct,
962 const struct lu_fid *fid,
965 char fid_name[FID_NOBRACE_LEN + 1];
968 snprintf(fid_name, sizeof(fid_name), DFID_NOBRACE, PFID(fid));
970 rc = fstatat(ct->open_by_fid_fd, fid_name, buf, 0);
971 return rc ? -errno : 0;
974 /** Create the destination volatile file for a restore operation.
976 * \param hcp Private copyaction handle.
977 * \param mdt_index MDT index where to create the volatile file.
978 * \param flags Volatile file creation flags.
979 * \return 0 on success.
981 static int create_restore_volatile(struct hsm_copyaction_private *hcp,
982 int mdt_index, int open_flags)
986 char parent[PATH_MAX + 1];
987 const char *mnt = hcp->ct_priv->mnt;
988 struct hsm_action_item *hai = &hcp->copy.hc_hai;
990 rc = fid_parent(mnt, &hai->hai_fid, parent, sizeof(parent));
992 /* fid_parent() failed, try to keep on going */
993 llapi_error(LLAPI_MSG_ERROR, rc,
994 "cannot get parent path to restore "DFID" "
995 "using '%s'", PFID(&hai->hai_fid), mnt);
996 snprintf(parent, sizeof(parent), "%s", mnt);
999 fd = llapi_create_volatile_idx(parent, mdt_index, open_flags);
1003 rc = fchown(fd, hcp->stat.st_uid, hcp->stat.st_gid);
1007 rc = llapi_fd2fid(fd, &hai->hai_dfid);
1022 /** Start processing an HSM action.
1023 * Should be called by copytools just before starting handling a request.
1024 * It could be skipped if copytool only want to directly report an error,
1025 * \see llapi_hsm_action_end().
1027 * \param hcp Opaque action handle to be passed to
1028 * llapi_hsm_action_progress and llapi_hsm_action_end.
1029 * \param ct Copytool handle acquired at registration.
1030 * \param hai The hsm_action_item describing the request.
1031 * \param restore_mdt_index On restore: MDT index where to create the volatile
1032 * file. Use -1 for default.
1033 * \param restore_open_flags On restore: volatile file creation mode. Use
1034 * O_LOV_DELAY_CREATE to manually set the LOVEA
1036 * \param is_error Whether this call is just to report an error.
1038 * \return 0 on success.
1040 int llapi_hsm_action_begin(struct hsm_copyaction_private **phcp,
1041 const struct hsm_copytool_private *ct,
1042 const struct hsm_action_item *hai,
1043 int restore_mdt_index, int restore_open_flags,
1046 struct hsm_copyaction_private *hcp;
1049 hcp = calloc(1, sizeof(*hcp));
1055 hcp->copy.hc_hai = *hai;
1056 hcp->copy.hc_hai.hai_len = sizeof(*hai);
1061 if (hai->hai_action == HSMA_RESTORE) {
1062 rc = ct_stat_by_fid(hcp->ct_priv, &hai->hai_fid, &hcp->stat);
1066 rc = create_restore_volatile(hcp, restore_mdt_index,
1067 restore_open_flags);
1072 rc = ioctl(ct->mnt_fd, LL_IOC_HSM_COPY_START, &hcp->copy);
1078 llapi_hsm_log_ct_progress(&hcp, hai, CT_START, 0, 0);
1081 hcp->magic = CP_PRIV_MAGIC;
1086 if (!(hcp->data_fd < 0))
1087 close(hcp->data_fd);
1094 /** Terminate an HSM action processing.
1095 * Should be called by copytools just having finished handling the request.
1096 * \param hdl[in,out] Handle returned by llapi_hsm_action_start.
1097 * \param he[in] The final range of copied data (for copy actions).
1098 * \param errval[in] The status code of the operation.
1099 * \param flags[in] The flags about the termination status (HP_FLAG_RETRY if
1100 * the error is retryable).
1102 * \return 0 on success.
1104 int llapi_hsm_action_end(struct hsm_copyaction_private **phcp,
1105 const struct hsm_extent *he, int hp_flags, int errval)
1107 struct hsm_copyaction_private *hcp;
1108 struct hsm_action_item *hai;
1111 if (phcp == NULL || *phcp == NULL || he == NULL)
1116 if (hcp->magic != CP_PRIV_MAGIC)
1119 hai = &hcp->copy.hc_hai;
1121 if (hai->hai_action == HSMA_RESTORE && errval == 0) {
1122 struct timeval tv[2];
1124 /* Set {a,m}time of volatile file to that of original. */
1125 tv[0].tv_sec = hcp->stat.st_atime;
1127 tv[1].tv_sec = hcp->stat.st_mtime;
1129 if (futimes(hcp->data_fd, tv) < 0) {
1134 rc = fsync(hcp->data_fd);
1142 /* In some cases, like restore, 2 FIDs are used.
1143 * Set the right FID to use here. */
1144 if (hai->hai_action == HSMA_ARCHIVE || hai->hai_action == HSMA_RESTORE)
1145 hai->hai_fid = hai->hai_dfid;
1147 /* Fill the last missing data that will be needed by
1148 * kernel to send a hsm_progress. */
1149 hcp->copy.hc_flags = hp_flags;
1150 hcp->copy.hc_errval = abs(errval);
1152 hcp->copy.hc_hai.hai_extent = *he;
1154 rc = ioctl(hcp->ct_priv->mnt_fd, LL_IOC_HSM_COPY_END, &hcp->copy);
1160 llapi_hsm_log_ct_progress(&hcp, hai, CT_FINISH, 0, 0);
1163 if (!(hcp->data_fd < 0))
1164 close(hcp->data_fd);
1172 /** Notify a progress in processing an HSM action.
1173 * \param hdl[in,out] handle returned by llapi_hsm_action_start.
1174 * \param he[in] the range of copied data (for copy actions).
1175 * \param total[in] the expected total of copied data (for copy actions).
1176 * \param hp_flags[in] HSM progress flags.
1177 * \return 0 on success.
1179 int llapi_hsm_action_progress(struct hsm_copyaction_private *hcp,
1180 const struct hsm_extent *he, __u64 total,
1184 struct hsm_progress hp;
1185 struct hsm_action_item *hai;
1187 if (hcp == NULL || he == NULL)
1190 if (hcp->magic != CP_PRIV_MAGIC)
1193 hai = &hcp->copy.hc_hai;
1195 memset(&hp, 0, sizeof(hp));
1197 hp.hp_cookie = hai->hai_cookie;
1198 hp.hp_flags = hp_flags;
1200 /* Progress is made on the data fid */
1201 hp.hp_fid = hai->hai_dfid;
1204 rc = ioctl(hcp->ct_priv->mnt_fd, LL_IOC_HSM_PROGRESS, &hp);
1208 llapi_hsm_log_ct_progress(&hcp, hai, CT_RUNNING, total, he->length);
1213 /** Get the fid of object to be used for copying data.
1214 * @return error code if the action is not a copy operation.
1216 int llapi_hsm_action_get_dfid(const struct hsm_copyaction_private *hcp,
1219 const struct hsm_action_item *hai = &hcp->copy.hc_hai;
1221 if (hcp->magic != CP_PRIV_MAGIC)
1224 if (hai->hai_action != HSMA_RESTORE && hai->hai_action != HSMA_ARCHIVE)
1227 *fid = hai->hai_dfid;
1233 * Get a file descriptor to be used for copying data. It's up to the
1234 * caller to close the FDs obtained from this function.
1236 * @retval a file descriptor on success.
1237 * @retval a negative error code on failure.
1239 int llapi_hsm_action_get_fd(const struct hsm_copyaction_private *hcp)
1241 const struct hsm_action_item *hai = &hcp->copy.hc_hai;
1244 if (hcp->magic != CP_PRIV_MAGIC)
1247 if (hai->hai_action == HSMA_ARCHIVE) {
1248 return ct_open_by_fid(hcp->ct_priv, &hai->hai_dfid,
1249 O_RDONLY | O_NOATIME | O_NOFOLLOW | O_NONBLOCK);
1250 } else if (hai->hai_action == HSMA_RESTORE) {
1251 fd = dup(hcp->data_fd);
1252 return fd < 0 ? -errno : fd;
1259 * Import an existing hsm-archived file into Lustre.
1261 * Caller must access file by (returned) newfid value from now on.
1263 * \param dst path to Lustre destination (e.g. /mnt/lustre/my/file).
1264 * \param archive archive number.
1265 * \param st struct stat buffer containing file ownership, perm, etc.
1266 * \param stripe_* Striping options. Currently ignored, since the restore
1267 * operation will set the striping. In V2, this striping might
1269 * \param newfid[out] Filled with new Lustre fid.
1271 int llapi_hsm_import(const char *dst, int archive, const struct stat *st,
1272 unsigned long long stripe_size, int stripe_offset,
1273 int stripe_count, int stripe_pattern, char *pool_name,
1276 struct hsm_user_import hui;
1280 if (stripe_pattern == 0)
1281 stripe_pattern = LOV_PATTERN_RAID0;
1283 /* Create a non-striped file */
1284 fd = llapi_file_open_pool(dst, O_CREAT | O_WRONLY, st->st_mode,
1285 stripe_size, stripe_offset, stripe_count,
1286 stripe_pattern | LOV_PATTERN_F_RELEASED,
1289 llapi_error(LLAPI_MSG_ERROR, fd,
1290 "cannot create '%s' for import", dst);
1294 /* Get the new fid in Lustre. Caller needs to use this fid
1296 rc = llapi_fd2fid(fd, newfid);
1298 llapi_error(LLAPI_MSG_ERROR, rc,
1299 "cannot get fid of '%s' for import", dst);
1303 hui.hui_uid = st->st_uid;
1304 hui.hui_gid = st->st_gid;
1305 hui.hui_mode = st->st_mode;
1306 hui.hui_size = st->st_size;
1307 hui.hui_archive_id = archive;
1308 hui.hui_atime = st->st_atime;
1309 hui.hui_atime_ns = st->st_atim.tv_nsec;
1310 hui.hui_mtime = st->st_mtime;
1311 hui.hui_mtime_ns = st->st_mtim.tv_nsec;
1312 rc = ioctl(fd, LL_IOC_HSM_IMPORT, &hui);
1315 llapi_error(LLAPI_MSG_ERROR, rc, "cannot import '%s'", dst);
1328 * Return the current HSM states and HSM requests related to file pointed by \a
1331 * \param hus Should be allocated by caller. Will be filled with current file
1334 * \retval 0 on success.
1335 * \retval -errno on error.
1337 int llapi_hsm_state_get_fd(int fd, struct hsm_user_state *hus)
1341 rc = ioctl(fd, LL_IOC_HSM_STATE_GET, hus);
1342 /* If error, save errno value */
1343 rc = rc ? -errno : 0;
1349 * Return the current HSM states and HSM requests related to file pointed by \a
1352 * see llapi_hsm_state_get_fd() for args use and return
1354 int llapi_hsm_state_get(const char *path, struct hsm_user_state *hus)
1359 fd = open(path, O_RDONLY | O_NONBLOCK);
1363 rc = llapi_hsm_state_get_fd(fd, hus);
1370 * Set HSM states of file pointed by \a fd
1372 * Using the provided bitmasks, the current HSM states for this file will be
1373 * changed. \a archive_id could be used to change the archive number also. Set
1374 * it to 0 if you do not want to change it.
1376 * \param setmask Bitmask for flag to be set.
1377 * \param clearmask Bitmask for flag to be cleared.
1378 * \param archive_id Archive number identifier to use. 0 means no change.
1380 * \retval 0 on success.
1381 * \retval -errno on error.
1383 int llapi_hsm_state_set_fd(int fd, __u64 setmask, __u64 clearmask,
1386 struct hsm_state_set hss;
1389 hss.hss_valid = HSS_SETMASK|HSS_CLEARMASK;
1390 hss.hss_setmask = setmask;
1391 hss.hss_clearmask = clearmask;
1392 /* Change archive_id if provided. We can only change
1393 * to set something different than 0. */
1394 if (archive_id > 0) {
1395 hss.hss_valid |= HSS_ARCHIVE_ID;
1396 hss.hss_archive_id = archive_id;
1398 rc = ioctl(fd, LL_IOC_HSM_STATE_SET, &hss);
1399 /* If error, save errno value */
1400 rc = rc ? -errno : 0;
1406 * Set HSM states of file pointed by \a path.
1408 * see llapi_hsm_state_set_fd() for args use and return
1410 int llapi_hsm_state_set(const char *path, __u64 setmask, __u64 clearmask,
1416 fd = open(path, O_WRONLY | O_LOV_DELAY_CREATE | O_NONBLOCK);
1420 rc = llapi_hsm_state_set_fd(fd, setmask, clearmask, archive_id);
1427 * Return the current HSM request related to file pointed by \a path.
1429 * \param hca Should be allocated by caller. Will be filled with current file
1432 * \retval 0 on success.
1433 * \retval -errno on error.
1435 int llapi_hsm_current_action(const char *path, struct hsm_current_action *hca)
1440 fd = open(path, O_RDONLY | O_NONBLOCK);
1444 rc = ioctl(fd, LL_IOC_HSM_ACTION, hca);
1445 /* If error, save errno value */
1446 rc = rc ? -errno : 0;
1453 * Allocate a hsm_user_request with the specified carateristics.
1454 * This structure should be freed with free().
1456 * \return an allocated structure on success, NULL otherwise.
1458 struct hsm_user_request *llapi_hsm_user_request_alloc(int itemcount,
1463 len += sizeof(struct hsm_user_request);
1464 len += sizeof(struct hsm_user_item) * itemcount;
1467 return (struct hsm_user_request *)malloc(len);
1471 * Send a HSM request to Lustre, described in \param request.
1473 * \param path Fullpath to the file to operate on.
1474 * \param request The request, allocated with llapi_hsm_user_request_alloc().
1476 * \return 0 on success, an error code otherwise.
1478 int llapi_hsm_request(const char *path, const struct hsm_user_request *request)
1483 rc = get_root_path(WANT_FD, NULL, &fd, (char *)path, -1);
1487 rc = ioctl(fd, LL_IOC_HSM_REQUEST, request);
1488 /* If error, save errno value */
1489 rc = rc ? -errno : 0;