Whamcloud - gitweb
LU-2446 build: Update Whamcloud copyright messages for Intel
[fs/lustre-release.git] / lnet / selftest / conctl.c
1 /*
2  * GPL HEADER START
3  *
4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5  *
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.
9  *
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).
15  *
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.sun.com/software/products/lustre/docs/GPLv2.pdf
19  *
20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
21  * CA 95054 USA or visit www.sun.com if you need additional information or
22  * have any questions.
23  *
24  * GPL HEADER END
25  */
26 /*
27  * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
28  * Use is subject to license terms.
29  */
30 /*
31  * This file is part of Lustre, http://www.lustre.org/
32  * Lustre is a trademark of Sun Microsystems, Inc.
33  *
34  * lnet/selftest/conctl.c
35  *
36  * IOC handle in kernel 
37  *
38  * Author: Liang Zhen <liangzhen@clusterfs.com>
39  */
40 #ifdef __KERNEL__
41
42 #include <libcfs/libcfs.h>
43 #include <lnet/lib-lnet.h>
44 #include <lnet/lnetst.h>
45 #include "console.h"
46
47 int
48 lst_session_new_ioctl(lstio_session_new_args_t *args)
49 {
50         char      *name;
51         int        rc;
52
53         if (args->lstio_ses_idp   == NULL || /* address for output sid */
54             args->lstio_ses_key   == 0 || /* no key is specified */
55             args->lstio_ses_namep == NULL || /* session name */
56             args->lstio_ses_nmlen <= 0 ||
57             args->lstio_ses_nmlen > LST_NAME_SIZE)
58                 return -EINVAL;
59
60         LIBCFS_ALLOC(name, args->lstio_ses_nmlen + 1);
61         if (name == NULL)
62                 return -ENOMEM;
63
64         if (cfs_copy_from_user(name,
65                                args->lstio_ses_namep,
66                                args->lstio_ses_nmlen)) {
67                 LIBCFS_FREE(name, args->lstio_ses_nmlen + 1);
68                 return -EFAULT;
69         }
70
71         name[args->lstio_ses_nmlen] = 0;
72
73         rc = lstcon_session_new(name,
74                                 args->lstio_ses_key,
75                                 args->lstio_ses_feats,
76                                 args->lstio_ses_force,
77                                 args->lstio_ses_timeout,
78                                 args->lstio_ses_idp);
79
80         LIBCFS_FREE(name, args->lstio_ses_nmlen + 1);
81         return rc;
82 }
83
84 int
85 lst_session_end_ioctl(lstio_session_end_args_t *args)
86 {
87         if (args->lstio_ses_key != console_session.ses_key)
88                 return -EACCES;
89
90         return lstcon_session_end();
91 }
92
93 int
94 lst_session_info_ioctl(lstio_session_info_args_t *args)
95 {
96         /* no checking of key */
97
98         if (args->lstio_ses_idp   == NULL || /* address for ouput sid */
99             args->lstio_ses_keyp  == NULL || /* address for ouput key */
100             args->lstio_ses_featp  == NULL || /* address for ouput features */
101             args->lstio_ses_ndinfo == NULL || /* address for output ndinfo */
102             args->lstio_ses_namep == NULL || /* address for ouput name */
103             args->lstio_ses_nmlen <= 0 ||
104             args->lstio_ses_nmlen > LST_NAME_SIZE)
105                 return -EINVAL;
106
107         return lstcon_session_info(args->lstio_ses_idp,
108                                    args->lstio_ses_keyp,
109                                    args->lstio_ses_featp,
110                                    args->lstio_ses_ndinfo,
111                                    args->lstio_ses_namep,
112                                    args->lstio_ses_nmlen);
113 }
114
115 int
116 lst_debug_ioctl(lstio_debug_args_t *args)
117 {
118         char   *name   = NULL;
119         int     client = 1;
120         int     rc;
121
122         if (args->lstio_dbg_key != console_session.ses_key)
123                 return -EACCES;
124
125         if (args->lstio_dbg_resultp == NULL)
126                 return -EINVAL;
127
128         if (args->lstio_dbg_namep != NULL && /* name of batch/group */
129             (args->lstio_dbg_nmlen <= 0 ||
130              args->lstio_dbg_nmlen > LST_NAME_SIZE))
131                 return -EINVAL;
132
133         if (args->lstio_dbg_namep != NULL) {
134                 LIBCFS_ALLOC(name, args->lstio_dbg_nmlen + 1);
135                 if (name == NULL)
136                         return -ENOMEM;
137
138                 if (cfs_copy_from_user(name, args->lstio_dbg_namep,
139                                        args->lstio_dbg_nmlen)) {
140                         LIBCFS_FREE(name, args->lstio_dbg_nmlen + 1);
141
142                         return -EFAULT;
143                 }
144
145                 name[args->lstio_dbg_nmlen] = 0;
146         }
147
148         rc = -EINVAL;
149
150         switch (args->lstio_dbg_type) {
151         case LST_OPC_SESSION:
152                 rc = lstcon_session_debug(args->lstio_dbg_timeout,
153                                           args->lstio_dbg_resultp);
154                 break;
155
156         case LST_OPC_BATCHSRV:
157                 client = 0;
158         case LST_OPC_BATCHCLI:
159                 if (name == NULL)
160                         goto out;
161
162                 rc = lstcon_batch_debug(args->lstio_dbg_timeout,
163                                         name, client, args->lstio_dbg_resultp);
164                 break;
165
166         case LST_OPC_GROUP:
167                 if (name == NULL)
168                         goto out;
169
170                 rc = lstcon_group_debug(args->lstio_dbg_timeout,
171                                         name, args->lstio_dbg_resultp);
172                 break;
173
174         case LST_OPC_NODES:
175                 if (args->lstio_dbg_count <= 0 ||
176                     args->lstio_dbg_idsp == NULL)
177                         goto out;
178
179                 rc = lstcon_nodes_debug(args->lstio_dbg_timeout,
180                                         args->lstio_dbg_count,
181                                         args->lstio_dbg_idsp,
182                                         args->lstio_dbg_resultp);
183                 break;
184
185         default:
186                 break;
187         }
188
189 out:
190         if (name != NULL)
191                 LIBCFS_FREE(name, args->lstio_dbg_nmlen + 1);
192
193         return rc;
194 }
195
196 int
197 lst_group_add_ioctl(lstio_group_add_args_t *args)
198 {
199         char           *name;
200         int             rc;
201
202         if (args->lstio_grp_key != console_session.ses_key)
203                 return -EACCES;
204
205         if (args->lstio_grp_namep == NULL||
206             args->lstio_grp_nmlen <= 0 || 
207             args->lstio_grp_nmlen > LST_NAME_SIZE)
208                 return -EINVAL;
209
210         LIBCFS_ALLOC(name, args->lstio_grp_nmlen + 1);
211         if (name == NULL)
212                 return -ENOMEM;
213
214         if (cfs_copy_from_user(name,
215                                args->lstio_grp_namep,
216                                args->lstio_grp_nmlen)) {
217                 LIBCFS_FREE(name, args->lstio_grp_nmlen);
218                 return -EFAULT;
219         }
220
221         name[args->lstio_grp_nmlen] = 0;
222
223         rc = lstcon_group_add(name);
224
225         LIBCFS_FREE(name, args->lstio_grp_nmlen + 1);
226
227         return rc;
228 }
229
230 int
231 lst_group_del_ioctl(lstio_group_del_args_t *args)
232 {
233         int     rc;
234         char   *name;
235
236         if (args->lstio_grp_key != console_session.ses_key)
237                 return -EACCES;
238
239         if (args->lstio_grp_namep == NULL ||
240             args->lstio_grp_nmlen <= 0 ||
241             args->lstio_grp_nmlen > LST_NAME_SIZE)
242                 return -EINVAL;
243
244         LIBCFS_ALLOC(name, args->lstio_grp_nmlen + 1);
245         if (name == NULL)
246                 return -ENOMEM;
247
248         if (cfs_copy_from_user(name,
249                                args->lstio_grp_namep,
250                                args->lstio_grp_nmlen)) {
251                 LIBCFS_FREE(name, args->lstio_grp_nmlen + 1);
252                 return -EFAULT;
253         }
254
255         name[args->lstio_grp_nmlen] = 0;
256
257         rc = lstcon_group_del(name);
258
259         LIBCFS_FREE(name, args->lstio_grp_nmlen + 1);
260
261         return rc;
262 }
263
264 int
265 lst_group_update_ioctl(lstio_group_update_args_t *args)
266 {
267         int     rc;
268         char   *name;
269
270         if (args->lstio_grp_key != console_session.ses_key)
271                 return -EACCES;
272
273         if (args->lstio_grp_resultp == NULL ||
274             args->lstio_grp_namep == NULL ||
275             args->lstio_grp_nmlen <= 0 || 
276             args->lstio_grp_nmlen > LST_NAME_SIZE)
277                 return -EINVAL;
278
279         LIBCFS_ALLOC(name, args->lstio_grp_nmlen + 1);
280         if (name == NULL)
281                 return -ENOMEM;
282
283         if (cfs_copy_from_user(name,
284                            args->lstio_grp_namep,
285                            args->lstio_grp_nmlen)) {
286                 LIBCFS_FREE(name, args->lstio_grp_nmlen + 1);
287                 return -EFAULT;
288         }
289
290         name[args->lstio_grp_nmlen] = 0;
291
292         switch (args->lstio_grp_opc) {
293         case LST_GROUP_CLEAN:
294                 rc = lstcon_group_clean(name, args->lstio_grp_args);
295                 break;
296
297         case LST_GROUP_REFRESH:
298                 rc = lstcon_group_refresh(name, args->lstio_grp_resultp);
299                 break;
300
301         case LST_GROUP_RMND:
302                 if (args->lstio_grp_count  <= 0 ||
303                     args->lstio_grp_idsp == NULL) {
304                         rc = -EINVAL;
305                         break;
306                 }
307                 rc = lstcon_nodes_remove(name, args->lstio_grp_count,
308                                          args->lstio_grp_idsp,
309                                          args->lstio_grp_resultp);
310                 break;
311
312         default:
313                 rc = -EINVAL;
314                 break;
315         }
316
317         LIBCFS_FREE(name, args->lstio_grp_nmlen + 1);
318
319         return rc;
320 }
321
322 int
323 lst_nodes_add_ioctl(lstio_group_nodes_args_t *args)
324 {
325         unsigned feats;
326         int     rc;
327         char   *name;
328
329         if (args->lstio_grp_key != console_session.ses_key)
330                 return -EACCES;
331
332         if (args->lstio_grp_idsp == NULL || /* array of ids */
333             args->lstio_grp_count <= 0 ||
334             args->lstio_grp_resultp == NULL ||
335             args->lstio_grp_featp == NULL ||
336             args->lstio_grp_namep == NULL ||
337             args->lstio_grp_nmlen <= 0 ||
338             args->lstio_grp_nmlen > LST_NAME_SIZE)
339                 return -EINVAL;
340
341         LIBCFS_ALLOC(name, args->lstio_grp_nmlen + 1);
342         if (name == NULL)
343                 return -ENOMEM;
344
345         if (cfs_copy_from_user(name, args->lstio_grp_namep,
346                                args->lstio_grp_nmlen)) {
347                 LIBCFS_FREE(name, args->lstio_grp_nmlen + 1);
348
349                 return -EFAULT;
350         }
351
352         name[args->lstio_grp_nmlen] = 0;
353
354         rc = lstcon_nodes_add(name, args->lstio_grp_count,
355                               args->lstio_grp_idsp, &feats,
356                               args->lstio_grp_resultp);
357
358         LIBCFS_FREE(name, args->lstio_grp_nmlen + 1);
359         if (rc == 0 &&
360             cfs_copy_to_user(args->lstio_grp_featp, &feats, sizeof(feats))) {
361                 return -EINVAL;
362         }
363
364         return rc;
365 }
366
367 int
368 lst_group_list_ioctl(lstio_group_list_args_t *args)
369 {
370         if (args->lstio_grp_key != console_session.ses_key) 
371                 return -EACCES;
372
373         if (args->lstio_grp_idx   < 0 ||
374             args->lstio_grp_namep == NULL ||
375             args->lstio_grp_nmlen <= 0 ||
376             args->lstio_grp_nmlen > LST_NAME_SIZE)
377                 return -EINVAL;
378
379         return lstcon_group_list(args->lstio_grp_idx,
380                               args->lstio_grp_nmlen,
381                               args->lstio_grp_namep);
382 }
383
384 int
385 lst_group_info_ioctl(lstio_group_info_args_t *args)
386 {
387         char           *name;
388         int             ndent;
389         int             index;
390         int             rc;
391
392         if (args->lstio_grp_key != console_session.ses_key)
393                 return -EACCES;
394
395         if (args->lstio_grp_namep == NULL ||
396             args->lstio_grp_nmlen <= 0 ||
397             args->lstio_grp_nmlen > LST_NAME_SIZE)
398                 return -EINVAL;
399
400         if (args->lstio_grp_entp  == NULL && /* output: group entry */
401             args->lstio_grp_dentsp == NULL)  /* output: node entry */
402                 return -EINVAL;
403
404         if (args->lstio_grp_dentsp != NULL) { /* have node entry */
405                 if (args->lstio_grp_idxp == NULL || /* node index */
406                     args->lstio_grp_ndentp == NULL) /* # of node entry */
407                         return -EINVAL;
408
409                 if (cfs_copy_from_user(&ndent, args->lstio_grp_ndentp,
410                                        sizeof(ndent)) ||
411                     cfs_copy_from_user(&index, args->lstio_grp_idxp,
412                                        sizeof(index)))
413                         return -EFAULT;
414
415                 if (ndent <= 0 || index < 0)
416                         return -EINVAL;
417         }
418
419         LIBCFS_ALLOC(name, args->lstio_grp_nmlen + 1);
420         if (name == NULL)
421                 return -ENOMEM;
422
423         if (cfs_copy_from_user(name,
424                                args->lstio_grp_namep,
425                                args->lstio_grp_nmlen)) {
426                 LIBCFS_FREE(name, args->lstio_grp_nmlen + 1);
427                 return -EFAULT;
428         }
429
430         name[args->lstio_grp_nmlen] = 0;
431
432         rc = lstcon_group_info(name, args->lstio_grp_entp,
433                                &index, &ndent, args->lstio_grp_dentsp);
434
435         LIBCFS_FREE(name, args->lstio_grp_nmlen + 1);
436
437         if (rc != 0) 
438                 return rc;
439
440         if (args->lstio_grp_dentsp != NULL && 
441             (cfs_copy_to_user(args->lstio_grp_idxp, &index, sizeof(index)) ||
442              cfs_copy_to_user(args->lstio_grp_ndentp, &ndent, sizeof(ndent))))
443                 rc = -EFAULT;
444
445         return 0;
446 }
447
448 int
449 lst_batch_add_ioctl(lstio_batch_add_args_t *args)
450 {
451         int             rc;
452         char           *name;
453
454         if (args->lstio_bat_key != console_session.ses_key)
455                 return -EACCES;
456
457         if (args->lstio_bat_namep == NULL ||
458             args->lstio_bat_nmlen <= 0 ||
459             args->lstio_bat_nmlen > LST_NAME_SIZE)
460                 return -EINVAL;
461
462         LIBCFS_ALLOC(name, args->lstio_bat_nmlen + 1);
463         if (name == NULL)
464                 return -ENOMEM;
465
466         if (cfs_copy_from_user(name,
467                                args->lstio_bat_namep,
468                                args->lstio_bat_nmlen)) {
469                 LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
470                 return -EFAULT;
471         }
472
473         name[args->lstio_bat_nmlen] = 0;
474
475         rc = lstcon_batch_add(name);
476
477         LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
478
479         return rc;
480 }
481
482 int
483 lst_batch_run_ioctl(lstio_batch_run_args_t *args)
484 {
485         int             rc;
486         char           *name;
487
488         if (args->lstio_bat_key != console_session.ses_key)
489                 return -EACCES;
490
491         if (args->lstio_bat_namep == NULL ||
492             args->lstio_bat_nmlen <= 0 ||
493             args->lstio_bat_nmlen > LST_NAME_SIZE)
494                 return -EINVAL;
495
496         LIBCFS_ALLOC(name, args->lstio_bat_nmlen + 1);
497         if (name == NULL)
498                 return -ENOMEM;
499
500         if (cfs_copy_from_user(name,
501                                args->lstio_bat_namep,
502                                args->lstio_bat_nmlen)) {
503                 LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
504                 return -EFAULT;
505         }
506
507         name[args->lstio_bat_nmlen] = 0;
508
509         rc = lstcon_batch_run(name, args->lstio_bat_timeout,
510                               args->lstio_bat_resultp);
511
512         LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
513
514         return rc;
515 }
516
517 int
518 lst_batch_stop_ioctl(lstio_batch_stop_args_t *args)
519 {
520         int             rc;
521         char           *name;
522
523         if (args->lstio_bat_key != console_session.ses_key)
524                 return -EACCES;
525
526         if (args->lstio_bat_resultp == NULL ||
527             args->lstio_bat_namep == NULL ||
528             args->lstio_bat_nmlen <= 0 ||
529             args->lstio_bat_nmlen > LST_NAME_SIZE)
530                 return -EINVAL;
531
532         LIBCFS_ALLOC(name, args->lstio_bat_nmlen + 1);
533         if (name == NULL)
534                 return -ENOMEM;
535
536         if (cfs_copy_from_user(name,
537                                args->lstio_bat_namep,
538                                args->lstio_bat_nmlen)) {
539                 LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
540                 return -EFAULT;
541         }
542
543         name[args->lstio_bat_nmlen] = 0;
544
545         rc = lstcon_batch_stop(name, args->lstio_bat_force,
546                                args->lstio_bat_resultp);
547
548         LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
549
550         return rc;
551 }
552
553 int
554 lst_batch_query_ioctl(lstio_batch_query_args_t *args)
555 {
556         char   *name;
557         int     rc;
558
559         if (args->lstio_bat_key != console_session.ses_key)
560                 return -EACCES;
561
562         if (args->lstio_bat_resultp == NULL ||
563             args->lstio_bat_namep == NULL ||
564             args->lstio_bat_nmlen <= 0 ||
565             args->lstio_bat_nmlen > LST_NAME_SIZE)
566                 return -EINVAL;
567
568         if (args->lstio_bat_testidx < 0)
569                 return -EINVAL;
570
571         LIBCFS_ALLOC(name, args->lstio_bat_nmlen + 1);
572         if (name == NULL)
573                 return -ENOMEM;
574
575         if (cfs_copy_from_user(name,
576                                args->lstio_bat_namep,
577                                args->lstio_bat_nmlen)) {
578                 LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
579                 return -EFAULT;
580         }
581
582         name[args->lstio_bat_nmlen] = 0;
583
584         rc = lstcon_test_batch_query(name,
585                                      args->lstio_bat_testidx,
586                                      args->lstio_bat_client,
587                                      args->lstio_bat_timeout,
588                                      args->lstio_bat_resultp);
589
590         LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
591
592         return rc;
593 }
594
595 int
596 lst_batch_list_ioctl(lstio_batch_list_args_t *args)
597 {
598         if (args->lstio_bat_key != console_session.ses_key)
599                 return -EACCES;
600
601         if (args->lstio_bat_idx   < 0 ||
602             args->lstio_bat_namep == NULL ||
603             args->lstio_bat_nmlen <= 0 ||
604             args->lstio_bat_nmlen > LST_NAME_SIZE)
605                 return -EINVAL;
606
607         return lstcon_batch_list(args->lstio_bat_idx,
608                               args->lstio_bat_nmlen,
609                               args->lstio_bat_namep);
610 }
611
612 int
613 lst_batch_info_ioctl(lstio_batch_info_args_t *args)
614 {
615         char           *name;
616         int             rc;
617         int             index;
618         int             ndent;
619
620         if (args->lstio_bat_key != console_session.ses_key)
621                 return -EACCES;
622
623         if (args->lstio_bat_namep == NULL || /* batch name */
624             args->lstio_bat_nmlen <= 0 ||
625             args->lstio_bat_nmlen > LST_NAME_SIZE)
626                 return -EINVAL;
627
628         if (args->lstio_bat_entp == NULL && /* output: batch entry */
629             args->lstio_bat_dentsp == NULL) /* output: node entry */
630                 return -EINVAL;
631
632         if (args->lstio_bat_dentsp != NULL) { /* have node entry */
633                 if (args->lstio_bat_idxp == NULL || /* node index */
634                     args->lstio_bat_ndentp == NULL) /* # of node entry */
635                         return -EINVAL;
636
637                 if (cfs_copy_from_user(&index, args->lstio_bat_idxp,
638                                        sizeof(index)) ||
639                     cfs_copy_from_user(&ndent, args->lstio_bat_ndentp,
640                                        sizeof(ndent)))
641                         return -EFAULT;
642
643                 if (ndent <= 0 || index < 0)
644                         return -EINVAL;
645         }
646
647         LIBCFS_ALLOC(name, args->lstio_bat_nmlen + 1);
648         if (name == NULL)
649                 return -ENOMEM;
650
651         if (cfs_copy_from_user(name,
652                                args->lstio_bat_namep, args->lstio_bat_nmlen)) {
653                 LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
654                 return -EFAULT;
655         }
656
657         name[args->lstio_bat_nmlen] = 0;
658
659         rc = lstcon_batch_info(name,
660                             args->lstio_bat_entp, args->lstio_bat_server,
661                             args->lstio_bat_testidx, &index, &ndent,
662                             args->lstio_bat_dentsp);
663
664         LIBCFS_FREE(name, args->lstio_bat_nmlen + 1);
665
666         if (rc != 0)
667                 return rc;
668
669         if (args->lstio_bat_dentsp != NULL && 
670             (cfs_copy_to_user(args->lstio_bat_idxp, &index, sizeof(index)) ||
671              cfs_copy_to_user(args->lstio_bat_ndentp, &ndent, sizeof(ndent))))
672                 rc = -EFAULT;
673
674         return rc;
675 }
676
677 int
678 lst_stat_query_ioctl(lstio_stat_args_t *args)
679 {
680         int             rc;
681         char           *name;
682
683         /* TODO: not finished */
684         if (args->lstio_sta_key != console_session.ses_key)
685                 return -EACCES;
686
687         if (args->lstio_sta_resultp == NULL ||
688             (args->lstio_sta_namep  == NULL &&
689              args->lstio_sta_idsp   == NULL) ||
690             args->lstio_sta_nmlen <= 0 ||
691             args->lstio_sta_nmlen > LST_NAME_SIZE)
692                 return -EINVAL;
693
694         if (args->lstio_sta_idsp != NULL &&
695             args->lstio_sta_count <= 0)
696                 return -EINVAL;
697
698         LIBCFS_ALLOC(name, args->lstio_sta_nmlen + 1);
699         if (name == NULL)
700                 return -ENOMEM;
701
702         if (cfs_copy_from_user(name, args->lstio_sta_namep,
703                                args->lstio_sta_nmlen)) {
704                 LIBCFS_FREE(name, args->lstio_sta_nmlen + 1);
705                 return -EFAULT;
706         }
707
708         if (args->lstio_sta_idsp == NULL) {
709                 rc = lstcon_group_stat(name, args->lstio_sta_timeout,
710                                        args->lstio_sta_resultp);
711         } else {
712                 rc = lstcon_nodes_stat(args->lstio_sta_count,
713                                        args->lstio_sta_idsp,
714                                        args->lstio_sta_timeout,
715                                        args->lstio_sta_resultp);
716         }
717
718         LIBCFS_FREE(name, args->lstio_sta_nmlen + 1);
719
720         return rc;
721 }
722
723 int lst_test_add_ioctl(lstio_test_args_t *args)
724 {
725         char           *name;
726         char           *srcgrp = NULL;
727         char           *dstgrp = NULL;
728         void           *param = NULL;
729         int             ret = 0;
730         int             rc = -ENOMEM;
731
732         if (args->lstio_tes_resultp == NULL ||
733             args->lstio_tes_retp == NULL ||
734             args->lstio_tes_bat_name == NULL || /* no specified batch */
735             args->lstio_tes_bat_nmlen <= 0 ||
736             args->lstio_tes_bat_nmlen > LST_NAME_SIZE ||
737             args->lstio_tes_sgrp_name == NULL || /* no source group */
738             args->lstio_tes_sgrp_nmlen <= 0 ||
739             args->lstio_tes_sgrp_nmlen > LST_NAME_SIZE ||
740             args->lstio_tes_dgrp_name == NULL || /* no target group */
741             args->lstio_tes_dgrp_nmlen <= 0 ||
742             args->lstio_tes_dgrp_nmlen > LST_NAME_SIZE)
743                 return -EINVAL;
744
745         if (args->lstio_tes_loop == 0 || /* negative is infinite */
746             args->lstio_tes_concur <= 0 ||
747             args->lstio_tes_dist <= 0 ||
748             args->lstio_tes_span <= 0)
749                 return -EINVAL;
750
751         /* have parameter, check if parameter length is valid */
752         if (args->lstio_tes_param != NULL &&
753             (args->lstio_tes_param_len <= 0 ||
754              args->lstio_tes_param_len > CFS_PAGE_SIZE - sizeof(lstcon_test_t)))
755                 return -EINVAL;
756
757         LIBCFS_ALLOC(name, args->lstio_tes_bat_nmlen + 1);
758         if (name == NULL)
759                 return rc;
760
761         LIBCFS_ALLOC(srcgrp, args->lstio_tes_sgrp_nmlen + 1);
762         if (srcgrp == NULL) 
763                 goto out;
764
765         LIBCFS_ALLOC(dstgrp, args->lstio_tes_dgrp_nmlen + 1);
766          if (dstgrp == NULL)
767                 goto out;
768
769         if (args->lstio_tes_param != NULL) {
770                 LIBCFS_ALLOC(param, args->lstio_tes_param_len);
771                 if (param == NULL) 
772                         goto out;
773         }
774
775         rc = -EFAULT;
776         if (cfs_copy_from_user(name,
777                               args->lstio_tes_bat_name,
778                               args->lstio_tes_bat_nmlen) ||
779             cfs_copy_from_user(srcgrp,
780                               args->lstio_tes_sgrp_name,
781                               args->lstio_tes_sgrp_nmlen) ||
782             cfs_copy_from_user(dstgrp,
783                               args->lstio_tes_dgrp_name,
784                               args->lstio_tes_dgrp_nmlen) ||
785             cfs_copy_from_user(param, args->lstio_tes_param,
786                               args->lstio_tes_param_len))
787                 goto out;
788
789         rc = lstcon_test_add(name,
790                             args->lstio_tes_type,
791                             args->lstio_tes_loop,
792                             args->lstio_tes_concur,
793                             args->lstio_tes_dist, args->lstio_tes_span,
794                             srcgrp, dstgrp, param, args->lstio_tes_param_len,
795                             &ret, args->lstio_tes_resultp);
796
797         if (ret != 0)
798                 rc = (cfs_copy_to_user(args->lstio_tes_retp, &ret,
799                                        sizeof(ret))) ? -EFAULT : 0;
800 out:
801         if (name != NULL)
802                 LIBCFS_FREE(name, args->lstio_tes_bat_nmlen + 1);
803
804         if (srcgrp != NULL)
805                 LIBCFS_FREE(srcgrp, args->lstio_tes_sgrp_nmlen + 1);
806
807         if (dstgrp != NULL)
808                 LIBCFS_FREE(dstgrp, args->lstio_tes_dgrp_nmlen + 1);
809
810         if (param != NULL)
811                 LIBCFS_FREE(param, args->lstio_tes_param_len);
812
813         return rc;
814 }
815
816 int
817 lstcon_ioctl_entry(unsigned int cmd, struct libcfs_ioctl_data *data)
818 {
819         char   *buf;
820         int     opc = data->ioc_u32[0];
821         int     rc;
822
823         if (cmd != IOC_LIBCFS_LNETST)
824                 return -EINVAL;
825
826         if (data->ioc_plen1 > CFS_PAGE_SIZE)
827                 return -EINVAL;
828
829         LIBCFS_ALLOC(buf, data->ioc_plen1);
830         if (buf == NULL)
831                 return -ENOMEM;
832
833         /* copy in parameter */
834         if (cfs_copy_from_user(buf, data->ioc_pbuf1, data->ioc_plen1)) {
835                 LIBCFS_FREE(buf, data->ioc_plen1);
836                 return -EFAULT;
837         }
838
839         mutex_lock(&console_session.ses_mutex);
840
841         console_session.ses_laststamp = cfs_time_current_sec();
842
843         if (console_session.ses_shutdown) {
844                 rc = -ESHUTDOWN;
845                 goto out;
846         }
847
848         if (console_session.ses_expired)
849                 lstcon_session_end();
850
851         if (opc != LSTIO_SESSION_NEW &&
852             console_session.ses_state == LST_SESSION_NONE) {
853                 CDEBUG(D_NET, "LST no active session\n");
854                 rc = -ESRCH;
855                 goto out;
856         }
857
858         memset(&console_session.ses_trans_stat, 0, sizeof(lstcon_trans_stat_t));
859
860         switch (opc) {
861                 case LSTIO_SESSION_NEW:
862                         rc = lst_session_new_ioctl((lstio_session_new_args_t *)buf);
863                         break;
864                 case LSTIO_SESSION_END:
865                         rc = lst_session_end_ioctl((lstio_session_end_args_t *)buf);
866                         break;
867                 case LSTIO_SESSION_INFO:
868                         rc = lst_session_info_ioctl((lstio_session_info_args_t *)buf);
869                         break;
870                 case LSTIO_DEBUG:
871                         rc = lst_debug_ioctl((lstio_debug_args_t *)buf);
872                         break;
873                 case LSTIO_GROUP_ADD:
874                         rc = lst_group_add_ioctl((lstio_group_add_args_t *)buf);
875                         break;
876                 case LSTIO_GROUP_DEL:
877                         rc = lst_group_del_ioctl((lstio_group_del_args_t *)buf);
878                         break;
879                 case LSTIO_GROUP_UPDATE:
880                         rc = lst_group_update_ioctl((lstio_group_update_args_t *)buf);
881                         break;
882                 case LSTIO_NODES_ADD:
883                         rc = lst_nodes_add_ioctl((lstio_group_nodes_args_t *)buf);
884                         break;
885                 case LSTIO_GROUP_LIST:
886                         rc = lst_group_list_ioctl((lstio_group_list_args_t *)buf);
887                         break;
888                 case LSTIO_GROUP_INFO:
889                         rc = lst_group_info_ioctl((lstio_group_info_args_t *)buf);
890                         break;
891                 case LSTIO_BATCH_ADD:
892                         rc = lst_batch_add_ioctl((lstio_batch_add_args_t *)buf);
893                         break;
894                 case LSTIO_BATCH_START:
895                         rc = lst_batch_run_ioctl((lstio_batch_run_args_t *)buf);
896                         break;
897                 case LSTIO_BATCH_STOP:
898                         rc = lst_batch_stop_ioctl((lstio_batch_stop_args_t *)buf);
899                         break;
900                 case LSTIO_BATCH_QUERY:
901                         rc = lst_batch_query_ioctl((lstio_batch_query_args_t *)buf);
902                         break;
903                 case LSTIO_BATCH_LIST:
904                         rc = lst_batch_list_ioctl((lstio_batch_list_args_t *)buf);
905                         break;
906                 case LSTIO_BATCH_INFO:
907                         rc = lst_batch_info_ioctl((lstio_batch_info_args_t *)buf);
908                         break;
909                 case LSTIO_TEST_ADD:
910                         rc = lst_test_add_ioctl((lstio_test_args_t *)buf);
911                         break;
912                 case LSTIO_STAT_QUERY:
913                         rc = lst_stat_query_ioctl((lstio_stat_args_t *)buf);
914                         break;
915                 default:
916                         rc = -EINVAL;
917         }
918
919         if (cfs_copy_to_user(data->ioc_pbuf2, &console_session.ses_trans_stat,
920                              sizeof(lstcon_trans_stat_t)))
921                 rc = -EFAULT;
922 out:
923         mutex_unlock(&console_session.ses_mutex);
924
925         LIBCFS_FREE(buf, data->ioc_plen1);
926
927         return rc;
928 }
929
930 EXPORT_SYMBOL(lstcon_ioctl_entry);
931
932 #endif