Description:
change memory resolution in report to kbytes
Commit status:
[Not Reviewed]
References:
Diff options:
Comments:
0 Commit comments
0 Inline Comments
Unresolved TODOs:
There are no unresolved TODOs
r164:447fdbbeac2c - - 3 files changed: 4 inserted, 4 deleted
@@ -1,254 +1,254 | |||
|
1 | 1 | # |
|
2 | 2 | # This part contains various test_request helpers for interfacing |
|
3 | 3 | # with Grader::Engine. There are TestRequestRoomMaker and |
|
4 | 4 | # TestRequestReporter. |
|
5 | 5 | |
|
6 | 6 | module Grader |
|
7 | 7 | |
|
8 | 8 | def self.link_or_copy(src, des) |
|
9 | 9 | begin |
|
10 | 10 | FileUtils.ln_s(src, des) |
|
11 | 11 | rescue NotImplementedError |
|
12 | 12 | FileUtils.cp(src,des) |
|
13 | 13 | end |
|
14 | 14 | end |
|
15 | 15 | |
|
16 | 16 | def self.call_and_log(error_message) |
|
17 | 17 | begin |
|
18 | 18 | yield |
|
19 | 19 | rescue |
|
20 | 20 | msg = "ERROR: #{error_message}" |
|
21 | 21 | raise msg |
|
22 | 22 | end |
|
23 | 23 | end |
|
24 | 24 | |
|
25 | 25 | # |
|
26 | 26 | # A TestRequestRoomMaker is a helper object for Engine |
|
27 | 27 | # - finds grading room: in user_result_dir/(user)/test_request/ ... |
|
28 | 28 | # - prepare problem configuration for grading --- basically it copy |
|
29 | 29 | # all config files, and copy user's input into the testcase |
|
30 | 30 | # directory. First, it finds the template from problem template |
|
31 | 31 | # directory; if it can't find a template, it'll use the template |
|
32 | 32 | # from default template. |
|
33 | 33 | class TestRequestRoomMaker |
|
34 | 34 | def initialize |
|
35 | 35 | @config = Grader::Configuration.get_instance |
|
36 | 36 | end |
|
37 | 37 | |
|
38 | 38 | def produce_grading_room(test_request) |
|
39 | 39 | grading_room = grading_room_dir(test_request) |
|
40 | 40 | FileUtils.mkdir_p(grading_room) |
|
41 | 41 | |
|
42 | 42 | # |
|
43 | 43 | # Also copy additional submitted file to this directory as well. |
|
44 | 44 | # The program would see this file only if it is copied |
|
45 | 45 | # to the sandbox directory later. The run script should do it. |
|
46 | 46 | # |
|
47 | 47 | if FileTest.exists?("#{test_request.input_file_name}.files") |
|
48 | 48 | FileUtils.cp_r("#{test_request.input_file_name}.files/.", |
|
49 | 49 | "#{grading_room}") |
|
50 | 50 | end |
|
51 | 51 | |
|
52 | 52 | grading_room |
|
53 | 53 | end |
|
54 | 54 | |
|
55 | 55 | def find_problem_home(test_request) |
|
56 | 56 | problem_name = test_request.problem_name |
|
57 | 57 | |
|
58 | 58 | template_dir = "#{@config.test_request_problem_templates_dir}/" + problem_name |
|
59 | 59 | |
|
60 | 60 | raise "Test Request: error template not found" if !File.exists?(template_dir) |
|
61 | 61 | |
|
62 | 62 | problem_home = problem_home_dir(test_request) |
|
63 | 63 | FileUtils.mkdir_p(problem_home) |
|
64 | 64 | |
|
65 | 65 | copy_problem_template(template_dir,problem_home) |
|
66 | 66 | link_input_file(test_request,problem_home) |
|
67 | 67 | |
|
68 | 68 | problem_home |
|
69 | 69 | end |
|
70 | 70 | |
|
71 | 71 | def save_source(test_request,source_name) |
|
72 | 72 | dir = self.produce_grading_room(test_request) |
|
73 | 73 | submission = test_request.submission |
|
74 | 74 | f = File.open("#{dir}/#{source_name}","w") |
|
75 | 75 | f.write(submission.source) |
|
76 | 76 | f.close |
|
77 | 77 | end |
|
78 | 78 | |
|
79 | 79 | def clean_up(test_request) |
|
80 | 80 | problem_home = problem_home_dir(test_request) |
|
81 | 81 | remove_data_files(problem_home) |
|
82 | 82 | end |
|
83 | 83 | |
|
84 | 84 | protected |
|
85 | 85 | def grading_room_dir(test_request) |
|
86 | 86 | problem_name = test_request.problem_name |
|
87 | 87 | user = test_request.user |
|
88 | 88 | grading_room = "#{@config.user_result_dir}" + |
|
89 | 89 | "/#{user.login}/test_request" + |
|
90 | 90 | "/#{problem_name}/#{test_request.id}" |
|
91 | 91 | grading_room |
|
92 | 92 | end |
|
93 | 93 | |
|
94 | 94 | def problem_home_dir(test_request) |
|
95 | 95 | problem_name = test_request.problem_name |
|
96 | 96 | user = test_request.user |
|
97 | 97 | "#{@config.user_result_dir}" + |
|
98 | 98 | "/#{user.login}/test_request/#{problem_name}" |
|
99 | 99 | end |
|
100 | 100 | |
|
101 | 101 | def copy_problem_template(template_dir,problem_home) |
|
102 | 102 | Grader::call_and_log("Test Request: cannot copy problem template") { |
|
103 | 103 | FileUtils.cp_r("#{template_dir}/.","#{problem_home}") |
|
104 | 104 | } |
|
105 | 105 | end |
|
106 | 106 | |
|
107 | 107 | def link_input_file(test_request, problem_home) |
|
108 | 108 | input_fname = "#{test_request.input_file_name}" |
|
109 | 109 | if !File.exists?(input_fname) |
|
110 | 110 | raise "Test Request: input file not found." |
|
111 | 111 | end |
|
112 | 112 | |
|
113 | 113 | input_fname_problem_home = "#{problem_home}/test_cases/1/input-1.txt" |
|
114 | 114 | if File.exists?(input_fname_problem_home) |
|
115 | 115 | FileUtils.rm([input_fname_problem_home], :force => true) |
|
116 | 116 | end |
|
117 | 117 | |
|
118 | 118 | Grader::link_or_copy("#{input_fname}", "#{input_fname_problem_home}") |
|
119 | 119 | end |
|
120 | 120 | |
|
121 | 121 | def remove_data_files(problem_home) |
|
122 | 122 | if File.exists?("#{problem_home}/test_cases/1/input-1.txt") |
|
123 | 123 | Grader::call_and_log("Test Request: cannot remove data files") { |
|
124 | 124 | FileUtils.rm Dir.glob("#{problem_home}/test_cases/1/*") |
|
125 | 125 | } |
|
126 | 126 | end |
|
127 | 127 | end |
|
128 | 128 | |
|
129 | 129 | end |
|
130 | 130 | |
|
131 | 131 | class TestRequestReporter |
|
132 | 132 | def initialize |
|
133 | 133 | @config = Grader::Configuration.get_instance |
|
134 | 134 | end |
|
135 | 135 | |
|
136 | 136 | def report(test_request,test_result_dir) |
|
137 | 137 | save_result(test_request,read_result(test_result_dir)) |
|
138 | 138 | end |
|
139 | 139 | |
|
140 | 140 | def report_error(test_request, msg) |
|
141 | 141 | save_result(test_request, {:running_stat => { |
|
142 | 142 | :msg => "#{msg}", |
|
143 | 143 | :running_time => nil, |
|
144 | 144 | :exit_status => "Some error occured. Program did not run", |
|
145 | 145 | :memory_usage => nil |
|
146 | 146 | }}) |
|
147 | 147 | end |
|
148 | 148 | |
|
149 | 149 | protected |
|
150 | 150 | def read_result(test_result_dir) |
|
151 | 151 | # TODO: |
|
152 | 152 | cmp_msg_fname = "#{test_result_dir}/compiler_message" |
|
153 | 153 | cmp_file = File.open(cmp_msg_fname) |
|
154 | 154 | cmp_msg = cmp_file.read |
|
155 | 155 | cmp_file.close |
|
156 | 156 | |
|
157 | 157 | result_file_name = "#{test_result_dir}/1/result" |
|
158 | 158 | |
|
159 | 159 | if File.exists?(result_file_name) |
|
160 | 160 | output_file_name = "#{test_result_dir}/1/output.txt" |
|
161 | 161 | results = [] |
|
162 | 162 | File.open("#{test_result_dir}/1/result") do |f| |
|
163 | 163 | results = f.readlines |
|
164 | 164 | end |
|
165 | 165 | stat = extract_running_stat(results) |
|
166 | 166 | |
|
167 | 167 | return { |
|
168 | 168 | :output_file_name => output_file_name, |
|
169 | 169 | :running_stat => stat, |
|
170 | 170 | :comment => "", |
|
171 | 171 | :cmp_msg => cmp_msg} |
|
172 | 172 | else |
|
173 | 173 | return { |
|
174 | 174 | :running_stat => nil, |
|
175 | 175 | :comment => "Compilation error", |
|
176 | 176 | :cmp_msg => cmp_msg} |
|
177 | 177 | end |
|
178 | 178 | end |
|
179 | 179 | |
|
180 | 180 | def extract_running_stat(results) |
|
181 | 181 | running_stat_line = results[-1] |
|
182 | 182 | |
|
183 | 183 | # extract exit status line |
|
184 | 184 | run_stat = "" |
|
185 | 185 | if !(/[Cc]orrect/.match(results[0])) |
|
186 | 186 | run_stat = results[0].chomp |
|
187 | 187 | else |
|
188 | 188 | run_stat = 'Program exited normally' |
|
189 | 189 | end |
|
190 | 190 | |
|
191 | 191 | # extract running time |
|
192 | 192 | if res = /r(.*)u(.*)s/.match(running_stat_line) |
|
193 | 193 | seconds = (res[1].to_f + res[2].to_f) |
|
194 | 194 | time_stat = "Time used: #{seconds} sec." |
|
195 | 195 | else |
|
196 | 196 | seconds = nil |
|
197 | 197 | time_stat = "Time used: n/a sec." |
|
198 | 198 | end |
|
199 | 199 | |
|
200 | 200 | # extract memory usage |
|
201 |
- if res = /s(.*) |
|
|
201 | + if res = /s(.*)kbytes/.match(running_stat_line) | |
|
202 | 202 | memory_used = res[1].to_i |
|
203 | 203 | else |
|
204 | 204 | memory_used = -1 |
|
205 | 205 | end |
|
206 | 206 | |
|
207 | 207 | return { |
|
208 | 208 | :msg => "#{run_stat}\n#{time_stat}", |
|
209 | 209 | :running_time => seconds, |
|
210 | 210 | :exit_status => run_stat, |
|
211 | 211 | :memory_usage => memory_used |
|
212 | 212 | } |
|
213 | 213 | end |
|
214 | 214 | |
|
215 | 215 | def save_result(test_request,result) |
|
216 | 216 | if result[:output_file_name]!=nil |
|
217 | 217 | test_request.output_file_name = link_output_file(test_request, |
|
218 | 218 | result[:output_file_name]) |
|
219 | 219 | end |
|
220 | 220 | test_request.graded_at = Time.now |
|
221 | 221 | test_request.compiler_message = (result[:cmp_msg] or '') |
|
222 | 222 | test_request.grader_comment = (result[:comment] or '') |
|
223 | 223 | if result[:running_stat]!=nil |
|
224 | 224 | test_request.running_stat = (result[:running_stat][:msg] or '') |
|
225 | 225 | test_request.running_time = (result[:running_stat][:running_time] or nil) |
|
226 | 226 | test_request.exit_status = result[:running_stat][:exit_status] |
|
227 | 227 | test_request.memory_usage = result[:running_stat][:memory_usage] |
|
228 | 228 | else |
|
229 | 229 | test_request.running_stat = '' |
|
230 | 230 | end |
|
231 | 231 | test_request.save |
|
232 | 232 | end |
|
233 | 233 | |
|
234 | 234 | protected |
|
235 | 235 | def link_output_file(test_request, fname) |
|
236 | 236 | target_file_name = random_output_file_name(test_request.user, |
|
237 | 237 | test_request.problem) |
|
238 | 238 | FileUtils.mkdir_p(File.dirname(target_file_name)) |
|
239 | 239 | Grader::link_or_copy("#{fname}", "#{target_file_name}") |
|
240 | 240 | return target_file_name |
|
241 | 241 | end |
|
242 | 242 | |
|
243 | 243 | def random_output_file_name(user,problem) |
|
244 | 244 | problem_name = TestRequest.name_of(problem) |
|
245 | 245 | begin |
|
246 | 246 | tmpname = "#{@config.test_request_output_base_dir}" + |
|
247 | 247 | "/#{user.login}/#{problem_name}/#{rand(10000)}" |
|
248 | 248 | end while File.exists?(tmpname) |
|
249 | 249 | tmpname |
|
250 | 250 | end |
|
251 | 251 | |
|
252 | 252 | end |
|
253 | 253 | |
|
254 | 254 | end |
@@ -1,980 +1,980 | |||
|
1 | 1 | /* |
|
2 | 2 | * A Simple Sandbox for Moe |
|
3 | 3 | * |
|
4 | 4 | * (c) 2001--2010 Martin Mares <mj@ucw.cz> |
|
5 | 5 | */ |
|
6 | 6 | |
|
7 | 7 | #define _LARGEFILE64_SOURCE |
|
8 | 8 | #define _GNU_SOURCE |
|
9 | 9 | |
|
10 | 10 | /* Generated automatically by ./configure, please don't touch manually. */ |
|
11 | 11 | #define CONFIG_BOX_KERNEL_AMD64 1 |
|
12 | 12 | #define CONFIG_BOX_USER_AMD64 1 |
|
13 | 13 | #define CONFIG_DIR "cf" |
|
14 | 14 | #define CONFIG_DIRECT_IO 1 |
|
15 | 15 | #define CONFIG_ISOLATE_BOX_DIR "/tmp/box" |
|
16 | 16 | #define CONFIG_ISOLATE_CGROUP_ROOT "/sys/fs/cgroup" |
|
17 | 17 | #define CONFIG_ISOLATE_FIRST_GID 60000 |
|
18 | 18 | #define CONFIG_ISOLATE_FIRST_UID 60000 |
|
19 | 19 | #define CONFIG_ISOLATE_NUM_BOXES 100 |
|
20 | 20 | #define CONFIG_LARGE_FILES 1 |
|
21 | 21 | #define CONFIG_LFS 1 |
|
22 | 22 | #define CONFIG_LINUX 1 |
|
23 | 23 | #define CONFIG_LOCAL 1 |
|
24 | 24 | #define CONFIG_UCW_PARTMAP_IS_MMAP 1 |
|
25 | 25 | #define CONFIG_UCW_PERL 1 |
|
26 | 26 | #define CONFIG_UCW_POOL_IS_MMAP 1 |
|
27 | 27 | #define CONFIG_UCW_RADIX_SORTER_BITS 10 |
|
28 | 28 | #define CONFIG_UCW_SHELL_UTILS 1 |
|
29 | 29 | #define CPU_64BIT_POINTERS 1 |
|
30 | 30 | #define CPU_ALLOW_UNALIGNED 1 |
|
31 | 31 | #define CPU_AMD64 1 |
|
32 | 32 | #define CPU_ARCH "default" |
|
33 | 33 | #define CPU_LITTLE_ENDIAN 1 |
|
34 | 34 | #define CPU_PAGE_SIZE 4096 |
|
35 | 35 | #define CPU_STRUCT_ALIGN 8 |
|
36 | 36 | #define CWARNS_OFF " -Wno-pointer-sign" |
|
37 | 37 | #define HAVE_ASCII_DOC "none" |
|
38 | 38 | #define INSTALL_BIN_DIR "bin" |
|
39 | 39 | #define INSTALL_CONFIG_DIR "cf" |
|
40 | 40 | #define INSTALL_DOC_DIR "share/doc" |
|
41 | 41 | #define INSTALL_INCLUDE_DIR "include" |
|
42 | 42 | #define INSTALL_LIB_DIR "lib" |
|
43 | 43 | #define INSTALL_LOG_DIR "log" |
|
44 | 44 | #define INSTALL_MAN_DIR "share/man" |
|
45 | 45 | #define INSTALL_PERL_DIR "lib/perl5" |
|
46 | 46 | #define INSTALL_PKGCONFIG_DIR "lib/pkgconfig" |
|
47 | 47 | #define INSTALL_PREFIX |
|
48 | 48 | #define INSTALL_RUN_DIR "run" |
|
49 | 49 | #define INSTALL_SBIN_DIR "sbin" |
|
50 | 50 | #define INSTALL_SHARE_DIR "share" |
|
51 | 51 | #define INSTALL_STATE_DIR "lib" |
|
52 | 52 | #define INSTALL_USR_PREFIX |
|
53 | 53 | #define INSTALL_VAR_PREFIX |
|
54 | 54 | #define SHERLOCK_VERSION "3.99.2" |
|
55 | 55 | #define SHERLOCK_VERSION_CODE 3099002 |
|
56 | 56 | #define SONAME_PREFIX "lib/" |
|
57 | 57 | #define UCW_VERSION "3.99.2" |
|
58 | 58 | #define UCW_VERSION_CODE 3099002 |
|
59 | 59 | |
|
60 | 60 | #include <errno.h> |
|
61 | 61 | #include <stdio.h> |
|
62 | 62 | #include <fcntl.h> |
|
63 | 63 | #include <stdlib.h> |
|
64 | 64 | #include <string.h> |
|
65 | 65 | #include <stdarg.h> |
|
66 | 66 | #include <stdint.h> |
|
67 | 67 | #include <unistd.h> |
|
68 | 68 | #include <getopt.h> |
|
69 | 69 | #include <time.h> |
|
70 | 70 | #include <sys/wait.h> |
|
71 | 71 | #include <sys/user.h> |
|
72 | 72 | #include <sys/time.h> |
|
73 | 73 | #include <sys/ptrace.h> |
|
74 | 74 | #include <sys/signal.h> |
|
75 | 75 | #include <sys/sysinfo.h> |
|
76 | 76 | #include <sys/resource.h> |
|
77 | 77 | #include <sys/utsname.h> |
|
78 | 78 | //#include <linux/ptrace.h> |
|
79 | 79 | |
|
80 | 80 | #if defined(CONFIG_BOX_KERNEL_AMD64) && !defined(CONFIG_BOX_USER_AMD64) |
|
81 | 81 | #include <asm/unistd_32.h> |
|
82 | 82 | #define NATIVE_NR_execve 59 /* 64-bit execve */ |
|
83 | 83 | #else |
|
84 | 84 | #include <asm/unistd.h> |
|
85 | 85 | #define NATIVE_NR_execve __NR_execve |
|
86 | 86 | #endif |
|
87 | 87 | |
|
88 | 88 | #define NONRET __attribute__((noreturn)) |
|
89 | 89 | #define UNUSED __attribute__((unused)) |
|
90 | 90 | #define ARRAY_SIZE(a) (int)(sizeof(a)/sizeof(a[0])) |
|
91 | 91 | |
|
92 | 92 | static int filter_syscalls; /* 0=off, 1=liberal, 2=totalitarian */ |
|
93 | 93 | static int timeout; /* milliseconds */ |
|
94 | 94 | static int wall_timeout; |
|
95 | 95 | static int extra_timeout; |
|
96 | 96 | static int pass_environ; |
|
97 | 97 | static int file_access; |
|
98 | 98 | static int verbose; |
|
99 | 99 | static int memory_limit; |
|
100 | 100 | static int stack_limit; |
|
101 | 101 | static char *redir_stdin, *redir_stdout, *redir_stderr; |
|
102 | 102 | static char *set_cwd; |
|
103 | 103 | |
|
104 | 104 | static pid_t box_pid; |
|
105 | 105 | static int is_ptraced; |
|
106 | 106 | static volatile int timer_tick; |
|
107 | 107 | static struct timeval start_time; |
|
108 | 108 | static int ticks_per_sec; |
|
109 | 109 | static int exec_seen; |
|
110 | 110 | static int partial_line; |
|
111 | 111 | |
|
112 | 112 | static int mem_peak_kb; |
|
113 | 113 | static int total_ms, wall_ms, sys_ms; |
|
114 | 114 | |
|
115 | 115 | static void die(char *msg, ...) NONRET; |
|
116 | 116 | static void sample_mem_peak(void); |
|
117 | 117 | |
|
118 | 118 | /*** Meta-files ***/ |
|
119 | 119 | |
|
120 | 120 | static FILE *metafile; |
|
121 | 121 | |
|
122 | 122 | static void |
|
123 | 123 | meta_open(const char *name) |
|
124 | 124 | { |
|
125 | 125 | if (!strcmp(name, "-")) |
|
126 | 126 | { |
|
127 | 127 | metafile = stdout; |
|
128 | 128 | return; |
|
129 | 129 | } |
|
130 | 130 | metafile = fopen(name, "w"); |
|
131 | 131 | if (!metafile) |
|
132 | 132 | die("Failed to open metafile '%s'",name); |
|
133 | 133 | } |
|
134 | 134 | |
|
135 | 135 | static void |
|
136 | 136 | meta_close(void) |
|
137 | 137 | { |
|
138 | 138 | if (metafile && metafile != stdout) |
|
139 | 139 | fclose(metafile); |
|
140 | 140 | } |
|
141 | 141 | |
|
142 | 142 | static void __attribute__((format(printf,1,2))) |
|
143 | 143 | meta_printf(const char *fmt, ...) |
|
144 | 144 | { |
|
145 | 145 | if (!metafile) |
|
146 | 146 | return; |
|
147 | 147 | |
|
148 | 148 | va_list args; |
|
149 | 149 | va_start(args, fmt); |
|
150 | 150 | vfprintf(metafile, fmt, args); |
|
151 | 151 | va_end(args); |
|
152 | 152 | } |
|
153 | 153 | |
|
154 | 154 | |
|
155 | 155 | static void print_running_stat(double wall_time, |
|
156 | 156 | double user_time, |
|
157 | 157 | double system_time, |
|
158 | 158 | int mem_usage) |
|
159 | 159 | { |
|
160 | 160 | //total is user |
|
161 | 161 | //wall is wall |
|
162 | 162 | // |
|
163 |
- fprintf(stderr,"%.4lfr%.4lfu%.4lfs%d |
|
|
163 | + fprintf(stderr,"%.4lfr%.4lfu%.4lfs%dkbytes\n", | |
|
164 | 164 | wall_time, user_time, system_time, mem_usage); |
|
165 | 165 | } |
|
166 | 166 | |
|
167 | 167 | static void |
|
168 | 168 | final_stats(struct rusage *rus) |
|
169 | 169 | { |
|
170 | 170 | struct timeval total, now, wall; |
|
171 | 171 | timeradd(&rus->ru_utime, &rus->ru_stime, &total); |
|
172 | 172 | total_ms = total.tv_sec*1000 + total.tv_usec/1000; |
|
173 | 173 | gettimeofday(&now, NULL); |
|
174 | 174 | timersub(&now, &start_time, &wall); |
|
175 | 175 | wall_ms = wall.tv_sec*1000 + wall.tv_usec/1000; |
|
176 | 176 | sys_ms = rus->ru_stime.tv_sec * 1000 + rus->ru_stime.tv_usec / 1000; |
|
177 | 177 | |
|
178 | 178 | meta_printf("time:%d.%03d\n", total_ms/1000, total_ms%1000); |
|
179 | 179 | meta_printf("time-wall:%d.%03d\n", wall_ms/1000, wall_ms%1000); |
|
180 | 180 | meta_printf("mem:%llu\n", (unsigned long long) mem_peak_kb * 1024); |
|
181 | 181 | } |
|
182 | 182 | |
|
183 | 183 | /*** Messages and exits ***/ |
|
184 | 184 | |
|
185 | 185 | static void NONRET |
|
186 | 186 | box_exit(int rc) |
|
187 | 187 | { |
|
188 | 188 | if (box_pid > 0) |
|
189 | 189 | { |
|
190 | 190 | sample_mem_peak(); |
|
191 | 191 | if (is_ptraced) |
|
192 | 192 | ptrace(PTRACE_KILL, box_pid); |
|
193 | 193 | kill(-box_pid, SIGKILL); |
|
194 | 194 | kill(box_pid, SIGKILL); |
|
195 | 195 | meta_printf("killed:1\n"); |
|
196 | 196 | |
|
197 | 197 | struct rusage rus; |
|
198 | 198 | int p, stat; |
|
199 | 199 | do |
|
200 | 200 | p = wait4(box_pid, &stat, 0, &rus); |
|
201 | 201 | while (p < 0 && errno == EINTR); |
|
202 | 202 | if (p < 0) |
|
203 | 203 | fprintf(stderr, "UGH: Lost track of the process (%m)\n"); |
|
204 | 204 | else { |
|
205 | 205 | final_stats(&rus); |
|
206 | 206 | } |
|
207 | 207 | } |
|
208 | 208 | print_running_stat( |
|
209 | 209 | (double)wall_ms/1000, |
|
210 | 210 | (double)total_ms/1000, |
|
211 | 211 | (double)sys_ms/1000, |
|
212 |
- |
|
|
212 | + mem_peak_kb); | |
|
213 | 213 | meta_close(); |
|
214 | 214 | exit(rc); |
|
215 | 215 | } |
|
216 | 216 | |
|
217 | 217 | static void |
|
218 | 218 | flush_line(void) |
|
219 | 219 | { |
|
220 | 220 | if (partial_line) |
|
221 | 221 | fputc('\n', stderr); |
|
222 | 222 | partial_line = 0; |
|
223 | 223 | } |
|
224 | 224 | |
|
225 | 225 | /* Report an error of the sandbox itself */ |
|
226 | 226 | static void NONRET __attribute__((format(printf,1,2))) |
|
227 | 227 | die(char *msg, ...) |
|
228 | 228 | { |
|
229 | 229 | va_list args; |
|
230 | 230 | va_start(args, msg); |
|
231 | 231 | flush_line(); |
|
232 | 232 | char buf[1024]; |
|
233 | 233 | vsnprintf(buf, sizeof(buf), msg, args); |
|
234 | 234 | meta_printf("status:XX\nmessage:%s\n", buf); |
|
235 | 235 | fputs(buf, stderr); |
|
236 | 236 | fputc('\n', stderr); |
|
237 | 237 | box_exit(2); |
|
238 | 238 | } |
|
239 | 239 | |
|
240 | 240 | /* Report an error of the program inside the sandbox */ |
|
241 | 241 | static void NONRET __attribute__((format(printf,1,2))) |
|
242 | 242 | err(char *msg, ...) |
|
243 | 243 | { |
|
244 | 244 | va_list args; |
|
245 | 245 | va_start(args, msg); |
|
246 | 246 | flush_line(); |
|
247 | 247 | if (msg[0] && msg[1] && msg[2] == ':' && msg[3] == ' ') |
|
248 | 248 | { |
|
249 | 249 | meta_printf("status:%c%c\n", msg[0], msg[1]); |
|
250 | 250 | msg += 4; |
|
251 | 251 | } |
|
252 | 252 | char buf[1024]; |
|
253 | 253 | vsnprintf(buf, sizeof(buf), msg, args); |
|
254 | 254 | meta_printf("message:%s\n", buf); |
|
255 | 255 | fputs(buf, stderr); |
|
256 | 256 | fputc('\n', stderr); |
|
257 | 257 | box_exit(1); |
|
258 | 258 | } |
|
259 | 259 | |
|
260 | 260 | /* Write a message, but only if in verbose mode */ |
|
261 | 261 | static void __attribute__((format(printf,1,2))) |
|
262 | 262 | msg(char *msg, ...) |
|
263 | 263 | { |
|
264 | 264 | va_list args; |
|
265 | 265 | va_start(args, msg); |
|
266 | 266 | if (verbose) |
|
267 | 267 | { |
|
268 | 268 | int len = strlen(msg); |
|
269 | 269 | if (len > 0) |
|
270 | 270 | partial_line = (msg[len-1] != '\n'); |
|
271 | 271 | vfprintf(stderr, msg, args); |
|
272 | 272 | fflush(stderr); |
|
273 | 273 | } |
|
274 | 274 | va_end(args); |
|
275 | 275 | } |
|
276 | 276 | |
|
277 | 277 | static void * |
|
278 | 278 | xmalloc(size_t size) |
|
279 | 279 | { |
|
280 | 280 | void *p = malloc(size); |
|
281 | 281 | if (!p) |
|
282 | 282 | die("Out of memory"); |
|
283 | 283 | return p; |
|
284 | 284 | } |
|
285 | 285 | |
|
286 | 286 | /*** Syscall rules ***/ |
|
287 | 287 | |
|
288 | 288 | static const char * const syscall_names[] = { |
|
289 | 289 | |
|
290 | 290 | /* Syscall table automatically generated by mk-syscall-table */ |
|
291 | 291 | |
|
292 | 292 | /* 0 */ [ __NR_read ] = "read", |
|
293 | 293 | /* 1 */ [ __NR_write ] = "write", |
|
294 | 294 | /* 2 */ [ __NR_open ] = "open", |
|
295 | 295 | /* 3 */ [ __NR_close ] = "close", |
|
296 | 296 | /* 4 */ [ __NR_stat ] = "stat", |
|
297 | 297 | /* 5 */ [ __NR_fstat ] = "fstat", |
|
298 | 298 | /* 6 */ [ __NR_lstat ] = "lstat", |
|
299 | 299 | /* 7 */ [ __NR_poll ] = "poll", |
|
300 | 300 | /* 8 */ [ __NR_lseek ] = "lseek", |
|
301 | 301 | /* 9 */ [ __NR_mmap ] = "mmap", |
|
302 | 302 | /* 10 */ [ __NR_mprotect ] = "mprotect", |
|
303 | 303 | /* 11 */ [ __NR_munmap ] = "munmap", |
|
304 | 304 | /* 12 */ [ __NR_brk ] = "brk", |
|
305 | 305 | /* 13 */ [ __NR_rt_sigaction ] = "rt_sigaction", |
|
306 | 306 | /* 14 */ [ __NR_rt_sigprocmask ] = "rt_sigprocmask", |
|
307 | 307 | /* 15 */ [ __NR_rt_sigreturn ] = "rt_sigreturn", |
|
308 | 308 | /* 16 */ [ __NR_ioctl ] = "ioctl", |
|
309 | 309 | /* 17 */ [ __NR_pread64 ] = "pread64", |
|
310 | 310 | /* 18 */ [ __NR_pwrite64 ] = "pwrite64", |
|
311 | 311 | /* 19 */ [ __NR_readv ] = "readv", |
|
312 | 312 | /* 20 */ [ __NR_writev ] = "writev", |
|
313 | 313 | /* 21 */ [ __NR_access ] = "access", |
|
314 | 314 | /* 22 */ [ __NR_pipe ] = "pipe", |
|
315 | 315 | /* 23 */ [ __NR_select ] = "select", |
|
316 | 316 | /* 24 */ [ __NR_sched_yield ] = "sched_yield", |
|
317 | 317 | /* 25 */ [ __NR_mremap ] = "mremap", |
|
318 | 318 | /* 26 */ [ __NR_msync ] = "msync", |
|
319 | 319 | /* 27 */ [ __NR_mincore ] = "mincore", |
|
320 | 320 | /* 28 */ [ __NR_madvise ] = "madvise", |
|
321 | 321 | /* 29 */ [ __NR_shmget ] = "shmget", |
|
322 | 322 | /* 30 */ [ __NR_shmat ] = "shmat", |
|
323 | 323 | /* 31 */ [ __NR_shmctl ] = "shmctl", |
|
324 | 324 | /* 32 */ [ __NR_dup ] = "dup", |
|
325 | 325 | /* 33 */ [ __NR_dup2 ] = "dup2", |
|
326 | 326 | /* 34 */ [ __NR_pause ] = "pause", |
|
327 | 327 | /* 35 */ [ __NR_nanosleep ] = "nanosleep", |
|
328 | 328 | /* 36 */ [ __NR_getitimer ] = "getitimer", |
|
329 | 329 | /* 37 */ [ __NR_alarm ] = "alarm", |
|
330 | 330 | /* 38 */ [ __NR_setitimer ] = "setitimer", |
|
331 | 331 | /* 39 */ [ __NR_getpid ] = "getpid", |
|
332 | 332 | /* 40 */ [ __NR_sendfile ] = "sendfile", |
|
333 | 333 | /* 41 */ [ __NR_socket ] = "socket", |
|
334 | 334 | /* 42 */ [ __NR_connect ] = "connect", |
|
335 | 335 | /* 43 */ [ __NR_accept ] = "accept", |
|
336 | 336 | /* 44 */ [ __NR_sendto ] = "sendto", |
|
337 | 337 | /* 45 */ [ __NR_recvfrom ] = "recvfrom", |
|
338 | 338 | /* 46 */ [ __NR_sendmsg ] = "sendmsg", |
|
339 | 339 | /* 47 */ [ __NR_recvmsg ] = "recvmsg", |
|
340 | 340 | /* 48 */ [ __NR_shutdown ] = "shutdown", |
|
341 | 341 | /* 49 */ [ __NR_bind ] = "bind", |
|
342 | 342 | /* 50 */ [ __NR_listen ] = "listen", |
|
343 | 343 | /* 51 */ [ __NR_getsockname ] = "getsockname", |
|
344 | 344 | /* 52 */ [ __NR_getpeername ] = "getpeername", |
|
345 | 345 | /* 53 */ [ __NR_socketpair ] = "socketpair", |
|
346 | 346 | /* 54 */ [ __NR_setsockopt ] = "setsockopt", |
|
347 | 347 | /* 55 */ [ __NR_getsockopt ] = "getsockopt", |
|
348 | 348 | /* 56 */ [ __NR_clone ] = "clone", |
|
349 | 349 | /* 57 */ [ __NR_fork ] = "fork", |
|
350 | 350 | /* 58 */ [ __NR_vfork ] = "vfork", |
|
351 | 351 | /* 59 */ [ __NR_execve ] = "execve", |
|
352 | 352 | /* 60 */ [ __NR_exit ] = "exit", |
|
353 | 353 | /* 61 */ [ __NR_wait4 ] = "wait4", |
|
354 | 354 | /* 62 */ [ __NR_kill ] = "kill", |
|
355 | 355 | /* 63 */ [ __NR_uname ] = "uname", |
|
356 | 356 | /* 64 */ [ __NR_semget ] = "semget", |
|
357 | 357 | /* 65 */ [ __NR_semop ] = "semop", |
|
358 | 358 | /* 66 */ [ __NR_semctl ] = "semctl", |
|
359 | 359 | /* 67 */ [ __NR_shmdt ] = "shmdt", |
|
360 | 360 | /* 68 */ [ __NR_msgget ] = "msgget", |
|
361 | 361 | /* 69 */ [ __NR_msgsnd ] = "msgsnd", |
|
362 | 362 | /* 70 */ [ __NR_msgrcv ] = "msgrcv", |
|
363 | 363 | /* 71 */ [ __NR_msgctl ] = "msgctl", |
|
364 | 364 | /* 72 */ [ __NR_fcntl ] = "fcntl", |
|
365 | 365 | /* 73 */ [ __NR_flock ] = "flock", |
|
366 | 366 | /* 74 */ [ __NR_fsync ] = "fsync", |
|
367 | 367 | /* 75 */ [ __NR_fdatasync ] = "fdatasync", |
|
368 | 368 | /* 76 */ [ __NR_truncate ] = "truncate", |
|
369 | 369 | /* 77 */ [ __NR_ftruncate ] = "ftruncate", |
|
370 | 370 | /* 78 */ [ __NR_getdents ] = "getdents", |
|
371 | 371 | /* 79 */ [ __NR_getcwd ] = "getcwd", |
|
372 | 372 | /* 80 */ [ __NR_chdir ] = "chdir", |
|
373 | 373 | /* 81 */ [ __NR_fchdir ] = "fchdir", |
|
374 | 374 | /* 82 */ [ __NR_rename ] = "rename", |
|
375 | 375 | /* 83 */ [ __NR_mkdir ] = "mkdir", |
|
376 | 376 | /* 84 */ [ __NR_rmdir ] = "rmdir", |
|
377 | 377 | /* 85 */ [ __NR_creat ] = "creat", |
|
378 | 378 | /* 86 */ [ __NR_link ] = "link", |
|
379 | 379 | /* 87 */ [ __NR_unlink ] = "unlink", |
|
380 | 380 | /* 88 */ [ __NR_symlink ] = "symlink", |
|
381 | 381 | /* 89 */ [ __NR_readlink ] = "readlink", |
|
382 | 382 | /* 90 */ [ __NR_chmod ] = "chmod", |
|
383 | 383 | /* 91 */ [ __NR_fchmod ] = "fchmod", |
|
384 | 384 | /* 92 */ [ __NR_chown ] = "chown", |
|
385 | 385 | /* 93 */ [ __NR_fchown ] = "fchown", |
|
386 | 386 | /* 94 */ [ __NR_lchown ] = "lchown", |
|
387 | 387 | /* 95 */ [ __NR_umask ] = "umask", |
|
388 | 388 | /* 96 */ [ __NR_gettimeofday ] = "gettimeofday", |
|
389 | 389 | /* 97 */ [ __NR_getrlimit ] = "getrlimit", |
|
390 | 390 | /* 98 */ [ __NR_getrusage ] = "getrusage", |
|
391 | 391 | /* 99 */ [ __NR_sysinfo ] = "sysinfo", |
|
392 | 392 | /* 100 */ [ __NR_times ] = "times", |
|
393 | 393 | /* 101 */ [ __NR_ptrace ] = "ptrace", |
|
394 | 394 | /* 102 */ [ __NR_getuid ] = "getuid", |
|
395 | 395 | /* 103 */ [ __NR_syslog ] = "syslog", |
|
396 | 396 | /* 104 */ [ __NR_getgid ] = "getgid", |
|
397 | 397 | /* 105 */ [ __NR_setuid ] = "setuid", |
|
398 | 398 | /* 106 */ [ __NR_setgid ] = "setgid", |
|
399 | 399 | /* 107 */ [ __NR_geteuid ] = "geteuid", |
|
400 | 400 | /* 108 */ [ __NR_getegid ] = "getegid", |
|
401 | 401 | /* 109 */ [ __NR_setpgid ] = "setpgid", |
|
402 | 402 | /* 110 */ [ __NR_getppid ] = "getppid", |
|
403 | 403 | /* 111 */ [ __NR_getpgrp ] = "getpgrp", |
|
404 | 404 | /* 112 */ [ __NR_setsid ] = "setsid", |
|
405 | 405 | /* 113 */ [ __NR_setreuid ] = "setreuid", |
|
406 | 406 | /* 114 */ [ __NR_setregid ] = "setregid", |
|
407 | 407 | /* 115 */ [ __NR_getgroups ] = "getgroups", |
|
408 | 408 | /* 116 */ [ __NR_setgroups ] = "setgroups", |
|
409 | 409 | /* 117 */ [ __NR_setresuid ] = "setresuid", |
|
410 | 410 | /* 118 */ [ __NR_getresuid ] = "getresuid", |
|
411 | 411 | /* 119 */ [ __NR_setresgid ] = "setresgid", |
|
412 | 412 | /* 120 */ [ __NR_getresgid ] = "getresgid", |
|
413 | 413 | /* 121 */ [ __NR_getpgid ] = "getpgid", |
|
414 | 414 | /* 122 */ [ __NR_setfsuid ] = "setfsuid", |
|
415 | 415 | /* 123 */ [ __NR_setfsgid ] = "setfsgid", |
|
416 | 416 | /* 124 */ [ __NR_getsid ] = "getsid", |
|
417 | 417 | /* 125 */ [ __NR_capget ] = "capget", |
|
418 | 418 | /* 126 */ [ __NR_capset ] = "capset", |
|
419 | 419 | /* 127 */ [ __NR_rt_sigpending ] = "rt_sigpending", |
|
420 | 420 | /* 128 */ [ __NR_rt_sigtimedwait ] = "rt_sigtimedwait", |
|
421 | 421 | /* 129 */ [ __NR_rt_sigqueueinfo ] = "rt_sigqueueinfo", |
|
422 | 422 | /* 130 */ [ __NR_rt_sigsuspend ] = "rt_sigsuspend", |
|
423 | 423 | /* 131 */ [ __NR_sigaltstack ] = "sigaltstack", |
|
424 | 424 | /* 132 */ [ __NR_utime ] = "utime", |
|
425 | 425 | /* 133 */ [ __NR_mknod ] = "mknod", |
|
426 | 426 | /* 134 */ [ __NR_uselib ] = "uselib", |
|
427 | 427 | /* 135 */ [ __NR_personality ] = "personality", |
|
428 | 428 | /* 136 */ [ __NR_ustat ] = "ustat", |
|
429 | 429 | /* 137 */ [ __NR_statfs ] = "statfs", |
|
430 | 430 | /* 138 */ [ __NR_fstatfs ] = "fstatfs", |
|
431 | 431 | /* 139 */ [ __NR_sysfs ] = "sysfs", |
|
432 | 432 | /* 140 */ [ __NR_getpriority ] = "getpriority", |
|
433 | 433 | /* 141 */ [ __NR_setpriority ] = "setpriority", |
|
434 | 434 | /* 142 */ [ __NR_sched_setparam ] = "sched_setparam", |
|
435 | 435 | /* 143 */ [ __NR_sched_getparam ] = "sched_getparam", |
|
436 | 436 | /* 144 */ [ __NR_sched_setscheduler ] = "sched_setscheduler", |
|
437 | 437 | /* 145 */ [ __NR_sched_getscheduler ] = "sched_getscheduler", |
|
438 | 438 | /* 146 */ [ __NR_sched_get_priority_max ] = "sched_get_priority_max", |
|
439 | 439 | /* 147 */ [ __NR_sched_get_priority_min ] = "sched_get_priority_min", |
|
440 | 440 | /* 148 */ [ __NR_sched_rr_get_interval ] = "sched_rr_get_interval", |
|
441 | 441 | /* 149 */ [ __NR_mlock ] = "mlock", |
|
442 | 442 | /* 150 */ [ __NR_munlock ] = "munlock", |
|
443 | 443 | /* 151 */ [ __NR_mlockall ] = "mlockall", |
|
444 | 444 | /* 152 */ [ __NR_munlockall ] = "munlockall", |
|
445 | 445 | /* 153 */ [ __NR_vhangup ] = "vhangup", |
|
446 | 446 | /* 154 */ [ __NR_modify_ldt ] = "modify_ldt", |
|
447 | 447 | /* 155 */ [ __NR_pivot_root ] = "pivot_root", |
|
448 | 448 | /* 156 */ [ __NR__sysctl ] = "_sysctl", |
|
449 | 449 | /* 157 */ [ __NR_prctl ] = "prctl", |
|
450 | 450 | /* 158 */ [ __NR_arch_prctl ] = "arch_prctl", |
|
451 | 451 | /* 159 */ [ __NR_adjtimex ] = "adjtimex", |
|
452 | 452 | /* 160 */ [ __NR_setrlimit ] = "setrlimit", |
|
453 | 453 | /* 161 */ [ __NR_chroot ] = "chroot", |
|
454 | 454 | /* 162 */ [ __NR_sync ] = "sync", |
|
455 | 455 | /* 163 */ [ __NR_acct ] = "acct", |
|
456 | 456 | /* 164 */ [ __NR_settimeofday ] = "settimeofday", |
|
457 | 457 | /* 165 */ [ __NR_mount ] = "mount", |
|
458 | 458 | /* 166 */ [ __NR_umount2 ] = "umount2", |
|
459 | 459 | /* 167 */ [ __NR_swapon ] = "swapon", |
|
460 | 460 | /* 168 */ [ __NR_swapoff ] = "swapoff", |
|
461 | 461 | /* 169 */ [ __NR_reboot ] = "reboot", |
|
462 | 462 | /* 170 */ [ __NR_sethostname ] = "sethostname", |
|
463 | 463 | /* 171 */ [ __NR_setdomainname ] = "setdomainname", |
|
464 | 464 | /* 172 */ [ __NR_iopl ] = "iopl", |
|
465 | 465 | /* 173 */ [ __NR_ioperm ] = "ioperm", |
|
466 | 466 | /* 174 */ [ __NR_create_module ] = "create_module", |
|
467 | 467 | /* 175 */ [ __NR_init_module ] = "init_module", |
|
468 | 468 | /* 176 */ [ __NR_delete_module ] = "delete_module", |
|
469 | 469 | /* 177 */ [ __NR_get_kernel_syms ] = "get_kernel_syms", |
|
470 | 470 | /* 178 */ [ __NR_query_module ] = "query_module", |
|
471 | 471 | /* 179 */ [ __NR_quotactl ] = "quotactl", |
|
472 | 472 | /* 180 */ [ __NR_nfsservctl ] = "nfsservctl", |
|
473 | 473 | /* 181 */ [ __NR_getpmsg ] = "getpmsg", |
|
474 | 474 | /* 182 */ [ __NR_putpmsg ] = "putpmsg", |
|
475 | 475 | /* 183 */ [ __NR_afs_syscall ] = "afs_syscall", |
|
476 | 476 | /* 184 */ [ __NR_tuxcall ] = "tuxcall", |
|
477 | 477 | /* 185 */ [ __NR_security ] = "security", |
|
478 | 478 | /* 186 */ [ __NR_gettid ] = "gettid", |
|
479 | 479 | /* 187 */ [ __NR_readahead ] = "readahead", |
|
480 | 480 | /* 188 */ [ __NR_setxattr ] = "setxattr", |
|
481 | 481 | /* 189 */ [ __NR_lsetxattr ] = "lsetxattr", |
|
482 | 482 | /* 190 */ [ __NR_fsetxattr ] = "fsetxattr", |
|
483 | 483 | /* 191 */ [ __NR_getxattr ] = "getxattr", |
|
484 | 484 | /* 192 */ [ __NR_lgetxattr ] = "lgetxattr", |
|
485 | 485 | /* 193 */ [ __NR_fgetxattr ] = "fgetxattr", |
|
486 | 486 | /* 194 */ [ __NR_listxattr ] = "listxattr", |
|
487 | 487 | /* 195 */ [ __NR_llistxattr ] = "llistxattr", |
|
488 | 488 | /* 196 */ [ __NR_flistxattr ] = "flistxattr", |
|
489 | 489 | /* 197 */ [ __NR_removexattr ] = "removexattr", |
|
490 | 490 | /* 198 */ [ __NR_lremovexattr ] = "lremovexattr", |
|
491 | 491 | /* 199 */ [ __NR_fremovexattr ] = "fremovexattr", |
|
492 | 492 | /* 200 */ [ __NR_tkill ] = "tkill", |
|
493 | 493 | /* 201 */ [ __NR_time ] = "time", |
|
494 | 494 | /* 202 */ [ __NR_futex ] = "futex", |
|
495 | 495 | /* 203 */ [ __NR_sched_setaffinity ] = "sched_setaffinity", |
|
496 | 496 | /* 204 */ [ __NR_sched_getaffinity ] = "sched_getaffinity", |
|
497 | 497 | /* 205 */ [ __NR_set_thread_area ] = "set_thread_area", |
|
498 | 498 | /* 206 */ [ __NR_io_setup ] = "io_setup", |
|
499 | 499 | /* 207 */ [ __NR_io_destroy ] = "io_destroy", |
|
500 | 500 | /* 208 */ [ __NR_io_getevents ] = "io_getevents", |
|
501 | 501 | /* 209 */ [ __NR_io_submit ] = "io_submit", |
|
502 | 502 | /* 210 */ [ __NR_io_cancel ] = "io_cancel", |
|
503 | 503 | /* 211 */ [ __NR_get_thread_area ] = "get_thread_area", |
|
504 | 504 | /* 212 */ [ __NR_lookup_dcookie ] = "lookup_dcookie", |
|
505 | 505 | /* 213 */ [ __NR_epoll_create ] = "epoll_create", |
|
506 | 506 | /* 214 */ [ __NR_epoll_ctl_old ] = "epoll_ctl_old", |
|
507 | 507 | /* 215 */ [ __NR_epoll_wait_old ] = "epoll_wait_old", |
|
508 | 508 | /* 216 */ [ __NR_remap_file_pages ] = "remap_file_pages", |
|
509 | 509 | /* 217 */ [ __NR_getdents64 ] = "getdents64", |
|
510 | 510 | /* 218 */ [ __NR_set_tid_address ] = "set_tid_address", |
|
511 | 511 | /* 219 */ [ __NR_restart_syscall ] = "restart_syscall", |
|
512 | 512 | /* 220 */ [ __NR_semtimedop ] = "semtimedop", |
|
513 | 513 | /* 221 */ [ __NR_fadvise64 ] = "fadvise64", |
|
514 | 514 | /* 222 */ [ __NR_timer_create ] = "timer_create", |
|
515 | 515 | /* 223 */ [ __NR_timer_settime ] = "timer_settime", |
|
516 | 516 | /* 224 */ [ __NR_timer_gettime ] = "timer_gettime", |
|
517 | 517 | /* 225 */ [ __NR_timer_getoverrun ] = "timer_getoverrun", |
|
518 | 518 | /* 226 */ [ __NR_timer_delete ] = "timer_delete", |
|
519 | 519 | /* 227 */ [ __NR_clock_settime ] = "clock_settime", |
|
520 | 520 | /* 228 */ [ __NR_clock_gettime ] = "clock_gettime", |
|
521 | 521 | /* 229 */ [ __NR_clock_getres ] = "clock_getres", |
|
522 | 522 | /* 230 */ [ __NR_clock_nanosleep ] = "clock_nanosleep", |
|
523 | 523 | /* 231 */ [ __NR_exit_group ] = "exit_group", |
|
524 | 524 | /* 232 */ [ __NR_epoll_wait ] = "epoll_wait", |
|
525 | 525 | /* 233 */ [ __NR_epoll_ctl ] = "epoll_ctl", |
|
526 | 526 | /* 234 */ [ __NR_tgkill ] = "tgkill", |
|
527 | 527 | /* 235 */ [ __NR_utimes ] = "utimes", |
|
528 | 528 | /* 236 */ [ __NR_vserver ] = "vserver", |
|
529 | 529 | /* 237 */ [ __NR_mbind ] = "mbind", |
|
530 | 530 | /* 238 */ [ __NR_set_mempolicy ] = "set_mempolicy", |
|
531 | 531 | /* 239 */ [ __NR_get_mempolicy ] = "get_mempolicy", |
|
532 | 532 | /* 240 */ [ __NR_mq_open ] = "mq_open", |
|
533 | 533 | /* 241 */ [ __NR_mq_unlink ] = "mq_unlink", |
|
534 | 534 | /* 242 */ [ __NR_mq_timedsend ] = "mq_timedsend", |
|
535 | 535 | /* 243 */ [ __NR_mq_timedreceive ] = "mq_timedreceive", |
|
536 | 536 | /* 244 */ [ __NR_mq_notify ] = "mq_notify", |
|
537 | 537 | /* 245 */ [ __NR_mq_getsetattr ] = "mq_getsetattr", |
|
538 | 538 | /* 246 */ [ __NR_kexec_load ] = "kexec_load", |
|
539 | 539 | /* 247 */ [ __NR_waitid ] = "waitid", |
|
540 | 540 | /* 248 */ [ __NR_add_key ] = "add_key", |
|
541 | 541 | /* 249 */ [ __NR_request_key ] = "request_key", |
|
542 | 542 | /* 250 */ [ __NR_keyctl ] = "keyctl", |
|
543 | 543 | /* 251 */ [ __NR_ioprio_set ] = "ioprio_set", |
|
544 | 544 | /* 252 */ [ __NR_ioprio_get ] = "ioprio_get", |
|
545 | 545 | /* 253 */ [ __NR_inotify_init ] = "inotify_init", |
|
546 | 546 | /* 254 */ [ __NR_inotify_add_watch ] = "inotify_add_watch", |
|
547 | 547 | /* 255 */ [ __NR_inotify_rm_watch ] = "inotify_rm_watch", |
|
548 | 548 | /* 256 */ [ __NR_migrate_pages ] = "migrate_pages", |
|
549 | 549 | /* 257 */ [ __NR_openat ] = "openat", |
|
550 | 550 | /* 258 */ [ __NR_mkdirat ] = "mkdirat", |
|
551 | 551 | /* 259 */ [ __NR_mknodat ] = "mknodat", |
|
552 | 552 | /* 260 */ [ __NR_fchownat ] = "fchownat", |
|
553 | 553 | /* 261 */ [ __NR_futimesat ] = "futimesat", |
|
554 | 554 | /* 262 */ [ __NR_newfstatat ] = "newfstatat", |
|
555 | 555 | /* 263 */ [ __NR_unlinkat ] = "unlinkat", |
|
556 | 556 | /* 264 */ [ __NR_renameat ] = "renameat", |
|
557 | 557 | /* 265 */ [ __NR_linkat ] = "linkat", |
|
558 | 558 | /* 266 */ [ __NR_symlinkat ] = "symlinkat", |
|
559 | 559 | /* 267 */ [ __NR_readlinkat ] = "readlinkat", |
|
560 | 560 | /* 268 */ [ __NR_fchmodat ] = "fchmodat", |
|
561 | 561 | /* 269 */ [ __NR_faccessat ] = "faccessat", |
|
562 | 562 | /* 270 */ [ __NR_pselect6 ] = "pselect6", |
|
563 | 563 | /* 271 */ [ __NR_ppoll ] = "ppoll", |
|
564 | 564 | /* 272 */ [ __NR_unshare ] = "unshare", |
|
565 | 565 | /* 273 */ [ __NR_set_robust_list ] = "set_robust_list", |
|
566 | 566 | /* 274 */ [ __NR_get_robust_list ] = "get_robust_list", |
|
567 | 567 | /* 275 */ [ __NR_splice ] = "splice", |
|
568 | 568 | /* 276 */ [ __NR_tee ] = "tee", |
|
569 | 569 | /* 277 */ [ __NR_sync_file_range ] = "sync_file_range", |
|
570 | 570 | /* 278 */ [ __NR_vmsplice ] = "vmsplice", |
|
571 | 571 | /* 279 */ [ __NR_move_pages ] = "move_pages", |
|
572 | 572 | /* 280 */ [ __NR_utimensat ] = "utimensat", |
|
573 | 573 | /* 281 */ [ __NR_epoll_pwait ] = "epoll_pwait", |
|
574 | 574 | /* 282 */ [ __NR_signalfd ] = "signalfd", |
|
575 | 575 | /* 283 */ [ __NR_timerfd_create ] = "timerfd_create", |
|
576 | 576 | /* 284 */ [ __NR_eventfd ] = "eventfd", |
|
577 | 577 | /* 285 */ [ __NR_fallocate ] = "fallocate", |
|
578 | 578 | /* 286 */ [ __NR_timerfd_settime ] = "timerfd_settime", |
|
579 | 579 | /* 287 */ [ __NR_timerfd_gettime ] = "timerfd_gettime", |
|
580 | 580 | /* 288 */ [ __NR_accept4 ] = "accept4", |
|
581 | 581 | /* 289 */ [ __NR_signalfd4 ] = "signalfd4", |
|
582 | 582 | /* 290 */ [ __NR_eventfd2 ] = "eventfd2", |
|
583 | 583 | /* 291 */ [ __NR_epoll_create1 ] = "epoll_create1", |
|
584 | 584 | /* 292 */ [ __NR_dup3 ] = "dup3", |
|
585 | 585 | /* 293 */ [ __NR_pipe2 ] = "pipe2", |
|
586 | 586 | /* 294 */ [ __NR_inotify_init1 ] = "inotify_init1", |
|
587 | 587 | /* 295 */ [ __NR_preadv ] = "preadv", |
|
588 | 588 | /* 296 */ [ __NR_pwritev ] = "pwritev", |
|
589 | 589 | /* 297 */ [ __NR_rt_tgsigqueueinfo ] = "rt_tgsigqueueinfo", |
|
590 | 590 | /* 298 */ [ __NR_perf_event_open ] = "perf_event_open", |
|
591 | 591 | /* 299 */ [ __NR_recvmmsg ] = "recvmmsg", |
|
592 | 592 | /* 300 */ [ __NR_fanotify_init ] = "fanotify_init", |
|
593 | 593 | /* 301 */ [ __NR_fanotify_mark ] = "fanotify_mark", |
|
594 | 594 | /* 302 */ [ __NR_prlimit64 ] = "prlimit64", |
|
595 | 595 | /* 303 */ [ __NR_name_to_handle_at ] = "name_to_handle_at", |
|
596 | 596 | /* 304 */ [ __NR_open_by_handle_at ] = "open_by_handle_at", |
|
597 | 597 | /* 305 */ [ __NR_clock_adjtime ] = "clock_adjtime", |
|
598 | 598 | /* 306 */ [ __NR_syncfs ] = "syncfs", |
|
599 | 599 | /* 307 */ [ __NR_sendmmsg ] = "sendmmsg", |
|
600 | 600 | /* 308 */ [ __NR_setns ] = "setns", |
|
601 | 601 | /* 309 */ [ __NR_getcpu ] = "getcpu", |
|
602 | 602 | /* 310 */ [ __NR_process_vm_readv ] = "process_vm_readv", |
|
603 | 603 | /* 311 */ [ __NR_process_vm_writev ] = "process_vm_writev", |
|
604 | 604 | /* 312 */ [ __NR_kcmp ] = "kcmp", |
|
605 | 605 | /* 313 */ [ __NR_finit_module ] = "finit_module", |
|
606 | 606 | }; |
|
607 | 607 | #define NUM_SYSCALLS ARRAY_SIZE(syscall_names) |
|
608 | 608 | #define NUM_ACTIONS (NUM_SYSCALLS+64) |
|
609 | 609 | |
|
610 | 610 | enum action { |
|
611 | 611 | A_DEFAULT, // Use the default action |
|
612 | 612 | A_NO, // Always forbid |
|
613 | 613 | A_YES, // Always permit |
|
614 | 614 | A_FILENAME, // Permit if arg1 is a known filename |
|
615 | 615 | A_ACTION_MASK = 15, |
|
616 | 616 | A_NO_RETVAL = 32, // Does not return a value |
|
617 | 617 | A_SAMPLE_MEM = 64, // Sample memory usage before the syscall |
|
618 | 618 | A_LIBERAL = 128, // Valid only in liberal mode |
|
619 | 619 | // Must fit in a unsigned char |
|
620 | 620 | }; |
|
621 | 621 | |
|
622 | 622 | static unsigned char syscall_action[NUM_ACTIONS] = { |
|
623 | 623 | #define S(x) [__NR_##x] |
|
624 | 624 | |
|
625 | 625 | // Syscalls permitted for specific file names |
|
626 | 626 | S(open) = A_FILENAME, |
|
627 | 627 | S(creat) = A_FILENAME, |
|
628 | 628 | S(unlink) = A_FILENAME, |
|
629 | 629 | S(access) = A_FILENAME, |
|
630 | 630 | S(truncate) = A_FILENAME, |
|
631 | 631 | S(stat) = A_FILENAME, |
|
632 | 632 | S(lstat) = A_FILENAME, |
|
633 | 633 | S(readlink) = A_FILENAME, |
|
634 | 634 | #ifndef CONFIG_BOX_USER_AMD64 |
|
635 | 635 | S(oldstat) = A_FILENAME, |
|
636 | 636 | S(oldlstat) = A_FILENAME, |
|
637 | 637 | S(truncate64) = A_FILENAME, |
|
638 | 638 | S(stat64) = A_FILENAME, |
|
639 | 639 | S(lstat64) = A_FILENAME, |
|
640 | 640 | #endif |
|
641 | 641 | |
|
642 | 642 | // Syscalls permitted always |
|
643 | 643 | S(exit) = A_YES | A_SAMPLE_MEM, |
|
644 | 644 | S(read) = A_YES, |
|
645 | 645 | S(write) = A_YES, |
|
646 | 646 | S(close) = A_YES, |
|
647 | 647 | S(lseek) = A_YES, |
|
648 | 648 | S(getpid) = A_YES, |
|
649 | 649 | S(getuid) = A_YES, |
|
650 | 650 | S(dup) = A_YES, |
|
651 | 651 | S(brk) = A_YES, |
|
652 | 652 | S(getgid) = A_YES, |
|
653 | 653 | S(geteuid) = A_YES, |
|
654 | 654 | S(getegid) = A_YES, |
|
655 | 655 | S(dup2) = A_YES, |
|
656 | 656 | S(ftruncate) = A_YES, |
|
657 | 657 | S(fstat) = A_YES, |
|
658 | 658 | S(personality) = A_YES, |
|
659 | 659 | S(readv) = A_YES, |
|
660 | 660 | S(writev) = A_YES, |
|
661 | 661 | S(getresuid) = A_YES, |
|
662 | 662 | #ifdef __NR_pread64 |
|
663 | 663 | S(pread64) = A_YES, |
|
664 | 664 | S(pwrite64) = A_YES, |
|
665 | 665 | #else |
|
666 | 666 | S(pread) = A_YES, |
|
667 | 667 | S(pwrite) = A_YES, |
|
668 | 668 | #endif |
|
669 | 669 | S(fcntl) = A_YES, |
|
670 | 670 | S(mmap) = A_YES, |
|
671 | 671 | S(munmap) = A_YES, |
|
672 | 672 | S(ioctl) = A_YES, |
|
673 | 673 | S(uname) = A_YES, |
|
674 | 674 | S(gettid) = A_YES, |
|
675 | 675 | S(set_thread_area) = A_YES, |
|
676 | 676 | S(get_thread_area) = A_YES, |
|
677 | 677 | S(set_tid_address) = A_YES, |
|
678 | 678 | S(exit_group) = A_YES | A_SAMPLE_MEM, |
|
679 | 679 | #ifdef CONFIG_BOX_USER_AMD64 |
|
680 | 680 | S(arch_prctl) = A_YES, |
|
681 | 681 | #else |
|
682 | 682 | S(oldfstat) = A_YES, |
|
683 | 683 | S(ftruncate64) = A_YES, |
|
684 | 684 | S(_llseek) = A_YES, |
|
685 | 685 | S(fstat64) = A_YES, |
|
686 | 686 | S(fcntl64) = A_YES, |
|
687 | 687 | S(mmap2) = A_YES, |
|
688 | 688 | #endif |
|
689 | 689 | |
|
690 | 690 | // Syscalls permitted only in liberal mode |
|
691 | 691 | S(time) = A_YES | A_LIBERAL, |
|
692 | 692 | S(alarm) = A_YES | A_LIBERAL, |
|
693 | 693 | S(pause) = A_YES | A_LIBERAL, |
|
694 | 694 | S(fchmod) = A_YES | A_LIBERAL, |
|
695 | 695 | S(getrlimit) = A_YES | A_LIBERAL, |
|
696 | 696 | S(getrusage) = A_YES | A_LIBERAL, |
|
697 | 697 | S(gettimeofday) = A_YES | A_LIBERAL, |
|
698 | 698 | S(select) = A_YES | A_LIBERAL, |
|
699 | 699 | S(setitimer) = A_YES | A_LIBERAL, |
|
700 | 700 | S(getitimer) = A_YES | A_LIBERAL, |
|
701 | 701 | S(mprotect) = A_YES | A_LIBERAL, |
|
702 | 702 | S(getdents) = A_YES | A_LIBERAL, |
|
703 | 703 | S(getdents64) = A_YES | A_LIBERAL, |
|
704 | 704 | S(fdatasync) = A_YES | A_LIBERAL, |
|
705 | 705 | S(mremap) = A_YES | A_LIBERAL, |
|
706 | 706 | S(poll) = A_YES | A_LIBERAL, |
|
707 | 707 | S(getcwd) = A_YES | A_LIBERAL, |
|
708 | 708 | S(nanosleep) = A_YES | A_LIBERAL, |
|
709 | 709 | S(rt_sigreturn) = A_YES | A_LIBERAL | A_NO_RETVAL, |
|
710 | 710 | S(rt_sigaction) = A_YES | A_LIBERAL, |
|
711 | 711 | S(rt_sigprocmask) = A_YES | A_LIBERAL, |
|
712 | 712 | S(rt_sigpending) = A_YES | A_LIBERAL, |
|
713 | 713 | S(rt_sigtimedwait) = A_YES | A_LIBERAL, |
|
714 | 714 | S(rt_sigqueueinfo) = A_YES | A_LIBERAL, |
|
715 | 715 | S(rt_sigsuspend) = A_YES | A_LIBERAL, |
|
716 | 716 | S(_sysctl) = A_YES | A_LIBERAL, |
|
717 | 717 | #ifndef CONFIG_BOX_USER_AMD64 |
|
718 | 718 | S(sigaction) = A_YES | A_LIBERAL, |
|
719 | 719 | S(sgetmask) = A_YES | A_LIBERAL, |
|
720 | 720 | S(ssetmask) = A_YES | A_LIBERAL, |
|
721 | 721 | S(sigsuspend) = A_YES | A_LIBERAL, |
|
722 | 722 | S(sigpending) = A_YES | A_LIBERAL, |
|
723 | 723 | S(sigreturn) = A_YES | A_LIBERAL | A_NO_RETVAL, |
|
724 | 724 | S(sigprocmask) = A_YES | A_LIBERAL, |
|
725 | 725 | S(ugetrlimit) = A_YES | A_LIBERAL, |
|
726 | 726 | S(readdir) = A_YES | A_LIBERAL, |
|
727 | 727 | S(signal) = A_YES | A_LIBERAL, |
|
728 | 728 | S(_newselect) = A_YES | A_LIBERAL, |
|
729 | 729 | #endif |
|
730 | 730 | |
|
731 | 731 | #undef S |
|
732 | 732 | }; |
|
733 | 733 | |
|
734 | 734 | static const char * |
|
735 | 735 | syscall_name(unsigned int id, char *buf) |
|
736 | 736 | { |
|
737 | 737 | if (id < NUM_SYSCALLS && syscall_names[id]) |
|
738 | 738 | return syscall_names[id]; |
|
739 | 739 | else |
|
740 | 740 | { |
|
741 | 741 | sprintf(buf, "#%d", id); |
|
742 | 742 | return buf; |
|
743 | 743 | } |
|
744 | 744 | } |
|
745 | 745 | |
|
746 | 746 | static int |
|
747 | 747 | syscall_by_name(char *name) |
|
748 | 748 | { |
|
749 | 749 | for (unsigned int i=0; i<NUM_SYSCALLS; i++) |
|
750 | 750 | if (syscall_names[i] && !strcmp(syscall_names[i], name)) |
|
751 | 751 | return i; |
|
752 | 752 | if (name[0] == '#') |
|
753 | 753 | name++; |
|
754 | 754 | if (!*name) |
|
755 | 755 | return -1; |
|
756 | 756 | char *ep; |
|
757 | 757 | unsigned long l = strtoul(name, &ep, 0); |
|
758 | 758 | if (*ep) |
|
759 | 759 | return -1; |
|
760 | 760 | if (l >= NUM_ACTIONS) |
|
761 | 761 | return NUM_ACTIONS; |
|
762 | 762 | return l; |
|
763 | 763 | } |
|
764 | 764 | |
|
765 | 765 | static int |
|
766 | 766 | set_syscall_action(char *a) |
|
767 | 767 | { |
|
768 | 768 | char *sep = strchr(a, '='); |
|
769 | 769 | enum action act = A_YES; |
|
770 | 770 | if (sep) |
|
771 | 771 | { |
|
772 | 772 | *sep++ = 0; |
|
773 | 773 | if (!strcmp(sep, "yes")) |
|
774 | 774 | act = A_YES; |
|
775 | 775 | else if (!strcmp(sep, "no")) |
|
776 | 776 | act = A_NO; |
|
777 | 777 | else if (!strcmp(sep, "file")) |
|
778 | 778 | act = A_FILENAME; |
|
779 | 779 | else |
|
780 | 780 | return 0; |
|
781 | 781 | } |
|
782 | 782 | |
|
783 | 783 | int sys = syscall_by_name(a); |
|
784 | 784 | if (sys < 0) |
|
785 | 785 | die("Unknown syscall `%s'", a); |
|
786 | 786 | if (sys >= NUM_ACTIONS) |
|
787 | 787 | die("Syscall `%s' out of range", a); |
|
788 | 788 | syscall_action[sys] = act; |
|
789 | 789 | return 1; |
|
790 | 790 | } |
|
791 | 791 | |
|
792 | 792 | /*** Path rules ***/ |
|
793 | 793 | |
|
794 | 794 | struct path_rule { |
|
795 | 795 | char *path; |
|
796 | 796 | enum action action; |
|
797 | 797 | struct path_rule *next; |
|
798 | 798 | }; |
|
799 | 799 | |
|
800 | 800 | static struct path_rule default_path_rules[] = { |
|
801 | 801 | { "/etc/", A_YES }, |
|
802 | 802 | { "/lib/", A_YES }, |
|
803 | 803 | { "/usr/lib/", A_YES }, |
|
804 | 804 | { "/opt/lib/", A_YES }, |
|
805 | 805 | { "/usr/share/zoneinfo/", A_YES }, |
|
806 | 806 | { "/usr/share/locale/", A_YES }, |
|
807 | 807 | { "/dev/null", A_YES }, |
|
808 | 808 | { "/dev/zero", A_YES }, |
|
809 | 809 | { "/proc/meminfo", A_YES }, |
|
810 | 810 | { "/proc/self/stat", A_YES }, |
|
811 | 811 | { "/proc/self/exe", A_YES }, // Needed by FPC 2.0.x runtime |
|
812 | 812 | { "/proc/self/maps", A_YES }, // Needed by glibc when it reports arena corruption |
|
813 | 813 | }; |
|
814 | 814 | |
|
815 | 815 | static struct path_rule *user_path_rules; |
|
816 | 816 | static struct path_rule **last_path_rule = &user_path_rules; |
|
817 | 817 | |
|
818 | 818 | static int |
|
819 | 819 | set_path_action(char *a) |
|
820 | 820 | { |
|
821 | 821 | char *sep = strchr(a, '='); |
|
822 | 822 | enum action act = A_YES; |
|
823 | 823 | if (sep) |
|
824 | 824 | { |
|
825 | 825 | *sep++ = 0; |
|
826 | 826 | if (!strcmp(sep, "yes")) |
|
827 | 827 | act = A_YES; |
|
828 | 828 | else if (!strcmp(sep, "no")) |
|
829 | 829 | act = A_NO; |
|
830 | 830 | else |
|
831 | 831 | return 0; |
|
832 | 832 | } |
|
833 | 833 | |
|
834 | 834 | struct path_rule *r = xmalloc(sizeof(*r) + strlen(a) + 1); |
|
835 | 835 | r->path = (char *)(r+1); |
|
836 | 836 | strcpy(r->path, a); |
|
837 | 837 | r->action = act; |
|
838 | 838 | r->next = NULL; |
|
839 | 839 | *last_path_rule = r; |
|
840 | 840 | last_path_rule = &r->next; |
|
841 | 841 | return 1; |
|
842 | 842 | } |
|
843 | 843 | |
|
844 | 844 | static enum action |
|
845 | 845 | match_path_rule(struct path_rule *r, char *path) |
|
846 | 846 | { |
|
847 | 847 | char *rr = r->path; |
|
848 | 848 | while (*rr) |
|
849 | 849 | if (*rr++ != *path++) |
|
850 | 850 | { |
|
851 | 851 | if (rr[-1] == '/' && !path[-1]) |
|
852 | 852 | break; |
|
853 | 853 | return A_DEFAULT; |
|
854 | 854 | } |
|
855 | 855 | if (rr > r->path && rr[-1] != '/' && *path) |
|
856 | 856 | return A_DEFAULT; |
|
857 | 857 | return r->action; |
|
858 | 858 | } |
|
859 | 859 | |
|
860 | 860 | /*** Environment rules ***/ |
|
861 | 861 | |
|
862 | 862 | struct env_rule { |
|
863 | 863 | char *var; // Variable to match |
|
864 | 864 | char *val; // ""=clear, NULL=inherit |
|
865 | 865 | int var_len; |
|
866 | 866 | struct env_rule *next; |
|
867 | 867 | }; |
|
868 | 868 | |
|
869 | 869 | static struct env_rule *first_env_rule; |
|
870 | 870 | static struct env_rule **last_env_rule = &first_env_rule; |
|
871 | 871 | |
|
872 | 872 | static struct env_rule default_env_rules[] = { |
|
873 | 873 | { "LIBC_FATAL_STDERR_", "1" } |
|
874 | 874 | }; |
|
875 | 875 | |
|
876 | 876 | static int |
|
877 | 877 | set_env_action(char *a0) |
|
878 | 878 | { |
|
879 | 879 | struct env_rule *r = xmalloc(sizeof(*r) + strlen(a0) + 1); |
|
880 | 880 | char *a = (char *)(r+1); |
|
881 | 881 | strcpy(a, a0); |
|
882 | 882 | |
|
883 | 883 | char *sep = strchr(a, '='); |
|
884 | 884 | if (sep == a) |
|
885 | 885 | return 0; |
|
886 | 886 | r->var = a; |
|
887 | 887 | if (sep) |
|
888 | 888 | { |
|
889 | 889 | *sep++ = 0; |
|
890 | 890 | r->val = sep; |
|
891 | 891 | } |
|
892 | 892 | else |
|
893 | 893 | r->val = NULL; |
|
894 | 894 | *last_env_rule = r; |
|
895 | 895 | last_env_rule = &r->next; |
|
896 | 896 | r->next = NULL; |
|
897 | 897 | return 1; |
|
898 | 898 | } |
|
899 | 899 | |
|
900 | 900 | static int |
|
901 | 901 | match_env_var(char *env_entry, struct env_rule *r) |
|
902 | 902 | { |
|
903 | 903 | if (strncmp(env_entry, r->var, r->var_len)) |
|
904 | 904 | return 0; |
|
905 | 905 | return (env_entry[r->var_len] == '='); |
|
906 | 906 | } |
|
907 | 907 | |
|
908 | 908 | static void |
|
909 | 909 | apply_env_rule(char **env, int *env_sizep, struct env_rule *r) |
|
910 | 910 | { |
|
911 | 911 | // First remove the variable if already set |
|
912 | 912 | int pos = 0; |
|
913 | 913 | while (pos < *env_sizep && !match_env_var(env[pos], r)) |
|
914 | 914 | pos++; |
|
915 | 915 | if (pos < *env_sizep) |
|
916 | 916 | { |
|
917 | 917 | (*env_sizep)--; |
|
918 | 918 | env[pos] = env[*env_sizep]; |
|
919 | 919 | env[*env_sizep] = NULL; |
|
920 | 920 | } |
|
921 | 921 | |
|
922 | 922 | // What is the new value? |
|
923 | 923 | char *new; |
|
924 | 924 | if (r->val) |
|
925 | 925 | { |
|
926 | 926 | if (!r->val[0]) |
|
927 | 927 | return; |
|
928 | 928 | new = xmalloc(r->var_len + 1 + strlen(r->val) + 1); |
|
929 | 929 | sprintf(new, "%s=%s", r->var, r->val); |
|
930 | 930 | } |
|
931 | 931 | else |
|
932 | 932 | { |
|
933 | 933 | pos = 0; |
|
934 | 934 | while (environ[pos] && !match_env_var(environ[pos], r)) |
|
935 | 935 | pos++; |
|
936 | 936 | if (!(new = environ[pos])) |
|
937 | 937 | return; |
|
938 | 938 | } |
|
939 | 939 | |
|
940 | 940 | // Add it at the end of the array |
|
941 | 941 | env[(*env_sizep)++] = new; |
|
942 | 942 | env[*env_sizep] = NULL; |
|
943 | 943 | } |
|
944 | 944 | |
|
945 | 945 | static char ** |
|
946 | 946 | setup_environment(void) |
|
947 | 947 | { |
|
948 | 948 | // Link built-in rules with user rules |
|
949 | 949 | for (int i=ARRAY_SIZE(default_env_rules)-1; i >= 0; i--) |
|
950 | 950 | { |
|
951 | 951 | default_env_rules[i].next = first_env_rule; |
|
952 | 952 | first_env_rule = &default_env_rules[i]; |
|
953 | 953 | } |
|
954 | 954 | |
|
955 | 955 | // Scan the original environment |
|
956 | 956 | char **orig_env = environ; |
|
957 | 957 | int orig_size = 0; |
|
958 | 958 | while (orig_env[orig_size]) |
|
959 | 959 | orig_size++; |
|
960 | 960 | |
|
961 | 961 | // For each rule, reserve one more slot and calculate length |
|
962 | 962 | int num_rules = 0; |
|
963 | 963 | for (struct env_rule *r = first_env_rule; r; r=r->next) |
|
964 | 964 | { |
|
965 | 965 | num_rules++; |
|
966 | 966 | r->var_len = strlen(r->var); |
|
967 | 967 | } |
|
968 | 968 | |
|
969 | 969 | // Create a new environment |
|
970 | 970 | char **env = xmalloc((orig_size + num_rules + 1) * sizeof(char *)); |
|
971 | 971 | int size; |
|
972 | 972 | if (pass_environ) |
|
973 | 973 | { |
|
974 | 974 | memcpy(env, environ, orig_size * sizeof(char *)); |
|
975 | 975 | size = orig_size; |
|
976 | 976 | } |
|
977 | 977 | else |
|
978 | 978 | size = 0; |
|
979 | 979 | env[size] = NULL; |
|
980 | 980 |
@@ -1,133 +1,133 | |||
|
1 | 1 | #!/usr/bin/env ruby |
|
2 | 2 | |
|
3 | 3 | CORRECT_MARK = 'P' |
|
4 | 4 | INCORRECT_MARK = '-' |
|
5 | 5 | TIMEOUT_MARK = 'T' |
|
6 | 6 | RUN_ERROR_MARK = 'x' |
|
7 | 7 | |
|
8 | 8 | def log(str='') |
|
9 | 9 | if ENV['TALKATIVE']!=nil |
|
10 | 10 | puts str |
|
11 | 11 | end |
|
12 | 12 | if ENV['GRADER_LOGGING']!=nil |
|
13 | 13 | log_fname = ENV['GRADER_LOGGING'] |
|
14 | 14 | fp = File.open(log_fname,"a") |
|
15 | 15 | fp.puts("grade: #{Time.new.strftime("%H:%M")} #{str}") |
|
16 | 16 | fp.close |
|
17 | 17 | end |
|
18 | 18 | end |
|
19 | 19 | |
|
20 | 20 | def char_comment(comment) |
|
21 | 21 | if comment =~ /[Ii]ncorrect/ |
|
22 | 22 | INCORRECT_MARK |
|
23 | 23 | elsif comment =~ /[Cc]orrect/ |
|
24 | 24 | CORRECT_MARK |
|
25 | 25 | elsif comment =~ /[Tt]ime/ |
|
26 | 26 | TIMEOUT_MARK |
|
27 | 27 | elsif res = /^[Cc]omment:(.*)$/.match(comment) |
|
28 | 28 | res[1] |
|
29 | 29 | else |
|
30 | 30 | RUN_ERROR_MARK # these are run time errors |
|
31 | 31 | end |
|
32 | 32 | end |
|
33 | 33 | |
|
34 | 34 | def extract_time(t) |
|
35 | 35 | puts "TIME: #{t}" |
|
36 |
- if (result=/^(.*)r(.*)u(.*)s(.*) |
|
|
36 | + if (result=/^(.*)r(.*)u(.*)s(.*)kbytes/.match(t)) | |
|
37 | 37 | {:real => result[1], :user => result[2], :sys => result[3], :mem => result[4]} |
|
38 | 38 | else |
|
39 | 39 | #{:real => 0, :user => 0, :sys => 0} |
|
40 | 40 | #puts "ERROR READING RUNNING TIME: #{t}" |
|
41 | 41 | raise "Error reading running time: #{t}" |
|
42 | 42 | end |
|
43 | 43 | end |
|
44 | 44 | |
|
45 | 45 | problem_home = ENV['PROBLEM_HOME'] |
|
46 | 46 | require "#{problem_home}/script/test_dsl.rb" |
|
47 | 47 | load "#{problem_home}/test_cases/all_tests.cfg" |
|
48 | 48 | problem = Problem.get_instance |
|
49 | 49 | |
|
50 | 50 | if problem.well_formed? == false |
|
51 | 51 | log "The problem specification is not well formed." |
|
52 | 52 | exit(127) |
|
53 | 53 | end |
|
54 | 54 | |
|
55 | 55 | all_score = 0 |
|
56 | 56 | all_comment = '' |
|
57 | 57 | peak_memory = -1 |
|
58 | 58 | max_runtime = -1 |
|
59 | 59 | (1..(problem.runs.length-1)).each do |k| |
|
60 | 60 | log "grade run #{k}" |
|
61 | 61 | run = problem.runs[k] |
|
62 | 62 | run_score = nil |
|
63 | 63 | run_comment = '' |
|
64 | 64 | run_comment_short = '' |
|
65 | 65 | run.tests.each do |test_num| |
|
66 | 66 | result_file_name = "#{test_num}/result" |
|
67 | 67 | if not File.exists?(result_file_name) |
|
68 | 68 | run_comment += "result file for test #{test_num} not found\n" |
|
69 | 69 | run_comment_short += RUN_ERROR_MARK |
|
70 | 70 | log "Cannot find the file #{test_num}/result!" |
|
71 | 71 | else |
|
72 | 72 | result_file = File.new(result_file_name, "r") |
|
73 | 73 | result_file_lines = result_file.readlines |
|
74 | 74 | if result_file_lines.length>=3 |
|
75 | 75 | current_run_score = result_file_lines[1].to_i |
|
76 | 76 | run_comment += result_file_lines[0] |
|
77 | 77 | run_comment_short += char_comment(result_file_lines[0].chomp) |
|
78 | 78 | |
|
79 | 79 | #update max runtime & memory |
|
80 | 80 | run_stat = extract_time result_file_lines[2] |
|
81 | 81 | peak_memory = [peak_memory,run_stat[:mem].to_i].max |
|
82 | 82 | max_runtime = [max_runtime,run_stat[:user].to_f + run_stat[:sys].to_f].max |
|
83 | 83 | else |
|
84 | 84 | current_run_score = 0 |
|
85 | 85 | run_comment += "result file for test #{test_num} error\n" |
|
86 | 86 | run_comment_short += RUN_ERROR_MARK |
|
87 | 87 | log "Error in #{test_num}/result!" |
|
88 | 88 | end |
|
89 | 89 | |
|
90 | 90 | # the score of this run should be the minimum of the score for |
|
91 | 91 | # each test case |
|
92 | 92 | if (run_score==nil) or (run_score>current_run_score) |
|
93 | 93 | run_score = current_run_score |
|
94 | 94 | end |
|
95 | 95 | result_file.close |
|
96 | 96 | end |
|
97 | 97 | end |
|
98 | 98 | |
|
99 | 99 | run_result_file = File.new("result-#{k}", "w") |
|
100 | 100 | run_result_file.write run_score |
|
101 | 101 | run_result_file.write "\n" |
|
102 | 102 | run_result_file.close |
|
103 | 103 | |
|
104 | 104 | run_comment_file = File.new("comment-#{k}", "w") |
|
105 | 105 | run_comment_file.write "#{run_comment}\n" |
|
106 | 106 | run_comment_file.close |
|
107 | 107 | |
|
108 | 108 | all_score = all_score + run_score |
|
109 | 109 | |
|
110 | 110 | # append comment for test run with many test cases |
|
111 | 111 | if run.tests.length > 1 |
|
112 | 112 | run_comment_short = '[' + run_comment_short + ']' |
|
113 | 113 | end |
|
114 | 114 | all_comment += run_comment_short |
|
115 | 115 | end |
|
116 | 116 | |
|
117 | 117 | result_file = File.new("result", "w") |
|
118 | 118 | result_file.write all_score |
|
119 | 119 | result_file.write "\n" |
|
120 | 120 | result_file.close |
|
121 | 121 | |
|
122 | 122 | comment_file = File.new("comment", "w") |
|
123 | 123 | comment_file.write "#{all_comment}\n" |
|
124 | 124 | comment_file.close |
|
125 | 125 | |
|
126 | 126 | |
|
127 | 127 | File.open("run_stat","w") do |file| |
|
128 | 128 | file.puts max_runtime |
|
129 | 129 | file.puts peak_memory |
|
130 | 130 | end |
|
131 | 131 | |
|
132 | 132 | log "score = #{all_score}\ncomment = #{all_comment}" |
|
133 | 133 | log "max_runtime = #{max_runtime}\npeak_memory = #{peak_memory}" |
You need to be logged in to leave comments.
Login now