Safe memory allocation in cli_handlers
[multipath-tools/.git] / multipathd / cli_handlers.c
1 /*
2  * Copyright (c) 2005 Christophe Varoqui
3  */
4 #include <checkers.h>
5 #include <memory.h>
6 #include <vector.h>
7 #include <structs.h>
8 #include <structs_vec.h>
9 #include <libdevmapper.h>
10 #include <devmapper.h>
11 #include <config.h>
12 #include <configure.h>
13 #include <blacklist.h>
14 #include <debug.h>
15 #include <print.h>
16 #include <sysfs.h>
17 #include <errno.h>
18
19 #include "main.h"
20 #include "cli.h"
21
22 #define REALLOC_REPLY(r, a, m)                                  \
23         do {                                                    \
24                 if ((a)) {                                      \
25                         (r) = REALLOC((r), (m) * 2);            \
26                         if ((r)) {                              \
27                                 memset((r) + (m), 0, (m));      \
28                                 (m) *= 2;                       \
29                         }                                       \
30                 }                                               \
31         } while (0)
32
33 int
34 show_paths (char ** r, int * len, struct vectors * vecs, char * style)
35 {
36         int i;
37         struct path * pp;
38         char * c;
39         char * reply;
40         unsigned int maxlen = INITIAL_REPLY_LEN;
41         int again = 1;
42
43         get_path_layout(vecs->pathvec, 1);
44         reply = MALLOC(maxlen);
45
46         while (again) {
47                 if (!reply)
48                         return 1;
49
50                 c = reply;
51
52                 if (VECTOR_SIZE(vecs->pathvec) > 0)
53                         c += snprint_path_header(c, reply + maxlen - c,
54                                                  style);
55
56                 vector_foreach_slot(vecs->pathvec, pp, i)
57                         c += snprint_path(c, reply + maxlen - c,
58                                           style, pp);
59
60                 again = ((c - reply) == (maxlen - 1));
61
62                 REALLOC_REPLY(reply, again, maxlen);
63         }
64         *r = reply;
65         *len = (int)(c - reply + 1);
66         return 0;
67 }
68
69 int
70 show_map_topology (char ** r, int * len, struct multipath * mpp)
71 {
72         char * c;
73         char * reply;
74         unsigned int maxlen = INITIAL_REPLY_LEN;
75         int again = 1;
76
77         reply = MALLOC(maxlen);
78
79         while (again) {
80                 if (!reply)
81                         return 1;
82
83                 c = reply;
84
85                 c += snprint_multipath_topology(c, reply + maxlen - c, mpp, 2);
86                 again = ((c - reply) == (maxlen - 1));
87
88                 REALLOC_REPLY(reply, again, maxlen);
89         }
90         *r = reply;
91         *len = (int)(c - reply + 1);
92         return 0;
93 }
94
95 int
96 show_maps_topology (char ** r, int * len, struct vectors * vecs)
97 {
98         int i;
99         struct multipath * mpp;
100         char * c;
101         char * reply;
102         unsigned int maxlen = INITIAL_REPLY_LEN;
103         int again = 1;
104  
105         get_path_layout(vecs->pathvec, 0);
106         reply = MALLOC(maxlen);
107
108         while (again) {
109                 if (!reply)
110                         return 1;
111
112                 c = reply;
113
114                 vector_foreach_slot(vecs->mpvec, mpp, i)
115                         c += snprint_multipath_topology(c, reply + maxlen - c,
116                                                         mpp, 2);
117
118                 again = ((c - reply) == (maxlen - 1));
119
120                 REALLOC_REPLY(reply, again, maxlen);
121         }
122         *r = reply;
123         *len = (int)(c - reply + 1);
124         return 0;
125 }
126
127 int
128 show_config (char ** r, int * len)
129 {
130         char * c;
131         char * reply;
132         unsigned int maxlen = INITIAL_REPLY_LEN;
133         int again = 1;
134
135         reply = MALLOC(maxlen);
136
137         while (again) {
138                 if (!reply)
139                         return 1;
140                 c = reply;
141                 c += snprint_defaults(c, reply + maxlen - c);
142                 again = ((c - reply) == maxlen);
143                 if (again) {
144                         reply = REALLOC(reply, maxlen * 2);
145                         if (!reply)
146                                 return 1;
147                         memset(reply + maxlen, 0, maxlen);
148                         maxlen *= 2;
149                         continue;
150                 }
151                 c += snprint_blacklist(c, reply + maxlen - c);
152                 again = ((c - reply) == maxlen);
153                 if (again) {
154                         reply = REALLOC(reply, maxlen * 2);
155                         if (!reply)
156                                 return 1;
157                         memset(reply + maxlen, 0, maxlen);
158                         maxlen *= 2;
159                         continue;
160                 }
161                 c += snprint_blacklist_except(c, reply + maxlen - c);
162                 again = ((c - reply) == maxlen);
163                 if (again) {
164                         reply = REALLOC(reply, maxlen * 2);
165                         if (!reply)
166                                 return 1;
167                         memset(reply + maxlen, 0, maxlen);
168                         maxlen *= 2;
169                         continue;
170                 }
171                 c += snprint_hwtable(c, reply + maxlen - c, conf->hwtable);
172                 again = ((c - reply) == maxlen);
173                 if (again) {
174                         reply = REALLOC(reply, maxlen * 2);
175                         if (!reply)
176                                 return 1;
177                         memset(reply + maxlen, 0, maxlen);
178                         maxlen *= 2;
179                         continue;
180                 }
181                 c += snprint_mptable(c, reply + maxlen - c, conf->mptable);
182                 again = ((c - reply) == maxlen);
183                 REALLOC_REPLY(reply, again, maxlen);
184         }
185         *r = reply;
186         *len = (int)(c - reply + 1);
187         return 0;
188 }
189
190 int
191 cli_list_config (void * v, char ** reply, int * len, void * data)
192 {
193         condlog(3, "list config (operator)");
194
195         return show_config(reply, len);
196 }
197
198 int
199 cli_list_paths (void * v, char ** reply, int * len, void * data)
200 {
201         struct vectors * vecs = (struct vectors *)data;
202
203         condlog(3, "list paths (operator)");
204
205         return show_paths(reply, len, vecs, PRINT_PATH_CHECKER);
206 }
207
208 int
209 cli_list_paths_fmt (void * v, char ** reply, int * len, void * data)
210 {
211         struct vectors * vecs = (struct vectors *)data;
212         char * fmt = get_keyparam(v, FMT);
213
214         condlog(3, "list paths (operator)");
215
216         return show_paths(reply, len, vecs, fmt);
217 }
218
219 int
220 cli_list_map_topology (void * v, char ** reply, int * len, void * data)
221 {
222         struct multipath * mpp;
223         struct vectors * vecs = (struct vectors *)data;
224         char * param = get_keyparam(v, MAP);
225         
226         get_path_layout(vecs->pathvec, 0);
227         mpp = find_mp_by_str(vecs->mpvec, param);
228
229         if (!mpp)
230                 return 1;
231
232         condlog(3, "list multipath %s (operator)", param);
233
234         return show_map_topology(reply, len, mpp);
235 }
236
237 int
238 cli_list_maps_topology (void * v, char ** reply, int * len, void * data)
239 {
240         struct vectors * vecs = (struct vectors *)data;
241
242         condlog(3, "list multipaths (operator)");
243
244         return show_maps_topology(reply, len, vecs);
245 }
246
247 int
248 cli_list_wildcards (void * v, char ** reply, int * len, void * data)
249 {
250         char * c;
251
252         *reply = MALLOC(INITIAL_REPLY_LEN);
253
254         if (!*reply)
255                 return 1;
256
257         c = *reply;
258         c += snprint_wildcards(c, INITIAL_REPLY_LEN);
259
260         *len = INITIAL_REPLY_LEN;
261         return 0;
262 }
263
264 int
265 show_status (char ** r, int *len, struct vectors * vecs)
266 {
267         char * c;
268         char * reply;
269
270         unsigned int maxlen = INITIAL_REPLY_LEN;
271         reply = MALLOC(maxlen);
272
273         if (!reply)
274                 return 1;
275
276         c = reply;
277         c += snprint_status(c, reply + maxlen - c, vecs);
278
279         *r = reply;
280         *len = (int)(c - reply + 1);
281         return 0;
282 }
283
284 int
285 show_maps (char ** r, int *len, struct vectors * vecs, char * style)
286 {
287         int i;
288         struct multipath * mpp;
289         char * c;
290         char * reply;
291         unsigned int maxlen = INITIAL_REPLY_LEN;
292         int again = 1;
293
294         get_multipath_layout(vecs->mpvec, 1);
295         reply = MALLOC(maxlen);
296
297         while (again) {
298                 if (!reply)
299                         return 1;
300
301                 c = reply;
302                 if (VECTOR_SIZE(vecs->mpvec) > 0)
303                         c += snprint_multipath_header(c, reply + maxlen - c,
304                                                       style);
305
306                 vector_foreach_slot(vecs->mpvec, mpp, i)
307                         c += snprint_multipath(c, reply + maxlen - c,
308                                                style, mpp);
309
310                 again = ((c - reply) == (maxlen - 1));
311
312                 REALLOC_REPLY(reply, again, maxlen);
313         }
314         *r = reply;
315         *len = (int)(c - reply + 1);
316         return 0;
317 }
318
319 int
320 cli_list_maps_fmt (void * v, char ** reply, int * len, void * data)
321 {
322         struct vectors * vecs = (struct vectors *)data;
323         char * fmt = get_keyparam(v, FMT);
324
325         condlog(3, "list maps (operator)");
326
327         return show_maps(reply, len, vecs, fmt);
328 }
329
330 int
331 cli_list_maps (void * v, char ** reply, int * len, void * data)
332 {
333         struct vectors * vecs = (struct vectors *)data;
334
335         condlog(3, "list maps (operator)");
336
337         return show_maps(reply, len, vecs, PRINT_MAP_NAMES);
338 }
339
340 int
341 cli_list_status (void * v, char ** reply, int * len, void * data)
342 {
343         struct vectors * vecs = (struct vectors *)data;
344
345         condlog(3, "list status (operator)");
346
347         return show_status(reply, len, vecs);
348 }
349
350 int
351 cli_list_maps_status (void * v, char ** reply, int * len, void * data)
352 {
353         struct vectors * vecs = (struct vectors *)data;
354
355         condlog(3, "list maps status (operator)");
356
357         return show_maps(reply, len, vecs, PRINT_MAP_STATUS);
358 }
359
360 int
361 cli_list_maps_stats (void * v, char ** reply, int * len, void * data)
362 {
363         struct vectors * vecs = (struct vectors *)data;
364
365         condlog(3, "list maps stats (operator)");
366
367         return show_maps(reply, len, vecs, PRINT_MAP_STATS);
368 }
369
370 int
371 cli_add_path (void * v, char ** reply, int * len, void * data)
372 {
373         struct vectors * vecs = (struct vectors *)data;
374         char * param = get_keyparam(v, PATH);
375         int r;
376
377         condlog(2, "%s: add path (operator)", param);
378
379         if (filter_devnode(conf->blist_devnode, conf->elist_devnode,
380             param) > 0 || (r = ev_add_path(param, vecs)) == 2) {
381                 *reply = strdup("blacklisted\n");
382                 *len = strlen(*reply) + 1;
383                 condlog(2, "%s: path blacklisted", param);
384                 return 0;
385         }
386         return r;
387 }
388
389 int
390 cli_del_path (void * v, char ** reply, int * len, void * data)
391 {
392         struct vectors * vecs = (struct vectors *)data;
393         char * param = get_keyparam(v, PATH);
394
395         condlog(2, "%s: remove path (operator)", param);
396
397         return ev_remove_path(param, vecs);
398 }
399
400 int
401 cli_add_map (void * v, char ** reply, int * len, void * data)
402 {
403         struct vectors * vecs = (struct vectors *)data;
404         char * param = get_keyparam(v, MAP);
405         int major, minor;
406         char dev_path[PATH_SIZE];
407         char *alias;
408         int rc;
409
410         condlog(2, "%s: add map (operator)", param);
411
412         if (filter_wwid(conf->blist_wwid, conf->elist_wwid, param) > 0) {
413                 *reply = strdup("blacklisted\n");
414                 *len = strlen(*reply) + 1;
415                 condlog(2, "%s: map blacklisted", param);
416                 return 0;
417         }
418         minor = dm_get_minor(param);
419         if (minor < 0) {
420                 condlog(2, "%s: not a device mapper table", param);
421                 return 0;
422         }
423         major = dm_get_major(param);
424         if (major < 0) {
425                 condlog(2, "%s: not a device mapper table", param);
426                 return 0;
427         }
428         sprintf(dev_path,"dm-%d", minor);
429         alias = dm_mapname(major, minor);
430         if (!alias) {
431                 condlog(2, "%s: mapname not found for %d:%d",
432                         param, major, minor);
433                 return 0;
434         }
435         rc = ev_add_map(dev_path, alias, vecs);
436         FREE(alias);
437         return rc;
438 }
439
440 int
441 cli_del_map (void * v, char ** reply, int * len, void * data)
442 {
443         struct vectors * vecs = (struct vectors *)data;
444         char * param = get_keyparam(v, MAP);
445         int major, minor;
446         char dev_path[PATH_SIZE];
447         char *alias;
448         int rc;
449
450         condlog(2, "%s: remove map (operator)", param);
451         minor = dm_get_minor(param);
452         if (minor < 0) {
453                 condlog(2, "%s: not a device mapper table", param);
454                 return 0;
455         }
456         major = dm_get_major(param);
457         if (major < 0) {
458                 condlog(2, "%s: not a device mapper table", param);
459                 return 0;
460         }
461         sprintf(dev_path,"dm-%d", minor);
462         alias = dm_mapname(major, minor);
463         if (!alias) {
464                 condlog(2, "%s: mapname not found for %d:%d",
465                         param, major, minor);
466                 return 0;
467         }
468         rc = ev_remove_map(param, alias, minor, vecs);
469         FREE(alias);
470         return rc;
471 }
472
473 int resize_map(struct multipath *mpp, unsigned long long size,
474                struct vectors * vecs)
475 {
476         char params[PARAMS_SIZE] = {0};
477
478         mpp->size = size;
479         update_mpp_paths(mpp, vecs->pathvec);
480         setup_map(mpp, params, PARAMS_SIZE);
481         mpp->action = ACT_RESIZE;
482         if (domap(mpp, params) <= 0) {
483                 condlog(0, "%s: failed to resize map : %s", mpp->alias,
484                         strerror(errno));
485                 return 1;
486         }
487         return 0;
488 }
489
490 int
491 cli_resize(void *v, char **reply, int *len, void *data)
492 {
493         struct vectors * vecs = (struct vectors *)data;
494         char * mapname = get_keyparam(v, MAP);
495         struct multipath *mpp;
496         int minor;
497         unsigned long long size;
498         struct pathgroup *pgp;
499         struct path *pp;
500
501         condlog(2, "%s: resize map (operator)", mapname);
502         if (sscanf(mapname, "dm-%d", &minor) == 1)
503                 mpp = find_mp_by_minor(vecs->mpvec, minor);
504         else
505                 mpp = find_mp_by_alias(vecs->mpvec, mapname);
506
507         if (!mpp) {
508                 condlog(0, "%s: invalid map name. cannot resize", mapname);
509                 return 1;
510         }
511
512         pgp = VECTOR_SLOT(mpp->pg, 0);
513         pp = VECTOR_SLOT(pgp->paths, 0);
514         if (sysfs_get_size(pp->sysdev, &size)) {
515                 condlog(0, "%s: couldn't get size for sysfs. cannot resize",
516                         mapname);
517                 return 1;
518         }
519         if (size == mpp->size) {
520                 condlog(0, "%s: map is still the same size (%llu)", mapname,
521                         mpp->size);
522                 return 0;
523         }
524         condlog(3, "%s old size is %llu, new size is %llu", mapname, mpp->size,
525                 size);
526
527         if (resize_map(mpp, size, vecs) != 0)
528                 return 1;
529
530         dm_lib_release();
531         setup_multipath(vecs, mpp);
532         sync_map_state(mpp);
533
534         return 0;
535 }
536
537 int
538 cli_restore_queueing(void *v, char **reply, int *len, void *data)
539 {
540         struct vectors * vecs = (struct vectors *)data;
541         char * mapname = get_keyparam(v, MAP);
542         struct multipath *mpp;
543         int minor;
544
545         condlog(2, "%s: restore map queueing (operator)", mapname);
546         if (sscanf(mapname, "dm-%d", &minor) == 1)
547                 mpp = find_mp_by_minor(vecs->mpvec, minor);
548         else
549                 mpp = find_mp_by_alias(vecs->mpvec, mapname);
550
551         if (!mpp) {
552                 condlog(0, "%s: invalid map name, cannot restore queueing", mapname);
553                 return 1;
554         }
555
556         if (mpp->no_path_retry != NO_PATH_RETRY_UNDEF &&
557                         mpp->no_path_retry != NO_PATH_RETRY_FAIL) {
558                 dm_queue_if_no_path(mpp->alias, 1);
559                 if (mpp->nr_active > 0)
560                         mpp->retry_tick = 0;
561                 else
562                         mpp->retry_tick = mpp->no_path_retry * conf->checkint;
563         }
564         return 0;
565 }
566
567 int
568 cli_restore_all_queueing(void *v, char **reply, int *len, void *data)
569 {
570         struct vectors * vecs = (struct vectors *)data;
571         struct multipath *mpp;
572         int i;
573
574         condlog(2, "restore queueing (operator)");
575         vector_foreach_slot(vecs->mpvec, mpp, i) {
576                 if (mpp->no_path_retry != NO_PATH_RETRY_UNDEF &&
577                     mpp->no_path_retry != NO_PATH_RETRY_FAIL) {
578                         dm_queue_if_no_path(mpp->alias, 1);
579                         if (mpp->nr_active > 0)
580                                 mpp->retry_tick = 0;
581                         else
582                                 mpp->retry_tick = mpp->no_path_retry * conf->checkint;
583                 }
584         }
585         return 0;
586 }
587
588 int
589 cli_disable_queueing(void *v, char **reply, int *len, void *data)
590 {
591         struct vectors * vecs = (struct vectors *)data;
592         char * mapname = get_keyparam(v, MAP);
593         struct multipath *mpp;
594         int minor;
595
596         condlog(2, "%s: disable map queueing (operator)", mapname);
597         if (sscanf(mapname, "dm-%d", &minor) == 1)
598                 mpp = find_mp_by_minor(vecs->mpvec, minor);
599         else
600                 mpp = find_mp_by_alias(vecs->mpvec, mapname);
601
602         if (!mpp) {
603                 condlog(0, "%s: invalid map name, cannot disable queueing", mapname);
604                 return 1;
605         }
606
607         mpp->retry_tick = 0;
608         dm_queue_if_no_path(mpp->alias, 0);
609         return 0;
610 }
611
612 int
613 cli_disable_all_queueing(void *v, char **reply, int *len, void *data)
614 {
615         struct vectors * vecs = (struct vectors *)data;
616         struct multipath *mpp;
617         int i;
618
619         condlog(2, "disable queueing (operator)");
620         vector_foreach_slot(vecs->mpvec, mpp, i) {
621                 mpp->retry_tick = 0;
622                 dm_queue_if_no_path(mpp->alias, 0);
623         }
624         return 0;
625 }
626
627 int
628 cli_switch_group(void * v, char ** reply, int * len, void * data)
629 {
630         char * mapname = get_keyparam(v, MAP);
631         int groupnum = atoi(get_keyparam(v, GROUP));
632
633         condlog(2, "%s: switch to path group #%i (operator)", mapname, groupnum);
634
635         return dm_switchgroup(mapname, groupnum);
636 }
637
638 int
639 cli_reconfigure(void * v, char ** reply, int * len, void * data)
640 {
641         struct vectors * vecs = (struct vectors *)data;
642
643         condlog(2, "reconfigure (operator)");
644
645         return reconfigure(vecs);
646 }
647
648 int
649 cli_suspend(void * v, char ** reply, int * len, void * data)
650 {
651         struct vectors * vecs = (struct vectors *)data;
652         char * param = get_keyparam(v, MAP);
653         int r = dm_simplecmd_noflush(DM_DEVICE_SUSPEND, param);
654
655         condlog(2, "%s: suspend (operator)", param);
656
657         if (!r) /* error */
658                 return 1;
659
660         struct multipath * mpp = find_mp_by_alias(vecs->mpvec, param);
661
662         if (!mpp)
663                 return 1;
664
665         dm_get_info(param, &mpp->dmi);
666         return 0;
667 }
668
669 int
670 cli_resume(void * v, char ** reply, int * len, void * data)
671 {
672         struct vectors * vecs = (struct vectors *)data;
673         char * param = get_keyparam(v, MAP);
674         int r = dm_simplecmd_noflush(DM_DEVICE_RESUME, param);
675
676         condlog(2, "%s: resume (operator)", param);
677
678         if (!r) /* error */
679                 return 1;
680
681         struct multipath * mpp = find_mp_by_alias(vecs->mpvec, param);
682
683         if (!mpp)
684                 return 1;
685
686         dm_get_info(param, &mpp->dmi);
687         return 0;
688 }
689
690 int
691 cli_reinstate(void * v, char ** reply, int * len, void * data)
692 {
693         struct vectors * vecs = (struct vectors *)data;
694         char * param = get_keyparam(v, PATH);
695         struct path * pp;
696
697         pp = find_path_by_dev(vecs->pathvec, param);
698
699         if (!pp)
700                  pp = find_path_by_devt(vecs->pathvec, param);
701
702         if (!pp || !pp->mpp || !pp->mpp->alias)
703                 return 1;
704
705         condlog(2, "%s: reinstate path %s (operator)",
706                 pp->mpp->alias, pp->dev_t);
707
708         checker_enable(&pp->checker);
709         return dm_reinstate_path(pp->mpp->alias, pp->dev_t);
710 }
711
712 int
713 cli_fail(void * v, char ** reply, int * len, void * data)
714 {
715         struct vectors * vecs = (struct vectors *)data;
716         char * param = get_keyparam(v, PATH);
717         struct path * pp;
718         int r;
719
720         pp = find_path_by_dev(vecs->pathvec, param);
721
722         if (!pp)
723                  pp = find_path_by_devt(vecs->pathvec, param);
724
725         if (!pp || !pp->mpp || !pp->mpp->alias)
726                 return 1;
727
728         condlog(2, "%s: fail path %s (operator)",
729                 pp->mpp->alias, pp->dev_t);
730
731         r = dm_fail_path(pp->mpp->alias, pp->dev_t);
732         /*
733          * Suspend path checking to avoid auto-reinstating the path
734          */
735         if (!r)
736                 checker_disable(&pp->checker);
737         return r;
738 }
739
740 int
741 show_blacklist (char ** r, int * len)
742 {
743         char *c = NULL;
744         char *reply = NULL;
745         unsigned int maxlen = INITIAL_REPLY_LEN;
746         int again = 1;
747
748         reply = MALLOC(maxlen);
749
750         while (again) {
751                 if (!reply)
752                         return 1;
753
754                 c = reply;
755                 c += snprint_blacklist_report(c, maxlen);
756                 again = ((c - reply) == maxlen);
757                 REALLOC_REPLY(reply, again, maxlen);
758         }
759
760         *r = reply;
761         *len = (int)(c - reply + 1);
762
763         return 0;
764 }
765
766 int
767 cli_list_blacklist (void * v, char ** reply, int * len, void * data)
768 {
769         condlog(3, "list blacklist (operator)");
770
771         return show_blacklist(reply, len);
772 }
773
774 int
775 show_devices (char ** r, int * len, struct vectors *vecs)
776 {
777         char *c = NULL;
778         char *reply = NULL;
779         unsigned int maxlen = INITIAL_REPLY_LEN;
780         int again = 1;
781
782         reply = MALLOC(maxlen);
783
784         while (again) {
785                 if (!reply)
786                         return 1;
787
788                 c = reply;
789                 c += snprint_devices(c, maxlen, vecs);
790                 again = ((c - reply) == maxlen);
791                 REALLOC_REPLY(reply, again, maxlen);
792         }
793
794         *r = reply;
795         *len = (int)(c - reply + 1);
796
797         return 0;
798 }
799
800 int
801 cli_list_devices (void * v, char ** reply, int * len, void * data)
802 {
803         struct vectors * vecs = (struct vectors *)data;
804
805         condlog(3, "list devices (operator)");
806
807         return show_devices(reply, len, vecs);
808 }
809
810 int
811 cli_quit (void * v, char ** reply, int * len, void * data)
812 {
813         return 0;
814 }