Description:
added redefine to protect against forbidded functions
Commit status:
[Not Reviewed]
References:
Diff options:
Comments:
0 Commit comments
0 Inline Comments
Unresolved TODOs:
There are no unresolved TODOs
r128:e73e7e444f1c - - 1 file changed: 19 inserted, 2 deleted
@@ -6,24 +6,25 | |||||
|
6 | # |
|
6 | # |
|
7 | # Standard Compile Script |
|
7 | # Standard Compile Script |
|
8 | # |
|
8 | # |
|
9 | # Supported compilers: |
|
9 | # Supported compilers: |
|
10 | # gcc, g++, and fpc. |
|
10 | # gcc, g++, and fpc. |
|
11 | # |
|
11 | # |
|
12 | ############################## |
|
12 | ############################## |
|
13 |
|
13 | ||
|
14 | def talk(msg) |
|
14 | def talk(msg) |
|
15 | if ENV['TALKATIVE']!=nil |
|
15 | if ENV['TALKATIVE']!=nil |
|
16 | puts str |
|
16 | puts str |
|
17 | end |
|
17 | end |
|
|
18 | + | ||
|
18 | if ENV['GRADER_LOGGING']!=nil |
|
19 | if ENV['GRADER_LOGGING']!=nil |
|
19 | log_fname = ENV['GRADER_LOGGING'] |
|
20 | log_fname = ENV['GRADER_LOGGING'] |
|
20 | fp = File.open(log_fname,"a") |
|
21 | fp = File.open(log_fname,"a") |
|
21 | fp.puts("run: #{Time.new.strftime("%H:%M")} #{msg}") |
|
22 | fp.puts("run: #{Time.new.strftime("%H:%M")} #{msg}") |
|
22 | fp.close |
|
23 | fp.close |
|
23 | end |
|
24 | end |
|
24 | end |
|
25 | end |
|
25 |
|
26 | ||
|
26 | C_COMPILER = "gcc" |
|
27 | C_COMPILER = "gcc" |
|
27 | CPLUSPLUS_COMPILER = "g++" |
|
28 | CPLUSPLUS_COMPILER = "g++" |
|
28 | PASCAL_COMPILER = "fpc" |
|
29 | PASCAL_COMPILER = "fpc" |
|
29 |
|
30 | ||
@@ -72,33 +73,49 | |||||
|
72 | if !FileTest.exists? params[:source_file] |
|
73 | if !FileTest.exists? params[:source_file] |
|
73 | talk("ERROR: The source file does not exist!") |
|
74 | talk("ERROR: The source file does not exist!") |
|
74 | open(params[:message_file],"w") do |f| |
|
75 | open(params[:message_file],"w") do |f| |
|
75 | f.puts "ERROR: The source file did not exist." |
|
76 | f.puts "ERROR: The source file did not exist." |
|
76 | end |
|
77 | end |
|
77 | exit(127) |
|
78 | exit(127) |
|
78 | end |
|
79 | end |
|
79 |
|
80 | ||
|
80 | if params[:prog_lang]=='cpp': |
|
81 | if params[:prog_lang]=='cpp': |
|
81 | params[:prog_lang] = 'c++' |
|
82 | params[:prog_lang] = 'c++' |
|
82 | end |
|
83 | end |
|
83 |
|
84 | ||
|
|
85 | + forbidden_functions = ['system', 'fopen', 'freopen', 'open', 'remove', 'rename'] | ||
|
|
86 | + redefine_list = [] | ||
|
|
87 | + forbidden_count = 0 | ||
|
|
88 | + forbidden_key = (rand(8999) + 1000).to_s | ||
|
|
89 | + forbidden_functions.each do |f| | ||
|
|
90 | + redefine_list << "-D#{f}=forbidden#{forbidden_key}#{forbidden_count}" | ||
|
|
91 | + forbidden_count += 1 | ||
|
|
92 | + end | ||
|
|
93 | + redefine_str = redefine_list.join ' ' | ||
|
|
94 | + puts redefine_str | ||
|
|
95 | + | ||
|
84 | # Compile. |
|
96 | # Compile. |
|
85 | case params[:prog_lang] |
|
97 | case params[:prog_lang] |
|
86 |
|
98 | ||
|
87 | when "c" |
|
99 | when "c" |
|
88 |
- command = "#{C_COMPILER} #{params[:source_file]} -o |
|
100 | + command = "#{C_COMPILER} -E #{params[:source_file]} -o source_prep.c" |
|
|
101 | + system(command) | ||
|
|
102 | + command = "#{C_COMPILER} source_prep.c #{redefine_str} -o #{params[:output_file]} #{C_OPTIONS} 2> #{params[:message_file]}" | ||
|
|
103 | + puts command | ||
|
89 | system(command) |
|
104 | system(command) |
|
90 |
|
105 | ||
|
91 | when "c++" |
|
106 | when "c++" |
|
92 |
- command = "#{CPLUSPLUS_COMPILER} #{params[:source_file]} -o |
|
107 | + command = "#{CPLUSPLUS_COMPILER} -E #{params[:source_file]} -o source_prep.cpp" |
|
|
108 | + system(command) | ||
|
|
109 | + command = "#{CPLUSPLUS_COMPILER} #{redefine_str} #{params[:source_file]} -o #{params[:output_file]} #{CPLUSPLUS_OPTIONS} 2> #{params[:message_file]}" | ||
|
93 | system(command) |
|
110 | system(command) |
|
94 |
|
111 | ||
|
95 | when "pas" |
|
112 | when "pas" |
|
96 | command = "#{PASCAL_COMPILER} #{params[:source_file]} -ooutpas #{PASCAL_OPTIONS} > #{params[:message_file]}" |
|
113 | command = "#{PASCAL_COMPILER} #{params[:source_file]} -ooutpas #{PASCAL_OPTIONS} > #{params[:message_file]}" |
|
97 | system(command) |
|
114 | system(command) |
|
98 | FileUtils.mv("output", params[:output_file]) |
|
115 | FileUtils.mv("output", params[:output_file]) |
|
99 |
|
116 | ||
|
100 | else |
|
117 | else |
|
101 | talk("ERROR: Invalid language specified!") |
|
118 | talk("ERROR: Invalid language specified!") |
|
102 | open(params[:message_file],"w") do |f| |
|
119 | open(params[:message_file],"w") do |f| |
|
103 | f.puts "ERROR: Invalid language specified!" |
|
120 | f.puts "ERROR: Invalid language specified!" |
|
104 | end |
|
121 | end |
You need to be logged in to leave comments.
Login now