PACT/src/cli.c at main · MoatLab/PACT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
/* SPDX-License-Identifier: MIT */
/* Copyright (c) 2026 MoatLab, Virginia Tech. */

/* cli.c — argv parsing for PACT runtime. See `pact --help` for the flag list. */

#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "cli.h"
#include "error.h" /* set_log_level */
#include "usage.h"
#include "sighandler.h"

int pact_parse_command_line_args(int argc, char *argv[], pact_config_t *config)
{
    for (int i = 1; i < argc; i++) {
        if (strcmp(argv[i], "--sampling-interval") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --sampling-interval requires an argument\n");
                return -1;
            }
            config->sampling_interval_ms = atoi(argv[i]);
        } else if (strcmp(argv[i], "--adaptive-interval") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --adaptive-interval requires an argument\n");
                return -1;
            }
            config->adaptive_interval_ms = atoi(argv[i]);
        } else if (strcmp(argv[i], "--stats-interval") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --stats-interval requires an argument\n");
                return -1;
            }
            config->stats_interval_ms = atoi(argv[i]);
        } else if (strcmp(argv[i], "--max-migrations-per-cycle") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --max-migrations-per-cycle requires an argument\n");
                return -1;
            }
            config->max_migrations_per_cycle = atoi(argv[i]);
        } else if (strcmp(argv[i], "--bin-count") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --bin-count requires an argument\n");
                return -1;
            }
            config->bin_count = atoi(argv[i]);
        } else if (strcmp(argv[i], "--bin-width") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --bin-width requires an argument\n");
                return -1;
            }
            config->bin_width = atof(argv[i]);
        } else if (strcmp(argv[i], "--cooling-alpha") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --cooling-alpha requires an argument\n");
                return -1;
            }
            config->cooling_alpha = atof(argv[i]);
            if (config->cooling_alpha < 0.0 || config->cooling_alpha > 1.0) {
                fprintf(stderr, "Error: --cooling-alpha must be in [0, 1]\n");
                return -1;
            }
        } else if (strcmp(argv[i], "--cooling-trigger-samples") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --cooling-trigger-samples requires an argument\n");
                return -1;
            }
            config->cooling_trigger_samples = strtoull(argv[i], NULL, 10);
        } else if (strcmp(argv[i], "-l") == 0 || strcmp(argv[i], "--logging") == 0) {
            config->enable_logging = true;
            if (i + 1 < argc && argv[i + 1][0] != '-') {
                i++;
                if (strcmp(argv[i], "csv") == 0 || strcmp(argv[i], "json") == 0) {
                    strncpy(config->log_format, argv[i], sizeof(config->log_format) - 1);
                    config->log_format[sizeof(config->log_format) - 1] = '\0';
                } else {
                    fprintf(stderr, "Warning: Unknown log format '%s', using 'csv'\n", argv[i]);
                }
            }
        } else if (strcmp(argv[i], "--log-file") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --log-file requires an argument\n");
                return -1;
            }
            strncpy(config->log_file, argv[i], sizeof(config->log_file) - 1);
            config->log_file[sizeof(config->log_file) - 1] = '\0';
        } else if (strcmp(argv[i], "--pebs-period") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --pebs-period requires an argument\n");
                return -1;
            }
            config->pebs_period = atoll(argv[i]);
            if (config->pebs_period < 1 || config->pebs_period > 1000000) {
                fprintf(stderr, "Error: PEBS period must be between 1 and 1000000\n");
                return -1;
            }
        } else if (strcmp(argv[i], "--workload") == 0) {
            /* --workload PID is required. CPU pinning is external
             * (taskset). Passing --workload more than once overwrites
             * the prior value with a warning. */
            if (++i >= argc) {
                fprintf(stderr, "Error: --workload requires a PID\n");
                return -1;
            }
            pid_t pid = atoi(argv[i]);
            if (pid <= 0) {
                fprintf(stderr, "Error: Invalid PID: %s\n", argv[i]);
                return -1;
            }
            if (config->target_pid > 0) {
                fprintf(stderr,
                        "Warning: --workload specified more than once; "
                        "overwriting PID %d with %d\n",
                        config->target_pid, pid);
            }
            config->target_pid = pid;
            snprintf(config->workload_name, sizeof(config->workload_name), "workload");
            printf("Workload: PID %d\n", pid);
        } else if (strcmp(argv[i], "--log-level") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --log-level requires an argument\n");
                return -1;
            }
            int log_level = atoi(argv[i]);
            if (log_level < 0 || log_level > 3) {
                fprintf(stderr, "Error: Log level must be between 0 and 3\n");
                return -1;
            }
            set_log_level(log_level);
            printf("Log level set to %d\n", log_level);
        } else if (strcmp(argv[i], "--monitor-cpu") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --monitor-cpu requires an argument\n");
                return -1;
            }
            config->monitor_cpu = atoi(argv[i]);
            printf("Monitor CPU affinity set to %d\n", config->monitor_cpu);
        } else if (strcmp(argv[i], "--migration-cpu") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --migration-cpu requires an argument\n");
                return -1;
            }
            config->migration_cpu = atoi(argv[i]);
            printf("Migration CPU affinity set to %d\n", config->migration_cpu);
        } else if (strcmp(argv[i], "--crash-marker") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --crash-marker requires a path argument\n");
                return -1;
            }
            pact_signal_set_crash_marker_path(argv[i]);
            printf("Crash marker path: %s\n", argv[i]);
#ifdef PACT_CGROUP_SUPPORT
        } else if (strcmp(argv[i], "--cgroup-name") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --cgroup-name requires an argument\n");
                return -1;
            }
            strncpy(config->cgroup_name, argv[i], sizeof(config->cgroup_name) - 1);
            config->cgroup_name[sizeof(config->cgroup_name) - 1] = '\0';
            printf("Cgroup name: %s\n", config->cgroup_name);
        } else if (strcmp(argv[i], "--cgroup-dir") == 0) {
            if (++i >= argc) {
                fprintf(stderr, "Error: --cgroup-dir requires an argument\n");
                return -1;
            }
            strncpy(config->cgroup_dir, argv[i], sizeof(config->cgroup_dir) - 1);
            config->cgroup_dir[sizeof(config->cgroup_dir) - 1] = '\0';
            printf("Cgroup directory: %s\n", config->cgroup_dir);
#endif
        } else if (strcmp(argv[i], "--pac-pool-max") == 0) {
            if (++i >= argc) {
                fprintf(stderr,
                        "Error: --pac-pool-max requires an argument (entries; 0=unlimited)\n");
                return -1;
            }
            config->pac_pool_max = (size_t)strtoull(argv[i], NULL, 10);
            printf("PAC metadata pool cap: %zu entries (~%zu MB at 128B/entry)\n",
                   config->pac_pool_max, config->pac_pool_max * 128 / (1024 * 1024));
        } else if (strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "--help") == 0) {
            pact_print_usage(argv[0]);
            return 1;
        } else if (strcmp(argv[i], "-V") == 0 || strcmp(argv[i], "--version") == 0) {
            pact_print_version();
            return 1;
        } else {
            fprintf(stderr, "Error: Unknown option: %s\n", argv[i]);
            pact_print_usage(argv[0]);
            return -1;
        }
    }

    if (config->target_pid <= 0) {
        pact_print_usage(argv[0]);
        fprintf(stderr, "\nError: --workload PID is required.\n");
        return -1;
    }
    return 0;
}