4 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License version 2 only,
8 * as published by the Free Software Foundation.
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License version 2 for more details (a copy is included
14 * in the LICENSE file that accompanied this code).
16 * You should have received a copy of the GNU General Public License
17 * version 2 along with this program; If not, see
18 * http://www.gnu.org/licenses/gpl-2.0.html
23 * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
24 * Use is subject to license terms.
26 * Copyright (c) 2012, 2014, Intel Corporation.
29 * This file is part of Lustre, http://www.lustre.org/
30 * Lustre is a trademark of Sun Microsystems, Inc.
32 * lnet/selftest/conctl.c
34 * Infrastructure of LST console
36 * Author: Liang Zhen <liangzhen@clusterfs.com>
39 #include <libcfs/libcfs.h>
40 #include <lnet/lib-lnet.h>
44 #define LST_NODE_STATE_COUNTER(nd, p) \
46 if ((nd)->nd_state == LST_NODE_ACTIVE) \
47 (p)->nle_nactive ++; \
48 else if ((nd)->nd_state == LST_NODE_BUSY) \
50 else if ((nd)->nd_state == LST_NODE_DOWN) \
53 (p)->nle_nunknown ++; \
57 struct lstcon_session console_session;
60 lstcon_node_get(struct lstcon_node *nd)
62 LASSERT (nd->nd_ref >= 1);
68 lstcon_node_find(struct lnet_process_id id, struct lstcon_node **ndpp,
71 struct lstcon_ndlink *ndl;
72 unsigned int idx = LNET_NIDADDR(id.nid) % LST_GLOBAL_HASHSIZE;
74 LASSERT(id.nid != LNET_NID_ANY);
76 list_for_each_entry(ndl, &console_session.ses_ndl_hash[idx],
78 if (ndl->ndl_node->nd_id.nid != id.nid ||
79 ndl->ndl_node->nd_id.pid != id.pid)
82 lstcon_node_get(ndl->ndl_node);
83 *ndpp = ndl->ndl_node;
90 LIBCFS_ALLOC(*ndpp, sizeof(**ndpp) + sizeof(*ndl));
94 ndl = (struct lstcon_ndlink *)(*ndpp + 1);
96 ndl->ndl_node = *ndpp;
98 ndl->ndl_node->nd_ref = 1;
99 ndl->ndl_node->nd_id = id;
100 ndl->ndl_node->nd_stamp = ktime_get();
101 ndl->ndl_node->nd_state = LST_NODE_UNKNOWN;
102 ndl->ndl_node->nd_timeout = 0;
103 memset(&ndl->ndl_node->nd_ping, 0, sizeof(ndl->ndl_node->nd_ping));
105 /* queued in global hash & list, no refcount is taken by
106 * global hash & list, if caller release his refcount,
107 * node will be released */
108 list_add_tail(&ndl->ndl_hlink, &console_session.ses_ndl_hash[idx]);
109 list_add_tail(&ndl->ndl_link, &console_session.ses_ndl_list);
115 lstcon_node_put(struct lstcon_node *nd)
117 struct lstcon_ndlink *ndl;
119 LASSERT(nd->nd_ref > 0);
121 if (--nd->nd_ref > 0)
124 ndl = (struct lstcon_ndlink *)(nd + 1);
126 LASSERT(!list_empty(&ndl->ndl_link));
127 LASSERT(!list_empty(&ndl->ndl_hlink));
129 /* remove from session */
130 list_del(&ndl->ndl_link);
131 list_del(&ndl->ndl_hlink);
133 LIBCFS_FREE(nd, sizeof(*nd) + sizeof(*ndl));
137 lstcon_ndlink_find(struct list_head *hash, struct lnet_process_id id,
138 struct lstcon_ndlink **ndlpp, int create)
140 unsigned int idx = LNET_NIDADDR(id.nid) % LST_NODE_HASHSIZE;
141 struct lstcon_ndlink *ndl;
142 struct lstcon_node *nd;
145 if (id.nid == LNET_NID_ANY)
149 list_for_each_entry(ndl, &hash[idx], ndl_hlink) {
150 if (ndl->ndl_node->nd_id.nid != id.nid ||
151 ndl->ndl_node->nd_id.pid != id.pid)
161 /* find or create in session hash */
162 rc = lstcon_node_find(id, &nd, (create == 1) ? 1 : 0);
166 LIBCFS_ALLOC(ndl, sizeof(*ndl));
175 INIT_LIST_HEAD(&ndl->ndl_link);
176 list_add_tail(&ndl->ndl_hlink, &hash[idx]);
182 lstcon_ndlink_release(struct lstcon_ndlink *ndl)
184 LASSERT(list_empty(&ndl->ndl_link));
185 LASSERT(!list_empty(&ndl->ndl_hlink));
187 list_del(&ndl->ndl_hlink); /* delete from hash */
188 lstcon_node_put(ndl->ndl_node);
190 LIBCFS_FREE(ndl, sizeof(*ndl));
194 lstcon_group_alloc(char *name, struct lstcon_group **grpp)
196 struct lstcon_group *grp;
199 LIBCFS_ALLOC(grp, offsetof(struct lstcon_group,
200 grp_ndl_hash[LST_NODE_HASHSIZE]));
206 if (strlen(name) > sizeof(grp->grp_name)-1) {
207 LIBCFS_FREE(grp, offsetof(struct lstcon_group,
208 grp_ndl_hash[LST_NODE_HASHSIZE]));
211 strncpy(grp->grp_name, name, sizeof(grp->grp_name));
214 INIT_LIST_HEAD(&grp->grp_link);
215 INIT_LIST_HEAD(&grp->grp_ndl_list);
216 INIT_LIST_HEAD(&grp->grp_trans_list);
218 for (i = 0; i < LST_NODE_HASHSIZE; i++)
219 INIT_LIST_HEAD(&grp->grp_ndl_hash[i]);
227 lstcon_group_addref(struct lstcon_group *grp)
232 static void lstcon_group_ndlink_release(struct lstcon_group *,
233 struct lstcon_ndlink *);
236 lstcon_group_drain(struct lstcon_group *grp, int keep)
238 struct lstcon_ndlink *ndl;
239 struct lstcon_ndlink *tmp;
241 list_for_each_entry_safe(ndl, tmp, &grp->grp_ndl_list, ndl_link) {
242 if ((ndl->ndl_node->nd_state & keep) == 0)
243 lstcon_group_ndlink_release(grp, ndl);
248 lstcon_group_decref(struct lstcon_group *grp)
252 if (--grp->grp_ref > 0)
255 if (!list_empty(&grp->grp_link))
256 list_del(&grp->grp_link);
258 lstcon_group_drain(grp, 0);
260 for (i = 0; i < LST_NODE_HASHSIZE; i++)
261 LASSERT(list_empty(&grp->grp_ndl_hash[i]));
263 LIBCFS_FREE(grp, offsetof(struct lstcon_group,
264 grp_ndl_hash[LST_NODE_HASHSIZE]));
268 lstcon_group_find(const char *name, struct lstcon_group **grpp)
270 struct lstcon_group *grp;
272 list_for_each_entry(grp, &console_session.ses_grp_list, grp_link) {
273 if (strncmp(grp->grp_name, name, LST_NAME_SIZE) != 0)
276 lstcon_group_addref(grp); /* +1 ref for caller */
285 lstcon_group_ndlink_find(struct lstcon_group *grp, struct lnet_process_id id,
286 struct lstcon_ndlink **ndlpp, int create)
290 rc = lstcon_ndlink_find(&grp->grp_ndl_hash[0], id, ndlpp, create);
294 if (!list_empty(&(*ndlpp)->ndl_link))
297 list_add_tail(&(*ndlpp)->ndl_link, &grp->grp_ndl_list);
304 lstcon_group_ndlink_release(struct lstcon_group *grp, struct lstcon_ndlink *ndl)
306 list_del_init(&ndl->ndl_link);
307 lstcon_ndlink_release(ndl);
312 lstcon_group_ndlink_move(struct lstcon_group *old,
313 struct lstcon_group *new, struct lstcon_ndlink *ndl)
315 unsigned int idx = LNET_NIDADDR(ndl->ndl_node->nd_id.nid) %
320 list_move_tail(&ndl->ndl_hlink, &new->grp_ndl_hash[idx]);
321 list_move_tail(&ndl->ndl_link, &new->grp_ndl_list);
326 lstcon_group_move(struct lstcon_group *old, struct lstcon_group *new)
328 struct lstcon_ndlink *ndl;
330 while (!list_empty(&old->grp_ndl_list)) {
331 ndl = list_entry(old->grp_ndl_list.next,
332 struct lstcon_ndlink, ndl_link);
333 lstcon_group_ndlink_move(old, new, ndl);
338 lstcon_sesrpc_condition(int transop, struct lstcon_node *nd, void *arg)
340 struct lstcon_group *grp = arg;
343 case LST_TRANS_SESNEW:
344 if (nd->nd_state == LST_NODE_ACTIVE)
348 case LST_TRANS_SESEND:
349 if (nd->nd_state != LST_NODE_ACTIVE)
352 if (grp != NULL && nd->nd_ref > 1)
356 case LST_TRANS_SESQRY:
367 lstcon_sesrpc_readent(int transop, struct srpc_msg *msg,
368 struct lstcon_rpc_ent __user *ent_up)
370 struct srpc_debug_reply *rep;
373 case LST_TRANS_SESNEW:
374 case LST_TRANS_SESEND:
377 case LST_TRANS_SESQRY:
378 rep = &msg->msg_body.dbg_reply;
380 if (copy_to_user(&ent_up->rpe_priv[0],
381 &rep->dbg_timeout, sizeof(int)) ||
382 copy_to_user(&ent_up->rpe_payload[0],
383 &rep->dbg_name, LST_NAME_SIZE))
396 lstcon_group_nodes_add(struct lstcon_group *grp,
397 int count, struct lnet_process_id __user *ids_up,
399 struct list_head __user *result_up)
401 struct lstcon_rpc_trans *trans;
402 struct lstcon_ndlink *ndl;
403 struct lstcon_group *tmp;
404 struct lnet_process_id id;
408 rc = lstcon_group_alloc(NULL, &tmp);
410 CERROR("Out of memory\n");
414 for (i = 0 ; i < count; i++) {
415 if (copy_from_user(&id, &ids_up[i], sizeof(id))) {
420 /* skip if it's in this group already */
421 rc = lstcon_group_ndlink_find(grp, id, &ndl, 0);
425 /* add to tmp group */
426 rc = lstcon_group_ndlink_find(tmp, id, &ndl, 1);
428 CERROR("Can't create ndlink, out of memory\n");
434 lstcon_group_decref(tmp);
438 rc = lstcon_rpc_trans_ndlist(&tmp->grp_ndl_list,
439 &tmp->grp_trans_list, LST_TRANS_SESNEW,
440 tmp, lstcon_sesrpc_condition, &trans);
442 CERROR("Can't create transaction: %d\n", rc);
443 lstcon_group_decref(tmp);
448 lstcon_rpc_trans_postwait(trans, LST_TRANS_TIMEOUT);
450 rc = lstcon_rpc_trans_interpreter(trans, result_up,
451 lstcon_sesrpc_readent);
452 *featp = trans->tas_features;
454 /* destroy all RPGs */
455 lstcon_rpc_trans_destroy(trans);
457 lstcon_group_move(tmp, grp);
458 lstcon_group_decref(tmp);
464 lstcon_group_nodes_remove(struct lstcon_group *grp,
465 int count, struct lnet_process_id __user *ids_up,
466 struct list_head __user *result_up)
468 struct lstcon_rpc_trans *trans;
469 struct lstcon_ndlink *ndl;
470 struct lstcon_group *tmp;
471 struct lnet_process_id id;
475 /* End session and remove node from the group */
477 rc = lstcon_group_alloc(NULL, &tmp);
479 CERROR("Out of memory\n");
483 for (i = 0; i < count; i++) {
484 if (copy_from_user(&id, &ids_up[i], sizeof(id))) {
489 /* move node to tmp group */
490 if (lstcon_group_ndlink_find(grp, id, &ndl, 0) == 0)
491 lstcon_group_ndlink_move(grp, tmp, ndl);
494 rc = lstcon_rpc_trans_ndlist(&tmp->grp_ndl_list,
495 &tmp->grp_trans_list, LST_TRANS_SESEND,
496 tmp, lstcon_sesrpc_condition, &trans);
498 CERROR("Can't create transaction: %d\n", rc);
502 lstcon_rpc_trans_postwait(trans, LST_TRANS_TIMEOUT);
504 rc = lstcon_rpc_trans_interpreter(trans, result_up, NULL);
506 lstcon_rpc_trans_destroy(trans);
507 /* release nodes anyway, because we can't rollback status */
508 lstcon_group_decref(tmp);
512 lstcon_group_move(tmp, grp);
513 lstcon_group_decref(tmp);
519 lstcon_group_add(char *name)
521 struct lstcon_group *grp;
524 rc = (lstcon_group_find(name, &grp) == 0)? -EEXIST: 0;
526 /* find a group with same name */
527 lstcon_group_decref(grp);
531 rc = lstcon_group_alloc(name, &grp);
533 CERROR("Can't allocate descriptor for group %s\n", name);
537 list_add_tail(&grp->grp_link, &console_session.ses_grp_list);
543 lstcon_nodes_add(char *name, int count, struct lnet_process_id __user *ids_up,
544 unsigned *featp, struct list_head __user *result_up)
546 struct lstcon_group *grp;
550 LASSERT (ids_up != NULL);
552 rc = lstcon_group_find(name, &grp);
554 CDEBUG(D_NET, "Can't find group %s\n", name);
558 if (grp->grp_ref > 2) {
559 /* referred by other threads or test */
560 CDEBUG(D_NET, "Group %s is busy\n", name);
561 lstcon_group_decref(grp);
566 rc = lstcon_group_nodes_add(grp, count, ids_up, featp, result_up);
568 lstcon_group_decref(grp);
574 lstcon_group_del(char *name)
576 struct lstcon_rpc_trans *trans;
577 struct lstcon_group *grp;
580 rc = lstcon_group_find(name, &grp);
582 CDEBUG(D_NET, "Can't find group: %s\n", name);
586 if (grp->grp_ref > 2) {
587 /* referred by others threads or test */
588 CDEBUG(D_NET, "Group %s is busy\n", name);
589 lstcon_group_decref(grp);
593 rc = lstcon_rpc_trans_ndlist(&grp->grp_ndl_list,
594 &grp->grp_trans_list, LST_TRANS_SESEND,
595 grp, lstcon_sesrpc_condition, &trans);
597 CERROR("Can't create transaction: %d\n", rc);
598 lstcon_group_decref(grp);
602 lstcon_rpc_trans_postwait(trans, LST_TRANS_TIMEOUT);
604 lstcon_rpc_trans_destroy(trans);
606 lstcon_group_decref(grp);
607 /* -ref for session, it's destroyed,
608 * status can't be rolled back, destroy group anway */
609 lstcon_group_decref(grp);
615 lstcon_group_clean(char *name, int args)
617 struct lstcon_group *grp = NULL;
620 rc = lstcon_group_find(name, &grp);
622 CDEBUG(D_NET, "Can't find group %s\n", name);
626 if (grp->grp_ref > 2) {
627 /* referred by test */
628 CDEBUG(D_NET, "Group %s is busy\n", name);
629 lstcon_group_decref(grp);
633 args = (LST_NODE_ACTIVE | LST_NODE_BUSY |
634 LST_NODE_DOWN | LST_NODE_UNKNOWN) & ~args;
636 lstcon_group_drain(grp, args);
638 lstcon_group_decref(grp);
639 /* release empty group */
640 if (list_empty(&grp->grp_ndl_list))
641 lstcon_group_decref(grp);
647 lstcon_nodes_remove(char *name, int count,
648 struct lnet_process_id __user *ids_up,
649 struct list_head __user *result_up)
651 struct lstcon_group *grp = NULL;
654 rc = lstcon_group_find(name, &grp);
656 CDEBUG(D_NET, "Can't find group: %s\n", name);
660 if (grp->grp_ref > 2) {
661 /* referred by test */
662 CDEBUG(D_NET, "Group %s is busy\n", name);
663 lstcon_group_decref(grp);
667 rc = lstcon_group_nodes_remove(grp, count, ids_up, result_up);
669 lstcon_group_decref(grp);
670 /* release empty group */
671 if (list_empty(&grp->grp_ndl_list))
672 lstcon_group_decref(grp);
678 lstcon_group_refresh(char *name, struct list_head __user *result_up)
680 struct lstcon_rpc_trans *trans;
681 struct lstcon_group *grp;
684 rc = lstcon_group_find(name, &grp);
686 CDEBUG(D_NET, "Can't find group: %s\n", name);
690 if (grp->grp_ref > 2) {
691 /* referred by test */
692 CDEBUG(D_NET, "Group %s is busy\n", name);
693 lstcon_group_decref(grp);
697 /* re-invite all inactive nodes int the group */
698 rc = lstcon_rpc_trans_ndlist(&grp->grp_ndl_list,
699 &grp->grp_trans_list, LST_TRANS_SESNEW,
700 grp, lstcon_sesrpc_condition, &trans);
702 /* local error, return */
703 CDEBUG(D_NET, "Can't create transaction: %d\n", rc);
704 lstcon_group_decref(grp);
708 lstcon_rpc_trans_postwait(trans, LST_TRANS_TIMEOUT);
710 rc = lstcon_rpc_trans_interpreter(trans, result_up, NULL);
712 lstcon_rpc_trans_destroy(trans);
714 lstcon_group_decref(grp);
720 lstcon_group_list(int index, int len, char __user *name_up)
722 struct lstcon_group *grp;
725 LASSERT(name_up != NULL);
727 list_for_each_entry(grp, &console_session.ses_grp_list, grp_link) {
729 return copy_to_user(name_up, grp->grp_name, len) ?
738 lstcon_nodes_getent(struct list_head *head, int *index_p,
739 int *count_p, struct lstcon_node_ent __user *dents_up)
741 struct lstcon_ndlink *ndl;
742 struct lstcon_node *nd;
746 LASSERT(index_p != NULL && count_p != NULL);
747 LASSERT(dents_up != NULL);
748 LASSERT(*index_p >= 0);
749 LASSERT(*count_p > 0);
751 list_for_each_entry(ndl, head, ndl_link) {
752 if (index++ < *index_p)
755 if (count >= *count_p)
759 if (copy_to_user(&dents_up[count].nde_id,
760 &nd->nd_id, sizeof(nd->nd_id)) ||
761 copy_to_user(&dents_up[count].nde_state,
762 &nd->nd_state, sizeof(nd->nd_state)))
768 if (index <= *index_p)
778 lstcon_group_info(char *name, struct lstcon_ndlist_ent __user *gents_p,
779 int *index_p, int *count_p,
780 struct lstcon_node_ent __user *dents_up)
782 struct lstcon_ndlist_ent *gentp;
783 struct lstcon_group *grp;
784 struct lstcon_ndlink *ndl;
787 rc = lstcon_group_find(name, &grp);
789 CDEBUG(D_NET, "Can't find group %s\n", name);
793 if (dents_up != NULL) {
795 rc = lstcon_nodes_getent(&grp->grp_ndl_list,
796 index_p, count_p, dents_up);
797 lstcon_group_decref(grp);
802 /* non-verbose query */
803 CFS_ALLOC_PTR(gentp);
805 CERROR("Can't allocate ndlist_ent\n");
806 lstcon_group_decref(grp);
811 list_for_each_entry(ndl, &grp->grp_ndl_list, ndl_link)
812 LST_NODE_STATE_COUNTER(ndl->ndl_node, gentp);
814 rc = copy_to_user(gents_p, gentp,
815 sizeof(struct lstcon_ndlist_ent)) ? -EFAULT : 0;
819 lstcon_group_decref(grp);
825 lstcon_batch_find(const char *name, struct lstcon_batch **batpp)
827 struct lstcon_batch *bat;
829 list_for_each_entry(bat, &console_session.ses_bat_list, bat_link) {
830 if (strncmp(bat->bat_name, name, LST_NAME_SIZE) == 0) {
840 lstcon_batch_add(char *name)
842 struct lstcon_batch *bat;
846 rc = (lstcon_batch_find(name, &bat) == 0)? -EEXIST: 0;
848 CDEBUG(D_NET, "Batch %s already exists\n", name);
852 LIBCFS_ALLOC(bat, sizeof(*bat));
854 CERROR("Can't allocate descriptor for batch %s\n", name);
858 LIBCFS_ALLOC(bat->bat_cli_hash,
859 sizeof(struct list_head) * LST_NODE_HASHSIZE);
860 if (bat->bat_cli_hash == NULL) {
861 CERROR("Can't allocate hash for batch %s\n", name);
862 LIBCFS_FREE(bat, sizeof(*bat));
867 LIBCFS_ALLOC(bat->bat_srv_hash,
868 sizeof(struct list_head) * LST_NODE_HASHSIZE);
869 if (bat->bat_srv_hash == NULL) {
870 CERROR("Can't allocate hash for batch %s\n", name);
871 LIBCFS_FREE(bat->bat_cli_hash, LST_NODE_HASHSIZE);
872 LIBCFS_FREE(bat, sizeof(*bat));
877 if (strlen(name) > sizeof(bat->bat_name)-1) {
878 LIBCFS_FREE(bat->bat_srv_hash, LST_NODE_HASHSIZE);
879 LIBCFS_FREE(bat->bat_cli_hash, LST_NODE_HASHSIZE);
880 LIBCFS_FREE(bat, sizeof(*bat));
883 strncpy(bat->bat_name, name, sizeof(bat->bat_name));
884 bat->bat_hdr.tsb_index = 0;
885 bat->bat_hdr.tsb_id.bat_id = ++console_session.ses_id_cookie;
888 bat->bat_state = LST_BATCH_IDLE;
890 INIT_LIST_HEAD(&bat->bat_cli_list);
891 INIT_LIST_HEAD(&bat->bat_srv_list);
892 INIT_LIST_HEAD(&bat->bat_test_list);
893 INIT_LIST_HEAD(&bat->bat_trans_list);
895 for (i = 0; i < LST_NODE_HASHSIZE; i++) {
896 INIT_LIST_HEAD(&bat->bat_cli_hash[i]);
897 INIT_LIST_HEAD(&bat->bat_srv_hash[i]);
900 list_add_tail(&bat->bat_link, &console_session.ses_bat_list);
906 lstcon_batch_list(int index, int len, char __user *name_up)
908 struct lstcon_batch *bat;
910 LASSERT(name_up != NULL);
913 list_for_each_entry(bat, &console_session.ses_bat_list, bat_link) {
915 return copy_to_user(name_up, bat->bat_name, len) ?
924 lstcon_batch_info(char *name, struct lstcon_test_batch_ent __user *ent_up,
925 int server, int testidx, int *index_p, int *ndent_p,
926 struct lstcon_node_ent __user *dents_up)
928 struct lstcon_test_batch_ent *entp;
929 struct list_head *clilst;
930 struct list_head *srvlst;
931 struct lstcon_test *test = NULL;
932 struct lstcon_batch *bat;
933 struct lstcon_ndlink *ndl;
936 rc = lstcon_batch_find(name, &bat);
938 CDEBUG(D_NET, "Can't find batch %s\n", name);
943 /* query test, test index start from 1 */
944 list_for_each_entry(test, &bat->bat_test_list, tes_link) {
950 CDEBUG(D_NET, "Can't find specified test in batch\n");
955 clilst = (test == NULL) ? &bat->bat_cli_list :
956 &test->tes_src_grp->grp_ndl_list;
957 srvlst = (test == NULL) ? &bat->bat_srv_list :
958 &test->tes_dst_grp->grp_ndl_list;
960 if (dents_up != NULL) {
961 rc = lstcon_nodes_getent((server ? srvlst: clilst),
962 index_p, ndent_p, dents_up);
966 /* non-verbose query */
972 entp->u.tbe_batch.bae_ntest = bat->bat_ntest;
973 entp->u.tbe_batch.bae_state = bat->bat_state;
977 entp->u.tbe_test.tse_type = test->tes_type;
978 entp->u.tbe_test.tse_loop = test->tes_loop;
979 entp->u.tbe_test.tse_concur = test->tes_concur;
982 list_for_each_entry(ndl, clilst, ndl_link)
983 LST_NODE_STATE_COUNTER(ndl->ndl_node, &entp->tbe_cli_nle);
985 list_for_each_entry(ndl, srvlst, ndl_link)
986 LST_NODE_STATE_COUNTER(ndl->ndl_node, &entp->tbe_srv_nle);
988 rc = copy_to_user(ent_up, entp,
989 sizeof(struct lstcon_test_batch_ent)) ? -EFAULT : 0;
997 lstcon_batrpc_condition(int transop, struct lstcon_node *nd, void *arg)
1000 case LST_TRANS_TSBRUN:
1001 if (nd->nd_state != LST_NODE_ACTIVE)
1005 case LST_TRANS_TSBSTOP:
1006 if (nd->nd_state != LST_NODE_ACTIVE)
1010 case LST_TRANS_TSBCLIQRY:
1011 case LST_TRANS_TSBSRVQRY:
1019 lstcon_batch_op(struct lstcon_batch *bat, int transop,
1020 struct list_head __user *result_up)
1022 struct lstcon_rpc_trans *trans;
1025 rc = lstcon_rpc_trans_ndlist(&bat->bat_cli_list,
1026 &bat->bat_trans_list, transop,
1027 bat, lstcon_batrpc_condition, &trans);
1029 CERROR("Can't create transaction: %d\n", rc);
1033 lstcon_rpc_trans_postwait(trans, LST_TRANS_TIMEOUT);
1035 rc = lstcon_rpc_trans_interpreter(trans, result_up, NULL);
1037 lstcon_rpc_trans_destroy(trans);
1043 lstcon_batch_run(char *name, int timeout, struct list_head __user *result_up)
1045 struct lstcon_batch *bat;
1048 if (lstcon_batch_find(name, &bat) != 0) {
1049 CDEBUG(D_NET, "Can't find batch %s\n", name);
1053 bat->bat_arg = timeout;
1055 rc = lstcon_batch_op(bat, LST_TRANS_TSBRUN, result_up);
1057 /* mark batch as running if it's started in any node */
1058 if (lstcon_tsbop_stat_success(lstcon_trans_stat(), 0) != 0)
1059 bat->bat_state = LST_BATCH_RUNNING;
1065 lstcon_batch_stop(char *name, int force, struct list_head __user *result_up)
1067 struct lstcon_batch *bat;
1070 if (lstcon_batch_find(name, &bat) != 0) {
1071 CDEBUG(D_NET, "Can't find batch %s\n", name);
1075 bat->bat_arg = force;
1077 rc = lstcon_batch_op(bat, LST_TRANS_TSBSTOP, result_up);
1079 /* mark batch as stopped if all RPCs finished */
1080 if (lstcon_tsbop_stat_failure(lstcon_trans_stat(), 0) == 0)
1081 bat->bat_state = LST_BATCH_IDLE;
1087 lstcon_batch_destroy(struct lstcon_batch *bat)
1089 struct lstcon_ndlink *ndl;
1090 struct lstcon_test *test;
1093 list_del(&bat->bat_link);
1095 while (!list_empty(&bat->bat_test_list)) {
1096 test = list_entry(bat->bat_test_list.next,
1097 struct lstcon_test, tes_link);
1098 LASSERT(list_empty(&test->tes_trans_list));
1100 list_del(&test->tes_link);
1102 lstcon_group_decref(test->tes_src_grp);
1103 lstcon_group_decref(test->tes_dst_grp);
1105 LIBCFS_FREE(test, offsetof(struct lstcon_test,
1106 tes_param[test->tes_paramlen]));
1109 LASSERT(list_empty(&bat->bat_trans_list));
1111 while (!list_empty(&bat->bat_cli_list)) {
1112 ndl = list_entry(bat->bat_cli_list.next,
1113 struct lstcon_ndlink, ndl_link);
1114 list_del_init(&ndl->ndl_link);
1116 lstcon_ndlink_release(ndl);
1119 while (!list_empty(&bat->bat_srv_list)) {
1120 ndl = list_entry(bat->bat_srv_list.next,
1121 struct lstcon_ndlink, ndl_link);
1122 list_del_init(&ndl->ndl_link);
1124 lstcon_ndlink_release(ndl);
1127 for (i = 0; i < LST_NODE_HASHSIZE; i++) {
1128 LASSERT(list_empty(&bat->bat_cli_hash[i]));
1129 LASSERT(list_empty(&bat->bat_srv_hash[i]));
1132 LIBCFS_FREE(bat->bat_cli_hash,
1133 sizeof(struct list_head) * LST_NODE_HASHSIZE);
1134 LIBCFS_FREE(bat->bat_srv_hash,
1135 sizeof(struct list_head) * LST_NODE_HASHSIZE);
1136 LIBCFS_FREE(bat, sizeof(*bat));
1140 lstcon_testrpc_condition(int transop, struct lstcon_node *nd, void *arg)
1142 struct lstcon_test *test = arg;
1143 struct lstcon_batch *batch;
1144 struct lstcon_ndlink *ndl;
1145 struct list_head *hash;
1146 struct list_head *head;
1148 LASSERT(test != NULL);
1150 batch = test->tes_batch;
1151 LASSERT(batch != NULL);
1153 if (test->tes_oneside &&
1154 transop == LST_TRANS_TSBSRVADD)
1157 if (nd->nd_state != LST_NODE_ACTIVE)
1160 if (transop == LST_TRANS_TSBCLIADD) {
1161 hash = batch->bat_cli_hash;
1162 head = &batch->bat_cli_list;
1165 LASSERT (transop == LST_TRANS_TSBSRVADD);
1167 hash = batch->bat_srv_hash;
1168 head = &batch->bat_srv_list;
1171 LASSERT (nd->nd_id.nid != LNET_NID_ANY);
1173 if (lstcon_ndlink_find(hash, nd->nd_id, &ndl, 1) != 0)
1176 if (list_empty(&ndl->ndl_link))
1177 list_add_tail(&ndl->ndl_link, head);
1183 lstcon_test_nodes_add(struct lstcon_test *test,
1184 struct list_head __user *result_up)
1186 struct lstcon_rpc_trans *trans;
1187 struct lstcon_group *grp;
1191 LASSERT (test->tes_src_grp != NULL);
1192 LASSERT (test->tes_dst_grp != NULL);
1194 transop = LST_TRANS_TSBSRVADD;
1195 grp = test->tes_dst_grp;
1197 rc = lstcon_rpc_trans_ndlist(&grp->grp_ndl_list,
1198 &test->tes_trans_list, transop,
1199 test, lstcon_testrpc_condition, &trans);
1201 CERROR("Can't create transaction: %d\n", rc);
1205 lstcon_rpc_trans_postwait(trans, LST_TRANS_TIMEOUT);
1207 if (lstcon_trans_stat()->trs_rpc_errno != 0 ||
1208 lstcon_trans_stat()->trs_fwk_errno != 0) {
1209 lstcon_rpc_trans_interpreter(trans, result_up, NULL);
1211 lstcon_rpc_trans_destroy(trans);
1212 /* return if any error */
1213 CDEBUG(D_NET, "Failed to add test %s, "
1214 "RPC error %d, framework error %d\n",
1215 transop == LST_TRANS_TSBCLIADD ? "client" : "server",
1216 lstcon_trans_stat()->trs_rpc_errno,
1217 lstcon_trans_stat()->trs_fwk_errno);
1222 lstcon_rpc_trans_destroy(trans);
1224 if (transop == LST_TRANS_TSBCLIADD)
1227 transop = LST_TRANS_TSBCLIADD;
1228 grp = test->tes_src_grp;
1229 test->tes_cliidx = 0;
1231 /* requests to test clients */
1236 lstcon_verify_batch(const char *name, struct lstcon_batch **batch)
1240 rc = lstcon_batch_find(name, batch);
1242 CDEBUG(D_NET, "Can't find batch %s\n", name);
1246 if ((*batch)->bat_state != LST_BATCH_IDLE) {
1247 CDEBUG(D_NET, "Can't change running batch %s\n", name);
1255 lstcon_verify_group(const char *name, struct lstcon_group **grp)
1258 struct lstcon_ndlink *ndl;
1260 rc = lstcon_group_find(name, grp);
1262 CDEBUG(D_NET, "can't find group %s\n", name);
1266 list_for_each_entry(ndl, &(*grp)->grp_ndl_list, ndl_link) {
1267 if (ndl->ndl_node->nd_state == LST_NODE_ACTIVE) {
1272 CDEBUG(D_NET, "Group %s has no ACTIVE nodes\n", name);
1278 lstcon_test_add(char *batch_name, int type, int loop,
1279 int concur, int dist, int span,
1280 char *src_name, char *dst_name,
1281 void *param, int paramlen, int *retp,
1282 struct list_head __user *result_up)
1284 struct lstcon_test *test = NULL;
1286 struct lstcon_group *src_grp = NULL;
1287 struct lstcon_group *dst_grp = NULL;
1288 struct lstcon_batch *batch = NULL;
1291 * verify that a batch of the given name exists, and the groups
1292 * that will be part of the batch exist and have at least one
1295 rc = lstcon_verify_batch(batch_name, &batch);
1299 rc = lstcon_verify_group(src_name, &src_grp);
1303 rc = lstcon_verify_group(dst_name, &dst_grp);
1307 if (dst_grp->grp_userland)
1310 LIBCFS_ALLOC(test, offsetof(struct lstcon_test, tes_param[paramlen]));
1312 CERROR("Can't allocate test descriptor\n");
1318 test->tes_hdr.tsb_id = batch->bat_hdr.tsb_id;
1319 test->tes_batch = batch;
1320 test->tes_type = type;
1321 test->tes_oneside = 0; /* TODO */
1322 test->tes_loop = loop;
1323 test->tes_concur = concur;
1324 test->tes_stop_onerr = 1; /* TODO */
1325 test->tes_span = span;
1326 test->tes_dist = dist;
1327 test->tes_cliidx = 0; /* just used for creating RPC */
1328 test->tes_src_grp = src_grp;
1329 test->tes_dst_grp = dst_grp;
1330 INIT_LIST_HEAD(&test->tes_trans_list);
1332 if (param != NULL) {
1333 test->tes_paramlen = paramlen;
1334 memcpy(&test->tes_param[0], param, paramlen);
1337 rc = lstcon_test_nodes_add(test, result_up);
1342 if (lstcon_trans_stat()->trs_rpc_errno != 0 ||
1343 lstcon_trans_stat()->trs_fwk_errno != 0)
1344 CDEBUG(D_NET, "Failed to add test %d to batch %s\n", type,
1347 /* add to test list anyway, so user can check what's going on */
1348 list_add_tail(&test->tes_link, &batch->bat_test_list);
1351 test->tes_hdr.tsb_index = batch->bat_ntest;
1353 /* hold groups so nobody can change them */
1357 LIBCFS_FREE(test, offsetof(struct lstcon_test,
1358 tes_param[paramlen]));
1360 if (dst_grp != NULL)
1361 lstcon_group_decref(dst_grp);
1363 if (src_grp != NULL)
1364 lstcon_group_decref(src_grp);
1370 lstcon_test_find(struct lstcon_batch *batch, int idx,
1371 struct lstcon_test **testpp)
1373 struct lstcon_test *test;
1375 list_for_each_entry(test, &batch->bat_test_list, tes_link) {
1376 if (idx == test->tes_hdr.tsb_index) {
1386 lstcon_tsbrpc_readent(int transop, struct srpc_msg *msg,
1387 struct lstcon_rpc_ent __user *ent_up)
1389 struct srpc_batch_reply *rep = &msg->msg_body.bat_reply;
1391 LASSERT (transop == LST_TRANS_TSBCLIQRY ||
1392 transop == LST_TRANS_TSBSRVQRY);
1394 /* positive errno, framework error code */
1395 if (copy_to_user(&ent_up->rpe_priv[0],
1396 &rep->bar_active, sizeof(rep->bar_active)))
1403 lstcon_test_batch_query(char *name, int testidx, int client,
1404 int timeout, struct list_head __user *result_up)
1406 struct lstcon_rpc_trans *trans;
1407 struct list_head *translist;
1408 struct list_head *ndlist;
1409 struct lstcon_tsb_hdr *hdr;
1410 struct lstcon_batch *batch;
1411 struct lstcon_test *test = NULL;
1415 rc = lstcon_batch_find(name, &batch);
1417 CDEBUG(D_NET, "Can't find batch: %s\n", name);
1422 translist = &batch->bat_trans_list;
1423 ndlist = &batch->bat_cli_list;
1424 hdr = &batch->bat_hdr;
1427 /* query specified test only */
1428 rc = lstcon_test_find(batch, testidx, &test);
1430 CDEBUG(D_NET, "Can't find test: %d\n", testidx);
1434 translist = &test->tes_trans_list;
1435 ndlist = &test->tes_src_grp->grp_ndl_list;
1436 hdr = &test->tes_hdr;
1439 transop = client ? LST_TRANS_TSBCLIQRY : LST_TRANS_TSBSRVQRY;
1441 rc = lstcon_rpc_trans_ndlist(ndlist, translist, transop, hdr,
1442 lstcon_batrpc_condition, &trans);
1444 CERROR("Can't create transaction: %d\n", rc);
1448 lstcon_rpc_trans_postwait(trans, timeout);
1450 if (testidx == 0 && /* query a batch, not a test */
1451 lstcon_rpc_stat_failure(lstcon_trans_stat(), 0) == 0 &&
1452 lstcon_tsbqry_stat_run(lstcon_trans_stat(), 0) == 0) {
1453 /* all RPCs finished, and no active test */
1454 batch->bat_state = LST_BATCH_IDLE;
1457 rc = lstcon_rpc_trans_interpreter(trans, result_up,
1458 lstcon_tsbrpc_readent);
1459 lstcon_rpc_trans_destroy(trans);
1465 lstcon_statrpc_readent(int transop, struct srpc_msg *msg,
1466 struct lstcon_rpc_ent __user *ent_up)
1468 struct srpc_stat_reply *rep = &msg->msg_body.stat_reply;
1469 struct sfw_counters __user *sfwk_stat;
1470 struct srpc_counters __user *srpc_stat;
1471 struct lnet_counters_common __user *lnet_stat;
1473 if (rep->str_status != 0)
1476 sfwk_stat = (struct sfw_counters __user *)&ent_up->rpe_payload[0];
1477 srpc_stat = (struct srpc_counters __user *)
1478 ((char __user *)sfwk_stat + sizeof(*sfwk_stat));
1479 lnet_stat = (struct lnet_counters_common __user *)
1480 ((char __user *)srpc_stat + sizeof(*srpc_stat));
1482 if (copy_to_user(sfwk_stat, &rep->str_fw, sizeof(*sfwk_stat)) ||
1483 copy_to_user(srpc_stat, &rep->str_rpc, sizeof(*srpc_stat)) ||
1484 copy_to_user(lnet_stat, &rep->str_lnet, sizeof(*lnet_stat)))
1491 lstcon_ndlist_stat(struct list_head *ndlist,
1492 int timeout, struct list_head __user *result_up)
1494 struct list_head head;
1495 struct lstcon_rpc_trans *trans;
1498 INIT_LIST_HEAD(&head);
1500 rc = lstcon_rpc_trans_ndlist(ndlist, &head,
1501 LST_TRANS_STATQRY, NULL, NULL, &trans);
1503 CERROR("Can't create transaction: %d\n", rc);
1507 lstcon_rpc_trans_postwait(trans, LST_VALIDATE_TIMEOUT(timeout));
1509 rc = lstcon_rpc_trans_interpreter(trans, result_up,
1510 lstcon_statrpc_readent);
1511 lstcon_rpc_trans_destroy(trans);
1517 lstcon_group_stat(char *grp_name, int timeout,
1518 struct list_head __user *result_up)
1520 struct lstcon_group *grp;
1523 rc = lstcon_group_find(grp_name, &grp);
1525 CDEBUG(D_NET, "Can't find group %s\n", grp_name);
1529 rc = lstcon_ndlist_stat(&grp->grp_ndl_list, timeout, result_up);
1531 lstcon_group_decref(grp);
1537 lstcon_nodes_stat(int count, struct lnet_process_id __user *ids_up,
1538 int timeout, struct list_head __user *result_up)
1540 struct lstcon_ndlink *ndl;
1541 struct lstcon_group *tmp;
1542 struct lnet_process_id id;
1546 rc = lstcon_group_alloc(NULL, &tmp);
1548 CERROR("Out of memory\n");
1552 for (i = 0 ; i < count; i++) {
1553 if (copy_from_user(&id, &ids_up[i], sizeof(id))) {
1558 /* add to tmp group */
1559 rc = lstcon_group_ndlink_find(tmp, id, &ndl, 2);
1561 CDEBUG((rc == -ENOMEM) ? D_ERROR : D_NET,
1562 "Failed to find or create %s: %d\n",
1563 libcfs_id2str(id), rc);
1569 lstcon_group_decref(tmp);
1573 rc = lstcon_ndlist_stat(&tmp->grp_ndl_list, timeout, result_up);
1575 lstcon_group_decref(tmp);
1581 lstcon_debug_ndlist(struct list_head *ndlist,
1582 struct list_head *translist,
1583 int timeout, struct list_head __user *result_up)
1585 struct lstcon_rpc_trans *trans;
1588 rc = lstcon_rpc_trans_ndlist(ndlist, translist, LST_TRANS_SESQRY,
1589 NULL, lstcon_sesrpc_condition, &trans);
1591 CERROR("Can't create transaction: %d\n", rc);
1595 lstcon_rpc_trans_postwait(trans, LST_VALIDATE_TIMEOUT(timeout));
1597 rc = lstcon_rpc_trans_interpreter(trans, result_up,
1598 lstcon_sesrpc_readent);
1599 lstcon_rpc_trans_destroy(trans);
1605 lstcon_session_debug(int timeout, struct list_head __user *result_up)
1607 return lstcon_debug_ndlist(&console_session.ses_ndl_list,
1608 NULL, timeout, result_up);
1612 lstcon_batch_debug(int timeout, char *name,
1613 int client, struct list_head __user *result_up)
1615 struct lstcon_batch *bat;
1618 rc = lstcon_batch_find(name, &bat);
1622 rc = lstcon_debug_ndlist(client ? &bat->bat_cli_list :
1624 NULL, timeout, result_up);
1630 lstcon_group_debug(int timeout, char *name,
1631 struct list_head __user *result_up)
1633 struct lstcon_group *grp;
1636 rc = lstcon_group_find(name, &grp);
1640 rc = lstcon_debug_ndlist(&grp->grp_ndl_list, NULL,
1641 timeout, result_up);
1642 lstcon_group_decref(grp);
1648 lstcon_nodes_debug(int timeout, int count,
1649 struct lnet_process_id __user *ids_up,
1650 struct list_head __user *result_up)
1652 struct lnet_process_id id;
1653 struct lstcon_ndlink *ndl;
1654 struct lstcon_group *grp;
1658 rc = lstcon_group_alloc(NULL, &grp);
1660 CDEBUG(D_NET, "Out of memory\n");
1664 for (i = 0; i < count; i++) {
1665 if (copy_from_user(&id, &ids_up[i], sizeof(id))) {
1670 /* node is added to tmp group */
1671 rc = lstcon_group_ndlink_find(grp, id, &ndl, 1);
1673 CERROR("Can't create node link\n");
1679 lstcon_group_decref(grp);
1683 rc = lstcon_debug_ndlist(&grp->grp_ndl_list, NULL,
1684 timeout, result_up);
1686 lstcon_group_decref(grp);
1692 lstcon_session_match(struct lst_sid sid)
1694 return (console_session.ses_id.ses_nid == sid.ses_nid &&
1695 console_session.ses_id.ses_stamp == sid.ses_stamp) ? 1: 0;
1699 lstcon_new_session_id(struct lst_sid *sid)
1701 struct lnet_process_id id;
1703 LASSERT(console_session.ses_state == LST_SESSION_NONE);
1706 sid->ses_nid = id.nid;
1707 sid->ses_stamp = div_u64(ktime_get_ns(), NSEC_PER_MSEC);
1711 lstcon_session_new(char *name, int key, unsigned feats,
1712 int timeout, int force, struct lst_sid __user *sid_up)
1717 if (console_session.ses_state != LST_SESSION_NONE) {
1718 /* session exists */
1720 CNETERR("Session %s already exists\n",
1721 console_session.ses_name);
1725 rc = lstcon_session_end();
1727 /* lstcon_session_end() only return local error */
1732 if ((feats & ~LST_FEATS_MASK) != 0) {
1733 CNETERR("Unknown session features %x\n",
1734 (feats & ~LST_FEATS_MASK));
1738 for (i = 0; i < LST_GLOBAL_HASHSIZE; i++)
1739 LASSERT(list_empty(&console_session.ses_ndl_hash[i]));
1741 lstcon_new_session_id(&console_session.ses_id);
1743 console_session.ses_key = key;
1744 console_session.ses_state = LST_SESSION_ACTIVE;
1745 console_session.ses_force = !!force;
1746 console_session.ses_features = feats;
1747 console_session.ses_feats_updated = 0;
1748 console_session.ses_timeout = (timeout <= 0) ?
1749 LST_CONSOLE_TIMEOUT : timeout;
1751 if (strlen(name) > sizeof(console_session.ses_name)-1)
1753 strlcpy(console_session.ses_name, name,
1754 sizeof(console_session.ses_name));
1756 rc = lstcon_batch_add(LST_DEFAULT_BATCH);
1760 rc = lstcon_rpc_pinger_start();
1762 struct lstcon_batch *bat = NULL;
1764 lstcon_batch_find(LST_DEFAULT_BATCH, &bat);
1765 lstcon_batch_destroy(bat);
1770 if (copy_to_user(sid_up, &console_session.ses_id,
1771 sizeof(struct lst_sid)) == 0)
1774 lstcon_session_end();
1780 lstcon_session_info(struct lst_sid __user *sid_up, int __user *key_up,
1781 unsigned __user *featp,
1782 struct lstcon_ndlist_ent __user *ndinfo_up,
1783 char __user *name_up, int len)
1785 struct lstcon_ndlist_ent *entp;
1786 struct lstcon_ndlink *ndl;
1789 if (console_session.ses_state != LST_SESSION_ACTIVE)
1792 LIBCFS_ALLOC(entp, sizeof(*entp));
1796 list_for_each_entry(ndl, &console_session.ses_ndl_list, ndl_link)
1797 LST_NODE_STATE_COUNTER(ndl->ndl_node, entp);
1799 if (copy_to_user(sid_up, &console_session.ses_id,
1800 sizeof(struct lst_sid)) ||
1801 copy_to_user(key_up, &console_session.ses_key,
1803 copy_to_user(featp, &console_session.ses_features,
1805 copy_to_user(ndinfo_up, entp, sizeof(*entp)) ||
1806 copy_to_user(name_up, console_session.ses_name, len))
1809 LIBCFS_FREE(entp, sizeof(*entp));
1815 lstcon_session_end(void)
1817 struct lstcon_rpc_trans *trans;
1818 struct lstcon_group *grp;
1819 struct lstcon_batch *bat;
1822 LASSERT (console_session.ses_state == LST_SESSION_ACTIVE);
1824 rc = lstcon_rpc_trans_ndlist(&console_session.ses_ndl_list,
1825 NULL, LST_TRANS_SESEND, NULL,
1826 lstcon_sesrpc_condition, &trans);
1828 CERROR("Can't create transaction: %d\n", rc);
1832 console_session.ses_shutdown = 1;
1834 lstcon_rpc_pinger_stop();
1836 lstcon_rpc_trans_postwait(trans, LST_TRANS_TIMEOUT);
1838 lstcon_rpc_trans_destroy(trans);
1839 /* User can do nothing even rpc failed, so go on */
1841 /* waiting for orphan rpcs to die */
1842 lstcon_rpc_cleanup_wait();
1844 console_session.ses_id = LST_INVALID_SID;
1845 console_session.ses_state = LST_SESSION_NONE;
1846 console_session.ses_key = 0;
1847 console_session.ses_force = 0;
1848 console_session.ses_feats_updated = 0;
1850 /* destroy all batches */
1851 while (!list_empty(&console_session.ses_bat_list)) {
1852 bat = list_entry(console_session.ses_bat_list.next,
1853 struct lstcon_batch, bat_link);
1855 lstcon_batch_destroy(bat);
1858 /* destroy all groups */
1859 while (!list_empty(&console_session.ses_grp_list)) {
1860 grp = list_entry(console_session.ses_grp_list.next,
1861 struct lstcon_group, grp_link);
1862 LASSERT(grp->grp_ref == 1);
1864 lstcon_group_decref(grp);
1867 /* all nodes should be released */
1868 LASSERT(list_empty(&console_session.ses_ndl_list));
1870 console_session.ses_shutdown = 0;
1871 console_session.ses_expired = 0;
1877 lstcon_session_feats_check(unsigned feats)
1881 if ((feats & ~LST_FEATS_MASK) != 0) {
1882 CERROR("Can't support these features: %x\n",
1883 (feats & ~LST_FEATS_MASK));
1887 spin_lock(&console_session.ses_rpc_lock);
1889 if (!console_session.ses_feats_updated) {
1890 console_session.ses_feats_updated = 1;
1891 console_session.ses_features = feats;
1894 if (console_session.ses_features != feats)
1897 spin_unlock(&console_session.ses_rpc_lock);
1900 CERROR("remote features %x do not match with "
1901 "session features %x of console\n",
1902 feats, console_session.ses_features);
1909 lstcon_acceptor_handle(struct srpc_server_rpc *rpc)
1911 struct srpc_msg *rep = &rpc->srpc_replymsg;
1912 struct srpc_msg *req = &rpc->srpc_reqstbuf->buf_msg;
1913 struct srpc_join_reqst *jreq = &req->msg_body.join_reqst;
1914 struct srpc_join_reply *jrep = &rep->msg_body.join_reply;
1915 struct lstcon_group *grp = NULL;
1916 struct lstcon_ndlink *ndl;
1919 sfw_unpack_message(req);
1921 mutex_lock(&console_session.ses_mutex);
1923 jrep->join_sid = console_session.ses_id;
1925 if (console_session.ses_id.ses_nid == LNET_NID_ANY) {
1926 jrep->join_status = ESRCH;
1930 if (lstcon_session_feats_check(req->msg_ses_feats) != 0) {
1931 jrep->join_status = EPROTO;
1935 if (jreq->join_sid.ses_nid != LNET_NID_ANY &&
1936 !lstcon_session_match(jreq->join_sid)) {
1937 jrep->join_status = EBUSY;
1941 if (lstcon_group_find(jreq->join_group, &grp) != 0) {
1942 rc = lstcon_group_alloc(jreq->join_group, &grp);
1944 CERROR("Out of memory\n");
1948 list_add_tail(&grp->grp_link,
1949 &console_session.ses_grp_list);
1950 lstcon_group_addref(grp);
1953 if (grp->grp_ref > 2) {
1954 /* Group in using */
1955 jrep->join_status = EBUSY;
1959 rc = lstcon_group_ndlink_find(grp, rpc->srpc_peer, &ndl, 0);
1961 jrep->join_status = EEXIST;
1965 rc = lstcon_group_ndlink_find(grp, rpc->srpc_peer, &ndl, 1);
1967 CERROR("Out of memory\n");
1971 ndl->ndl_node->nd_state = LST_NODE_ACTIVE;
1972 ndl->ndl_node->nd_timeout = console_session.ses_timeout;
1974 if (grp->grp_userland == 0)
1975 grp->grp_userland = 1;
1977 strlcpy(jrep->join_session, console_session.ses_name,
1978 sizeof(jrep->join_session));
1979 jrep->join_timeout = console_session.ses_timeout;
1980 jrep->join_status = 0;
1983 rep->msg_ses_feats = console_session.ses_features;
1985 lstcon_group_decref(grp);
1987 mutex_unlock(&console_session.ses_mutex);
1992 static struct srpc_service lstcon_acceptor_service;
1994 static void lstcon_init_acceptor_service(void)
1996 /* initialize selftest console acceptor service table */
1997 lstcon_acceptor_service.sv_name = "join session";
1998 lstcon_acceptor_service.sv_handler = lstcon_acceptor_handle;
1999 lstcon_acceptor_service.sv_id = SRPC_SERVICE_JOIN;
2000 lstcon_acceptor_service.sv_wi_total = SFW_FRWK_WI_MAX;
2003 static struct notifier_block lstcon_ioctl_handler = {
2004 .notifier_call = lstcon_ioctl_entry,
2007 /* initialize console */
2009 lstcon_console_init(void)
2014 console_session.ses_id = LST_INVALID_SID;
2015 console_session.ses_state = LST_SESSION_NONE;
2016 console_session.ses_timeout = 0;
2017 console_session.ses_force = 0;
2018 console_session.ses_expired = 0;
2019 console_session.ses_feats_updated = 0;
2020 console_session.ses_features = LST_FEATS_MASK;
2021 console_session.ses_laststamp = ktime_get_real_seconds();
2023 mutex_init(&console_session.ses_mutex);
2025 INIT_LIST_HEAD(&console_session.ses_ndl_list);
2026 INIT_LIST_HEAD(&console_session.ses_grp_list);
2027 INIT_LIST_HEAD(&console_session.ses_bat_list);
2028 INIT_LIST_HEAD(&console_session.ses_trans_list);
2030 LIBCFS_ALLOC(console_session.ses_ndl_hash,
2031 sizeof(struct list_head) * LST_GLOBAL_HASHSIZE);
2032 if (console_session.ses_ndl_hash == NULL)
2035 for (i = 0; i < LST_GLOBAL_HASHSIZE; i++)
2036 INIT_LIST_HEAD(&console_session.ses_ndl_hash[i]);
2039 /* initialize acceptor service table */
2040 lstcon_init_acceptor_service();
2042 rc = srpc_add_service(&lstcon_acceptor_service);
2043 LASSERT(rc != -EBUSY);
2045 LIBCFS_FREE(console_session.ses_ndl_hash,
2046 sizeof(struct list_head) * LST_GLOBAL_HASHSIZE);
2050 rc = srpc_service_add_buffers(&lstcon_acceptor_service,
2051 lstcon_acceptor_service.sv_wi_total);
2057 rc = blocking_notifier_chain_register(&libcfs_ioctl_list,
2058 &lstcon_ioctl_handler);
2060 lstcon_rpc_module_init();
2065 srpc_shutdown_service(&lstcon_acceptor_service);
2066 srpc_remove_service(&lstcon_acceptor_service);
2068 LIBCFS_FREE(console_session.ses_ndl_hash,
2069 sizeof(struct list_head) * LST_GLOBAL_HASHSIZE);
2071 srpc_wait_service_shutdown(&lstcon_acceptor_service);
2077 lstcon_console_fini(void)
2081 blocking_notifier_chain_unregister(&libcfs_ioctl_list,
2082 &lstcon_ioctl_handler);
2084 mutex_lock(&console_session.ses_mutex);
2086 srpc_shutdown_service(&lstcon_acceptor_service);
2087 srpc_remove_service(&lstcon_acceptor_service);
2089 if (console_session.ses_state != LST_SESSION_NONE)
2090 lstcon_session_end();
2092 lstcon_rpc_module_fini();
2094 mutex_unlock(&console_session.ses_mutex);
2096 LASSERT(list_empty(&console_session.ses_ndl_list));
2097 LASSERT(list_empty(&console_session.ses_grp_list));
2098 LASSERT(list_empty(&console_session.ses_bat_list));
2099 LASSERT(list_empty(&console_session.ses_trans_list));
2101 for (i = 0; i < LST_NODE_HASHSIZE; i++)
2102 LASSERT(list_empty(&console_session.ses_ndl_hash[i]));
2104 LIBCFS_FREE(console_session.ses_ndl_hash,
2105 sizeof(struct list_head) * LST_GLOBAL_HASHSIZE);
2107 srpc_wait_service_shutdown(&lstcon_acceptor_service);