@@ -115,64 +115,82 @@ int pman_update_capture_settings(struct capture_settings* settings) {
115
115
116
116
void pman_set_snaplen (uint32_t desired_snaplen ) {
117
117
struct capture_settings settings ;
118
- pman_get_capture_settings (& settings );
118
+ if (pman_get_capture_settings (& settings ) != 0 ) {
119
+ return ;
120
+ }
119
121
settings .snaplen = desired_snaplen ;
120
122
pman_update_capture_settings (& settings );
121
123
}
122
124
123
125
void pman_set_boot_time (uint64_t boot_time ) {
124
126
struct capture_settings settings ;
125
- pman_get_capture_settings (& settings );
127
+ if (pman_get_capture_settings (& settings ) != 0 ) {
128
+ return ;
129
+ }
126
130
settings .boot_time = boot_time ;
127
131
pman_update_capture_settings (& settings );
128
132
}
129
133
130
134
void pman_set_dropping_mode (bool value ) {
131
135
struct capture_settings settings ;
132
- pman_get_capture_settings (& settings );
136
+ if (pman_get_capture_settings (& settings ) != 0 ) {
137
+ return ;
138
+ }
133
139
settings .dropping_mode = value ;
134
140
pman_update_capture_settings (& settings );
135
141
}
136
142
137
143
void pman_set_sampling_ratio (uint32_t value ) {
138
144
struct capture_settings settings ;
139
- pman_get_capture_settings (& settings );
145
+ if (pman_get_capture_settings (& settings ) != 0 ) {
146
+ return ;
147
+ }
140
148
settings .sampling_ratio = value ;
141
149
pman_update_capture_settings (& settings );
142
150
}
143
151
144
152
void pman_set_drop_failed (bool drop_failed ) {
145
153
struct capture_settings settings ;
146
- pman_get_capture_settings (& settings );
154
+ if (pman_get_capture_settings (& settings ) != 0 ) {
155
+ return ;
156
+ }
147
157
settings .drop_failed = drop_failed ;
148
158
pman_update_capture_settings (& settings );
149
159
}
150
160
151
161
void pman_set_do_dynamic_snaplen (bool do_dynamic_snaplen ) {
152
162
struct capture_settings settings ;
153
- pman_get_capture_settings (& settings );
163
+ if (pman_get_capture_settings (& settings ) != 0 ) {
164
+ return ;
165
+ }
154
166
settings .do_dynamic_snaplen = do_dynamic_snaplen ;
155
167
pman_update_capture_settings (& settings );
156
168
}
157
169
158
170
void pman_set_fullcapture_port_range (uint16_t range_start , uint16_t range_end ) {
159
171
struct capture_settings settings ;
160
- pman_get_capture_settings (& settings );
172
+ if (pman_get_capture_settings (& settings ) != 0 ) {
173
+ return ;
174
+ }
161
175
settings .fullcapture_port_range_start = range_start ;
162
176
settings .fullcapture_port_range_end = range_end ;
163
177
pman_update_capture_settings (& settings );
164
178
}
165
179
166
180
void pman_set_statsd_port (uint16_t statsd_port ) {
167
181
struct capture_settings settings ;
168
- pman_get_capture_settings (& settings );
182
+ if (pman_get_capture_settings (& settings ) != 0 ) {
183
+ return ;
184
+ }
169
185
settings .statsd_port = statsd_port ;
170
186
pman_update_capture_settings (& settings );
171
187
}
172
188
173
189
void pman_set_scap_tid (int32_t scap_tid ) {
174
190
struct capture_settings settings ;
175
- pman_get_capture_settings (& settings );
191
+ if (pman_get_capture_settings (& settings ) != 0 ) {
192
+ return ;
193
+ }
176
194
settings .scap_tid = scap_tid ;
177
195
pman_update_capture_settings (& settings );
178
196
}
@@ -198,28 +216,6 @@ void pman_fill_syscall_sampling_table() {
198
216
}
199
217
}
200
218
201
- int pman_init_settings_map () {
202
- char error_message [MAX_ERROR_MESSAGE_LEN ];
203
- struct capture_settings settings = {};
204
- uint32_t key = 0 ;
205
- int fd = bpf_map__fd (g_state .skel -> maps .capture_settings );
206
- if (fd <= 0 ) {
207
- snprintf (error_message , MAX_ERROR_MESSAGE_LEN , "unable to get capture_settings map!" );
208
- pman_print_error ((const char * )error_message );
209
- return errno ;
210
- }
211
-
212
- if (bpf_map_update_elem (fd , & key , & settings , BPF_ANY )) {
213
- snprintf (error_message ,
214
- MAX_ERROR_MESSAGE_LEN ,
215
- "unable to initialize capture_settings map!" );
216
- pman_print_error ((const char * )error_message );
217
- return errno ;
218
- }
219
-
220
- return 0 ;
221
- }
222
-
223
219
void pman_fill_ia32_to_64_table () {
224
220
for (int syscall_id = 0 ; syscall_id < SYSCALL_TABLE_SIZE ; syscall_id ++ ) {
225
221
// Note: we will map all syscalls from the upper limit of the ia32 table
@@ -452,7 +448,8 @@ int pman_prepare_maps_before_loading() {
452
448
453
449
int pman_finalize_maps_after_loading () {
454
450
int err ;
455
- err = pman_init_settings_map ();
451
+ struct capture_settings settings = {};
452
+ err = pman_update_capture_settings (& settings );
456
453
if (err != 0 ) {
457
454
return err ;
458
455
}
0 commit comments