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
|
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/time.h>
#include <time.h>
#include <ttyrec.h>
#include "common/config.h"
#include "recording.h"
#include "state.h"
#ifdef SESSION_RECORDING
static recorder recorder_handle = NULL;
void clean_recorder(void)
{
ttyrec_w_close(recorder_handle);
recorder_handle = NULL;
}
static char * // returns NULL if error, this char * is to be freed from the calling code
make_filename(void)
{
char * format = LOG_FILENAME_FORMAT;
char * filename = NULL;
unsigned int fname_pos = 0;
unsigned int format_pos = 0;
filename = malloc(LOG_FILENAME_MAX_LEN+1);
size_t format_len = strlen(format);
while (format_pos < format_len + 1 && fname_pos < LOG_FILENAME_MAX_LEN +1) {
if (format[format_pos] == '$') {
format_pos++;
if (format[format_pos] == 'd') {
time_t t;
struct tm * tm;
time(&t);
tm = localtime(&t);
fname_pos += strftime(filename + fname_pos, LOG_FILENAME_MAX_LEN - fname_pos, "%F", tm);
} else if (format[format_pos] == 'h') {
const char * hostname = state_get_ssh_destination();
size_t len = strlen(hostname);
strcpy(filename + fname_pos, hostname);
fname_pos += len;
} else if (format[format_pos] == 'u') {
const char * username = state_get_bastion_username();
size_t len = strlen(username);
strcpy(filename + fname_pos, username);
fname_pos += len;
} else if (format[format_pos] == 'i') {
sprintf(filename + fname_pos, "%llu", state_get_session_id());
fname_pos += strlen(filename + fname_pos);
}
format_pos++;
} else {
filename[fname_pos] = format[format_pos];
if (filename[fname_pos] == '/') { // We create the corresponding directory if it doesn't exist
filename[fname_pos+1] = '\0';
DIR* dir = opendir(filename);
if (dir)
closedir(dir);
else {
int ret = mkdir(filename, LOG_DIRECTORY_MODE);
if (ret != 0) {
fprintf(stderr, "Couldn't create log directory %s : %s\n", filename, strerror( errno ));
}
}
}
format_pos++;
fname_pos++;
}
}
if (filename[fname_pos-1] != '\0') {
fprintf(stderr, "Log file name is too long, check LOG_FILENAME_FORMAT and LOG_FILENAME_MAX_LEN\n");
free(filename);
filename = NULL;
}
return filename;
}
char // returns 0 if ok, 1 otherwise
init_recorder(void)
{
char * filename = make_filename();
if (filename == NULL)
return 1;
struct timeval tm;
if (gettimeofday(&tm, NULL) != 0) {
fprintf(stderr, "OUPS gettimeofday failed!\n");
return 1;
}
recorder_handle = ttyrec_w_open(-1, "ttyrec", filename, &tm);
free(filename);
if (recorder_handle == NULL) {
fprintf(stderr, "Couldn't open the session termrec log file.\n");
return 1;
}
return 0;
}
char // returns 0 if ok, greater than 0 otherwise
record(void* data, size_t len)
{
if(recorder_handle == NULL)
return 0;
struct timeval tm;
if (gettimeofday(&tm, NULL) != 0) {
fprintf(stderr, "OUPS gettimeofday failed!\n");
return 1;
}
if (ttyrec_w_write(recorder_handle, &tm, data, (int) len) == 0) {
fprintf(stderr, "OUPS ttyrec_w_write failed!\n");
return 2;
}
return 0;
}
#endif
|
