]> git.ipfire.org Git - ipfire-2.x.git/blame - config/cfgroot/ids-functions.pl
ids.cgi: Fix upstream proxy validation
[ipfire-2.x.git] / config / cfgroot / ids-functions.pl
CommitLineData
8dcebe53
SS
1#!/usr/bin/perl -w
2############################################################################
3# #
4# This file is part of the IPFire Firewall. #
5# #
6# IPFire is free software; you can redistribute it and/or modify #
7# it under the terms of the GNU General Public License as published by #
8# the Free Software Foundation; either version 2 of the License, or #
9# (at your option) any later version. #
10# #
11# IPFire is distributed in the hope that it will be useful, #
12# but WITHOUT ANY WARRANTY; without even the implied warranty of #
13# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
14# GNU General Public License for more details. #
15# #
16# You should have received a copy of the GNU General Public License #
17# along with IPFire; if not, write to the Free Software #
18# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA #
19# #
20# Copyright (C) 2018 IPFire Team <info@ipfire.org>. #
21# #
22############################################################################
23
24package IDS;
25
26require '/var/ipfire/general-functions.pl';
8dcebe53 27
02844177 28# Location where all config and settings files are stored.
164eab66 29our $settingsdir = "${General::swroot}/suricata";
02844177 30
b02e30fd
SS
31# File where the used rulefiles are stored.
32our $used_rulefiles_file = "$settingsdir/suricata-used-rulefiles.yaml";
33
34# File where the addresses of the homenet are stored.
35our $homenet_file = "$settingsdir/suricata-homenet.yaml";
36
37# File which contains the enabled sids.
38our $enabled_sids_file = "$settingsdir/oinkmaster-enabled-sids.conf";
39
40# File which contains the disabled sids.
41our $disabled_sids_file = "$settingsdir/oinkmaster-disabled-sids.conf";
42
43# File which contains wheater the rules should be changed.
44our $modify_sids_file = "$settingsdir/oinkmaster-modify-sids.conf";
45
46# File which stores the configured IPS settings.
47our $ids_settings_file = "$settingsdir/settings";
48
49# File which stores the configured rules-settings.
50our $rules_settings_file = "$settingsdir/rules-settings";
51
52# File which stores the configured settings for whitelisted addresses.
53our $ignored_file = "$settingsdir/ignored";
54
eea2670b 55# Location and name of the tarball which contains the ruleset.
164eab66 56our $rulestarball = "/var/tmp/idsrules.tar.gz";
eea2670b 57
3983aebd 58# File to store any errors, which also will be read and displayed by the wui.
77910792 59our $storederrorfile = "/tmp/ids_storederror";
3983aebd 60
8076deba
SS
61# File to lock the WUI, while the autoupdate script runs.
62our $ids_page_lock_file = "/tmp/ids_page_locked";
63
298ef5ba 64# Location where the rulefiles are stored.
21cab141 65our $rulespath = "/var/lib/suricata";
298ef5ba 66
b02e30fd
SS
67# File which contains the rules to whitelist addresses on suricata.
68our $whitelist_file = "$rulespath/whitelist.rules";
69
bce84f39
SS
70# File which contains a list of all supported ruleset sources.
71# (Sourcefire, Emergingthreads, etc..)
72our $rulesetsourcesfile = "$settingsdir/ruleset-sources";
73
796eea21
SS
74# The pidfile of the IDS.
75our $idspidfile = "/var/run/suricata.pid";
76
5240a809
SS
77# Location of suricatactrl.
78my $suricatactrl = "/usr/local/bin/suricatactrl";
79
80# Array with allowed commands of suricatactrl.
ed06bc81
SS
81my @suricatactrl_cmds = ( 'start', 'stop', 'restart', 'reload', 'fix-rules-dir', 'cron' );
82
83# Array with supported cron intervals.
84my @cron_intervals = ('off', 'daily', 'weekly' );
5240a809 85
b02e30fd
SS
86#
87## Function to check and create all IDS related files, if the does not exist.
88#
89sub check_and_create_filelayout() {
90 # Check if the files exist and if not, create them.
91 unless (-f "$enabled_sids_file") { &create_empty_file($enabled_sids_file); }
92 unless (-f "$disabled_sids_file") { &create_empty_file($disabled_sids_file); }
93 unless (-f "$modify_sids_file") { &create_empty_file($modify_sids_file); }
94 unless (-f "$used_rulefiles_file") { &create_empty_file($used_rulefiles_file); }
95 unless (-f "$ids_settings_file") { &create_empty_file($ids_settings_file); }
96 unless (-f "$rules_settings_file") { &create_empty_file($rules_settings_file); }
97 unless (-f "$ignored_file") { &create_empty_file($ignored_file); }
98 unless (-f "$whitelist_file" ) { &create_empty_file($whitelist_file); }
99}
100
8dcebe53
SS
101#
102## Function for checking if at least 300MB of free disk space are available
103## on the "/var" partition.
104#
105sub checkdiskspace () {
106 # Call diskfree to gather the free disk space of /var.
107 my @df = `/bin/df -B M /var`;
108
109 # Loop through the output.
110 foreach my $line (@df) {
111 # Ignore header line.
112 next if $line =~ m/^Filesystem/;
113
114 # Search for a line with the device information.
115 if ($line =~ m/dev/ ) {
116 # Split the line into single pieces.
117 my @values = split(' ', $line);
118 my ($filesystem, $blocks, $used, $available, $used_perenctage, $mounted_on) = @values;
119
120 # Check if the available disk space is more than 300MB.
121 if ($available < 300) {
434001d0
SS
122 # Log error to syslog.
123 &_log_to_syslog("Not enough free disk space on /var. Only $available MB from 300 MB available.");
8dcebe53 124
434001d0
SS
125 # Exit function and return "1" - False.
126 return 1;
8dcebe53
SS
127 }
128 }
129 }
130
131 # Everything okay, return nothing.
132 return;
133}
134
eea2670b 135#
b749416a 136## This function is responsible for downloading the configured IDS ruleset.
eea2670b 137##
b749416a 138## * At first it obtains from the stored rules settings which ruleset should be downloaded.
eea2670b
SS
139## * The next step is to get the download locations for all available rulesets.
140## * After that, the function will check if an upstream proxy should be used and grab the settings.
141## * The last step will be to generate the final download url, by obtaining the URL for the desired
142## ruleset, add the settings for the upstream proxy and final grab the rules tarball from the server.
143#
144sub downloadruleset {
b749416a
SS
145 # Get rules settings.
146 my %rulessettings=();
147 &General::readhash("$rules_settings_file", \%rulessettings);
eea2670b 148
be52c68a 149 # Check if a ruleset has been configured.
b749416a 150 unless($rulessettings{'RULES'}) {
be52c68a
SS
151 # Log that no ruleset has been configured and abort.
152 &_log_to_syslog("No ruleset source has been configured.");
153
154 # Return "1".
155 return 1;
156 }
157
eea2670b
SS
158 # Get all available ruleset locations.
159 my %rulesetsources=();
bce84f39 160 &General::readhash($rulesetsourcesfile, \%rulesetsources);
eea2670b
SS
161
162 # Read proxysettings.
163 my %proxysettings=();
164 &General::readhash("${General::swroot}/proxy/settings", \%proxysettings);
165
166 # Load required perl module to handle the download.
167 use LWP::UserAgent;
168
169 # Init the download module.
170 my $downloader = LWP::UserAgent->new;
171
172 # Set timeout to 10 seconds.
173 $downloader->timeout(10);
174
175 # Check if an upstream proxy is configured.
176 if ($proxysettings{'UPSTREAM_PROXY'}) {
eea2670b
SS
177 my $proxy_url;
178
40407aee 179 $proxy_url = "http://";
eea2670b 180
40407aee
PM
181 # Check if the proxy requires authentication.
182 if (($proxysettings{'UPSTREAM_USER'}) && ($proxysettings{'UPSTREAM_PASSWORD'})) {
183 $proxy_url .= "$proxysettings{'UPSTREAM_USER'}\:$proxysettings{'UPSTREAM_PASSWORD'}\@";
eea2670b
SS
184 }
185
40407aee
PM
186 # Add proxy server address and port.
187 $proxy_url .= $proxysettings{'UPSTREAM_PROXY'};
188
eea2670b 189 # Setup proxy settings.
6994f001 190 $downloader->proxy(['http', 'https'], $proxy_url);
eea2670b
SS
191 }
192
193 # Grab the right url based on the configured vendor.
b749416a 194 my $url = $rulesetsources{$rulessettings{'RULES'}};
eea2670b
SS
195
196 # Check if the vendor requires an oinkcode and add it if needed.
b749416a 197 $url =~ s/\<oinkcode\>/$rulessettings{'OINKCODE'}/g;
eea2670b
SS
198
199 # Abort if no url could be determined for the vendor.
200 unless ($url) {
434001d0
SS
201 # Log error and abort.
202 &_log_to_syslog("Unable to gather a download URL for the selected ruleset.");
203 return 1;
eea2670b
SS
204 }
205
155b3b56
SS
206 # Variable to store the filesize of the remote object.
207 my $remote_filesize;
96da5803 208
155b3b56
SS
209 # The sourcfire (snort rules) does not allow to send "HEAD" requests, so skip this check
210 # for this webserver.
211 #
212 # Check if the ruleset source contains "snort.org".
213 unless ($url =~ /\.snort\.org/) {
214 # Pass the requrested url to the downloader.
215 my $request = HTTP::Request->new(HEAD => $url);
96da5803 216
155b3b56
SS
217 # Accept the html header.
218 $request->header('Accept' => 'text/html');
96da5803 219
155b3b56
SS
220 # Perform the request and fetch the html header.
221 my $response = $downloader->request($request);
96da5803 222
155b3b56
SS
223 # Check if there was any error.
224 unless ($response->is_success) {
225 # Obtain error.
226 my $error = $response->status_line();
96da5803 227
155b3b56
SS
228 # Log error message.
229 &_log_to_syslog("Unable to download the ruleset. \($error\)");
230
231 # Return "1" - false.
232 return 1;
233 }
96da5803 234
155b3b56
SS
235 # Assign the fetched header object.
236 my $header = $response->headers();
96da5803 237
155b3b56
SS
238 # Grab the remote file size from the object and store it in the
239 # variable.
240 $remote_filesize = $header->content_length;
241 }
96da5803 242
25b6545a
SS
243 # Load perl module to deal with temporary files.
244 use File::Temp;
245
246 # Generate temporay file name, located in "/var/tmp" and with a suffix of ".tar.gz".
247 my $tmp = File::Temp->new( SUFFIX => ".tar.gz", DIR => "/var/tmp/", UNLINK => 0 );
248 my $tmpfile = $tmp->filename();
249
eea2670b
SS
250 # Pass the requested url to the downloader.
251 my $request = HTTP::Request->new(GET => $url);
252
25b6545a
SS
253 # Perform the request and save the output into the tmpfile.
254 my $response = $downloader->request($request, $tmpfile);
eea2670b
SS
255
256 # Check if there was any error.
257 unless ($response->is_success) {
88daf7eb
SS
258 # Obtain error.
259 my $error = $response->content;
260
434001d0 261 # Log error message.
88daf7eb 262 &_log_to_syslog("Unable to download the ruleset. \($error\)");
434001d0
SS
263
264 # Return "1" - false.
265 return 1;
eea2670b
SS
266 }
267
96da5803
SS
268 # Load perl stat module.
269 use File::stat;
270
25b6545a
SS
271 # Perform stat on the tmpfile.
272 my $stat = stat($tmpfile);
96da5803
SS
273
274 # Grab the local filesize of the downloaded tarball.
275 my $local_filesize = $stat->size;
276
277 # Check if both file sizes match.
155b3b56 278 if (($remote_filesize) && ($remote_filesize ne $local_filesize)) {
96da5803
SS
279 # Log error message.
280 &_log_to_syslog("Unable to completely download the ruleset. ");
281 &_log_to_syslog("Only got $local_filesize Bytes instead of $remote_filesize Bytes. ");
282
25b6545a
SS
283 # Delete temporary file.
284 unlink("$tmpfile");
285
96da5803
SS
286 # Return "1" - false.
287 return 1;
288 }
289
25b6545a
SS
290 # Load file copy module, which contains the move() function.
291 use File::Copy;
292
293 # Overwrite existing rules tarball with the new downloaded one.
294 move("$tmpfile", "$rulestarball");
295
eea2670b
SS
296 # If we got here, everything worked fine. Return nothing.
297 return;
298}
8dcebe53 299
25f5cb0d
SS
300#
301## A tiny wrapper function to call the oinkmaster script.
302#
303sub oinkmaster () {
330759d8
SS
304 # Check if the files in rulesdir have the correct permissions.
305 &_check_rulesdir_permissions();
306
883820bd
SS
307 # Cleanup the rules directory before filling it with the new rulest.
308 &_cleanup_rulesdir();
309
0e40e1e7
SS
310 # Load perl module to talk to the kernel syslog.
311 use Sys::Syslog qw(:DEFAULT setlogsock);
312
313 # Establish the connection to the syslog service.
314 openlog('oinkmaster', 'cons,pid', 'user');
315
25f5cb0d 316 # Call oinkmaster to generate ruleset.
26dc79a6 317 open(OINKMASTER, "/usr/local/bin/oinkmaster.pl -s -u file://$rulestarball -C $settingsdir/oinkmaster.conf -o $rulespath 2>&1 |") or die "Could not execute oinkmaster $!\n";
0e40e1e7
SS
318
319 # Log output of oinkmaster to syslog.
320 while(<OINKMASTER>) {
321 # The syslog function works best with an array based input,
322 # so generate one before passing the message details to syslog.
323 my @syslog = ("INFO", "$_");
324
325 # Send the log message.
326 syslog(@syslog);
327 }
328
329 # Close the pipe to oinkmaster process.
330 close(OINKMASTER);
331
332 # Close the log handle.
333 closelog();
25f5cb0d
SS
334}
335
3983aebd
SS
336#
337## Function to do all the logging stuff if the downloading or updating of the ruleset fails.
338#
339sub log_error ($) {
340 my ($error) = @_;
341
342 # Remove any newline.
343 chomp($error);
344
eb5592c1
SS
345 # Call private function to log the error message to syslog.
346 &_log_to_syslog($error);
347
3983aebd
SS
348 # Call private function to write/store the error message in the storederrorfile.
349 &_store_error_message($error);
350}
351
eb5592c1
SS
352#
353## Function to log a given error message to the kernel syslog.
354#
355sub _log_to_syslog ($) {
356 my ($message) = @_;
357
358 # Load perl module to talk to the kernel syslog.
359 use Sys::Syslog qw(:DEFAULT setlogsock);
360
361 # The syslog function works best with an array based input,
362 # so generate one before passing the message details to syslog.
363 my @syslog = ("ERR", "<ERROR> $message");
364
365 # Establish the connection to the syslog service.
366 openlog('oinkmaster', 'cons,pid', 'user');
367
368 # Send the log message.
369 syslog(@syslog);
370
371 # Close the log handle.
372 closelog();
373}
374
3983aebd
SS
375#
376## Private function to write a given error message to the storederror file.
377#
378sub _store_error_message ($) {
379 my ($message) = @_;
380
381 # Remove any newline.
382 chomp($message);
383
384 # Open file for writing.
385 open (ERRORFILE, ">$storederrorfile") or die "Could not write to $storederrorfile. $!\n";
386
387 # Write error to file.
388 print ERRORFILE "$message\n";
389
390 # Close file.
391 close (ERRORFILE);
3c59b1fa
SS
392
393 # Set correct ownership for the file.
394 &set_ownership("$storederrorfile");
3983aebd
SS
395}
396
1cae702c
SS
397#
398## Function to get a list of all available network zones.
399#
400sub get_available_network_zones () {
401 # Get netsettings.
402 my %netsettings = ();
403 &General::readhash("${General::swroot}/ethernet/settings", \%netsettings);
404
405 # Obtain the configuration type from the netsettings hash.
406 my $config_type = $netsettings{'CONFIG_TYPE'};
407
408 # Hash which contains the conversation from the config mode
409 # to the existing network interface names. They are stored like
410 # an array.
411 #
412 # Mode "0" red is a modem and green
413 # Mode "1" red is a netdev and green
414 # Mode "2" red, green and orange
415 # Mode "3" red, green and blue
416 # Mode "4" red, green, blue, orange
417 my %config_type_to_interfaces = (
418 "0" => [ "red", "green" ],
419 "1" => [ "red", "green" ],
420 "2" => [ "red", "green", "orange" ],
421 "3" => [ "red", "green", "blue" ],
422 "4" => [ "red", "green", "blue", "orange" ]
423 );
424
425 # Obtain and dereference the corresponding network interaces based on the read
426 # network config type.
427 my @network_zones = @{ $config_type_to_interfaces{$config_type} };
428
429 # Return them.
430 return @network_zones;
431}
432
796eea21
SS
433#
434## Function to check if the IDS is running.
435#
436sub ids_is_running () {
437 if(-f $idspidfile) {
438 # Open PID file for reading.
439 open(PIDFILE, "$idspidfile") or die "Could not open $idspidfile. $!\n";
440
441 # Grab the process-id.
442 my $pid = <PIDFILE>;
443
444 # Close filehandle.
445 close(PIDFILE);
446
447 # Remove any newline.
448 chomp($pid);
449
450 # Check if a directory for the process-id exists in proc.
451 if(-d "/proc/$pid") {
452 # The IDS daemon is running return the process id.
453 return $pid;
454 }
455 }
456
457 # Return nothing - IDS is not running.
458 return;
459}
460
5240a809
SS
461#
462## Function to call suricatactrl binary with a given command.
463#
464sub call_suricatactrl ($) {
465 # Get called option.
ed06bc81 466 my ($option, $interval) = @_;
5240a809
SS
467
468 # Loop through the array of supported commands and check if
469 # the given one is part of it.
470 foreach my $cmd (@suricatactrl_cmds) {
471 # Skip current command unless the given one has been found.
472 next unless($cmd eq $option);
473
ed06bc81
SS
474 # Check if the given command is "cron".
475 if ($option eq "cron") {
476 # Check if an interval has been given.
477 if ($interval) {
478 # Check if the given interval is valid.
479 foreach my $element (@cron_intervals) {
480 # Skip current element until the given one has been found.
481 next unless($element eq $interval);
482
483 # Call the suricatactrl binary and pass the "cron" command
484 # with the requrested interval.
485 system("$suricatactrl $option $interval &>/dev/null");
486
487 # Return "1" - True.
488 return 1;
489 }
490 }
5240a809 491
ed06bc81
SS
492 # If we got here, the given interval is not supported or none has been given. - Return nothing.
493 return;
494 } else {
495 # Call the suricatactrl binary and pass the requrested
496 # option to it.
497 system("$suricatactrl $option &>/dev/null");
498
499 # Return "1" - True.
500 return 1;
501 }
5240a809
SS
502 }
503
504 # Command not found - return nothing.
505 return;
506}
507
308ba5e7
SS
508#
509## Function to create a new empty file.
510#
511sub create_empty_file($) {
512 my ($file) = @_;
513
514 # Check if the given file exists.
515 if(-e $file) {
516 # Do nothing to prevent from overwriting existing files.
517 return;
518 }
519
520 # Open the file for writing.
521 open(FILE, ">$file") or die "Could not write to $file. $!\n";
522
523 # Close file handle.
524 close(FILE);
525
526 # Return true.
527 return 1;
528}
529
330759d8
SS
530#
531## Private function to check if the file permission of the rulespath are correct.
532## If not, call suricatactrl to fix them.
533#
534sub _check_rulesdir_permissions() {
e568796b
SS
535 # Check if the rulepath main directory is writable.
536 unless (-W $rulespath) {
537 # If not call suricatctrl to fix it.
538 &call_suricatactrl("fix-rules-dir");
539 }
540
330759d8
SS
541 # Open snort rules directory and do a directory listing.
542 opendir(DIR, $rulespath) or die $!;
543 # Loop through the direcory.
544 while (my $file = readdir(DIR)) {
545 # We only want files.
546 next unless (-f "$rulespath/$file");
547
548 # Check if the file is writable by the user.
549 if (-W "$rulespath/$file") {
550 # Everything is okay - go on to the next file.
551 next;
552 } else {
553 # There are wrong permissions, call suricatactrl to fix it.
554 &call_suricatactrl("fix-rules-dir");
555 }
556 }
557}
558
b59cdbee
SS
559#
560## Private function to cleanup the directory which contains
561## the IDS rules, before extracting and modifing the new ruleset.
562#
563sub _cleanup_rulesdir() {
8cf04a16
SS
564 # Open rules directory and do a directory listing.
565 opendir(DIR, $rulespath) or die $!;
566
567 # Loop through the direcory.
568 while (my $file = readdir(DIR)) {
569 # We only want files.
570 next unless (-f "$rulespath/$file");
571
572 # Skip element if it has config as file extension.
573 next if ($file =~ m/\.config$/);
b59cdbee 574
8cf04a16 575 # Delete the current processed file, if not, exit this function
b59cdbee 576 # and return an error message.
1201c1e7 577 unlink("$rulespath/$file") or return "Could not delete $rulespath/$file. $!\n";
b59cdbee
SS
578 }
579
4ce42488 580 # Return nothing;
b59cdbee
SS
581 return;
582}
583
b02e30fd
SS
584#
585## Function to generate the file which contains the home net information.
586#
587sub generate_home_net_file() {
588 my %netsettings;
589
590 # Read-in network settings.
591 &General::readhash("${General::swroot}/ethernet/settings", \%netsettings);
592
593 # Get available network zones.
594 my @network_zones = &get_available_network_zones();
595
596 # Temporary array to store network address and prefix of the configured
597 # networks.
598 my @networks;
599
600 # Loop through the array of available network zones.
601 foreach my $zone (@network_zones) {
bcbc9897
SS
602 # Check if the current processed zone is red.
603 if($zone eq "red") {
604 # Grab the IP-address of the red interface.
605 my $red_address = &get_red_address();
b02e30fd 606
bcbc9897
SS
607 # Check if an address has been obtained.
608 if ($red_address) {
609 # Generate full network string.
610 my $red_network = join("/", $red_address, "32");
b02e30fd 611
bcbc9897
SS
612 # Add the red network to the array of networks.
613 push(@networks, $red_network);
614 }
23c0347a 615
23c0347a
SS
616 # Check if the configured RED_TYPE is static.
617 if ($netsettings{'RED_TYPE'} eq "STATIC") {
618 # Get configured and enabled aliases.
619 my @aliases = &get_aliases();
620
621 # Loop through the array.
622 foreach my $alias (@aliases) {
623 # Add "/32" prefix.
624 my $network = join("/", $alias, "32");
625
626 # Add the generated network to the array of networks.
627 push(@networks, $network);
628 }
629 }
bcbc9897
SS
630 # Process remaining network zones.
631 } else {
632 # Convert current zone name into upper case.
633 $zone = uc($zone);
634
635 # Generate key to access the required data from the netsettings hash.
636 my $zone_netaddress = $zone . "_NETADDRESS";
637 my $zone_netmask = $zone . "_NETMASK";
638
639 # Obtain the settings from the netsettings hash.
640 my $netaddress = $netsettings{$zone_netaddress};
641 my $netmask = $netsettings{$zone_netmask};
642
643 # Convert the subnetmask into prefix notation.
644 my $prefix = &Network::convert_netmask2prefix($netmask);
645
646 # Generate full network string.
647 my $network = join("/", $netaddress,$prefix);
648
649 # Check if the network is valid.
650 if(&Network::check_subnet($network)) {
651 # Add the generated network to the array of networks.
652 push(@networks, $network);
653 }
23c0347a 654 }
b02e30fd
SS
655 }
656
657 # Format home net declaration.
658 my $line = "\"\[";
659
660 # Loop through the array of networks.
661 foreach my $network (@networks) {
662 # Add the network to the line.
663 $line = "$line" . "$network";
664
665 # Check if the current network was the last in the array.
666 if ($network eq $networks[-1]) {
667 # Close the line.
668 $line = "$line" . "\]\"";
669 } else {
670 # Add "," for the next network.
671 $line = "$line" . "\,";
672 }
673 }
674
675 # Open file to store the addresses of the home net.
676 open(FILE, ">$homenet_file") or die "Could not open $homenet_file. $!\n";
677
678 # Print yaml header.
679 print FILE "%YAML 1.1\n";
680 print FILE "---\n\n";
681
682 # Print notice about autogenerated file.
683 print FILE "#Autogenerated file. Any custom changes will be overwritten!\n";
684
685 # Print the generated and required HOME_NET declaration to the file.
686 print FILE "HOME_NET:\t$line\n";
687
688 # Close file handle.
689 close(FILE);
690}
691
692#
693## Function to generate and write the file for used rulefiles.
694#
695sub write_used_rulefiles_file(@) {
696 my @files = @_;
697
698 # Open file for used rulefiles.
699 open (FILE, ">$used_rulefiles_file") or die "Could not write to $used_rulefiles_file. $!\n";
700
701 # Write yaml header to the file.
702 print FILE "%YAML 1.1\n";
703 print FILE "---\n\n";
704
705 # Write header to file.
706 print FILE "#Autogenerated file. Any custom changes will be overwritten!\n";
707
708 # Allways use the whitelist.
709 print FILE " - whitelist.rules\n";
710
711 # Loop through the array of given files.
712 foreach my $file (@files) {
713 # Check if the given filename exists and write it to the file of used rulefiles.
714 if(-f "$rulespath/$file") {
715 print FILE " - $file\n";
716 }
717 }
718
719 # Close file after writing.
720 close(FILE);
721}
722
74cc8f5a
SS
723#
724## Function to generate and write the file for modify the ruleset.
725#
726sub write_modify_sids_file($) {
727 my ($ruleaction) = @_;
728
729 # Open modify sid's file for writing.
2ee51088 730 open(FILE, ">$modify_sids_file") or die "Could not write to $modify_sids_file. $!\n";
74cc8f5a
SS
731
732 # Write file header.
733 print FILE "#Autogenerated file. Any custom changes will be overwritten!\n";
734
735 # Check if the traffic only should be monitored.
736 unless($ruleaction eq "alert") {
737 # Tell oinkmaster to switch all rules from alert to drop.
738 print FILE "modifysid \* \"alert\" \| \"drop\"\n";
739 }
740
741 # Close file handle.
742 close(FILE);
743}
744
04a0d07c
SS
745#
746## Function to gather the version of suricata.
747#
748sub get_suricata_version($) {
749 my ($format) = @_;
750
751 # Execute piped suricata command and return the version information.
752 open(SURICATA, "suricata -V |") or die "Couldn't execute program: $!";
753
754 # Grab and store the output of the piped program.
755 my $version_string = <SURICATA>;
756
757 # Close pipe.
758 close(SURICATA);
759
760 # Remove newlines.
761 chomp($version_string);
762
763 # Grab the version from the version string.
764 $version_string =~ /([0-9]+([.][0-9]+)+)/;
765
766 # Splitt the version into single chunks.
767 my ($major_ver, $minor_ver, $patchlevel) = split(/\./, $1);
768
769 # Check and return the requested version sheme.
770 if ($format eq "major") {
771 # Return the full version.
772 return "$major_ver";
773 } elsif ($format eq "minor") {
774 # Return the major and minor part.
775 return "$major_ver.$minor_ver";
776 } else {
777 # Return the full version string.
778 return "$major_ver.$minor_ver.$patchlevel";
779 }
780}
781
9283e9b9
SS
782#
783## Function to generate the rules file with whitelisted addresses.
784#
785sub generate_ignore_file() {
786 my %ignored = ();
787
788 # SID range 1000000-1999999 Reserved for Local Use
789 # Put your custom rules in this range to avoid conflicts
790 my $sid = 1500000;
791
792 # Read-in ignoredfile.
793 &General::readhasharray($IDS::ignored_file, \%ignored);
794
795 # Open ignorefile for writing.
796 open(FILE, ">$IDS::whitelist_file") or die "Could not write to $IDS::whitelist_file. $!\n";
797
798 # Config file header.
799 print FILE "# Autogenerated file.\n";
800 print FILE "# All user modifications will be overwritten.\n\n";
801
802 # Add all user defined addresses to the whitelist.
803 #
804 # Check if the hash contains any elements.
805 if (keys (%ignored)) {
806 # Loop through the entire hash and write the host/network
807 # and remark to the ignore file.
808 while ( (my $key) = each %ignored) {
809 my $address = $ignored{$key}[0];
810 my $remark = $ignored{$key}[1];
811 my $status = $ignored{$key}[2];
812
813 # Check if the status of the entry is "enabled".
814 if ($status eq "enabled") {
815 # Check if the address/network is valid.
816 if ((&General::validip($address)) || (&General::validipandmask($address))) {
817 # Write rule line to the file to pass any traffic from this IP
818 print FILE "pass ip $address any -> any any (msg:\"pass all traffic from/to $address\"\; sid:$sid\;)\n";
819
820 # Increment sid.
821 $sid++;
822 }
823 }
824 }
825 }
826
827 close(FILE);
828}
829
1fedede6
SS
830#
831## Function to set correct ownership for single files and directories.
832#
833
834sub set_ownership($) {
835 my ($target) = @_;
836
837 # User and group of the WUI.
838 my $uname = "nobody";
839 my $grname = "nobody";
840
841 # The chown function implemented in perl requies the user and group as nummeric id's.
842 my $uid = getpwnam($uname);
843 my $gid = getgrnam($grname);
844
845 # Check if the given target exists.
846 unless ($target) {
847 # Stop the script and print error message.
848 die "The $target does not exist. Cannot change the ownership!\n";
849 }
850
851 # Check weather the target is a file or directory.
852 if (-f $target) {
853 # Change ownership ot the single file.
854 chown($uid, $gid, "$target");
855 } elsif (-d $target) {
856 # Do a directory listing.
857 opendir(DIR, $target) or die $!;
858 # Loop through the direcory.
859 while (my $file = readdir(DIR)) {
860
861 # We only want files.
862 next unless (-f "$target/$file");
863
864 # Set correct ownership for the files.
865 chown($uid, $gid, "$target/$file");
866 }
867
868 closedir(DIR);
869
870 # Change ownership of the directory.
871 chown($uid, $gid, "$target");
872 }
873}
77c31301
SS
874
875#
876## Function to read-in the aliases file and returns all configured and enabled aliases.
877#
878sub get_aliases() {
879 # Location of the aliases file.
880 my $aliases_file = "${General::swroot}/ethernet/aliases";
881
882 # Array to store the aliases.
883 my @aliases;
884
885 # Check if the file is empty.
886 if (-z $aliases_file) {
887 # Abort nothing to do.
888 return;
889 }
890
891 # Open the aliases file.
892 open(ALIASES, $aliases_file) or die "Could not open $aliases_file. $!\n";
893
894 # Loop through the file content.
895 while (my $line = <ALIASES>) {
896 # Remove newlines.
897 chomp($line);
898
899 # Splitt line content into single chunks.
900 my ($address, $state, $remark) = split(/\,/, $line);
901
902 # Check if the state of the current processed alias is "on".
903 if ($state eq "on") {
904 # Check if the address is valid.
905 if(&Network::check_ip_address($address)) {
906 # Add the alias to the array of aliases.
907 push(@aliases, $address);
908 }
909 }
910 }
911
912 # Close file handle.
913 close(ALIASES);
914
915 # Return the array.
916 return @aliases;
917}
918
de8e1e5b
SS
919#
920## Function to grab the current assigned IP-address on red.
921#
922sub get_red_address() {
923 # File, which contains the current IP-address of the red interface.
924 my $file = "${General::swroot}/red/local-ipaddress";
925
926 # Check if the file exists.
927 if (-e $file) {
928 # Open the given file.
929 open(FILE, "$file") or die "Could not open $file.";
930
931 # Obtain the address from the first line of the file.
932 my $address = <FILE>;
933
934 # Close filehandle
935 close(FILE);
936
937 # Remove newlines.
938 chomp $address;
939
940 # Check if the grabbed address is valid.
941 if (&General::validip($address)) {
942 # Return the address.
943 return $address;
944 }
945 }
946
947 # Return nothing.
948 return;
949}
77c31301 950
8076deba
SS
951#
952## Function to write the lock file for locking the WUI, while
953## the autoupdate script runs.
954#
955sub lock_ids_page() {
956 # Call subfunction to create the file.
957 &create_empty_file($ids_page_lock_file);
958}
959
960#
961## Function to release the lock of the WUI, again.
962#
963sub unlock_ids_page() {
964 # Delete lock file.
965 unlink($ids_page_lock_file);
966}
967
8dcebe53 9681;