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
|
/**************************************************************************
**
** sngrep - SIP Messages flow viewer
**
** Copyright (C) 2013-2018 Ivan Alonso (Kaian)
** Copyright (C) 2013-2018 Irontec SL. All rights reserved.
**
** This program is free software: you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation, either version 3 of the License, or
** (at your option) any later version.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program. If not, see <http://www.gnu.org/licenses/>.
**
****************************************************************************/
/**
* @file test_input.c
* @author Ivan Alonso [aka Kaian] <kaian@irontec.com>
*
* Basic input injector for sngrep testing
*/
#include "config.h"
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <sys/wait.h>
#ifndef TEST_MAX_DURATION
#define TEST_MAX_DURATION 60
#endif
#ifndef TEST_INITIAL_WAIT
#define TEST_INITIAL_WAIT 1000 * 1200
#endif
#ifndef TEST_KEY_DELAY
#define TEST_KEY_DELAY 3000
#endif
#ifndef TEST_PCAP_INPUT
#define TEST_PCAP_INPUT "aaa.pcap"
#endif
/* keys array needs to be of type "char" */
const char keys[];
int
main()
{
int ppipe[2];
int unused, ret = 0;
int child;
unused = pipe(ppipe);
// Max test duration
alarm(TEST_MAX_DURATION);
if ((child = fork()) < 0) {
fprintf(stderr, "Fatal: unable to fork test.\n");
return 127;
} else if (child == 0) {
// Child process, run sngrep with test pcap
dup2(ppipe[0], STDIN_FILENO);
#ifdef CMAKE_CURRENT_BINARY_DIR
char *argv[] = { CMAKE_CURRENT_BINARY_DIR "/sngrep", "-I", TEST_PCAP_INPUT, 0 };
#else
char *argv[] = { "../src/sngrep", "-I", TEST_PCAP_INPUT, 0 };
#endif
execv(argv[0], argv);
} else {
// Parent process, send keys to child stdin
usleep(TEST_INITIAL_WAIT);
int i;
for (i = 0; keys[i]; i++) {
unused = write(ppipe[1], &keys[i], sizeof(char));
usleep(TEST_KEY_DELAY);
}
unused = wait(&ret);
}
return ret;
}
|
