cafe-grader-judge-scripts

Commit r128:e73e7e444f1c public

parent | child
Description:
added redefine to protect against forbidded functions
Commit status:
[Not Reviewed]
References:
default
Diff options:
Raw Diff | Patch Diff | Download Diff | Show whitespace | Increase context
Comments:
0 Commit comments 0 Inline Comments
Unresolved TODOs:
There are no unresolved TODOs
Side by Side Unified
Expand All Files Collapse All Files Wide Mode Diff
Add another comment
Browse Files

r128:e73e7e444f1c - - 1 file changed: 19 inserted, 2 deleted

Show file before | Show file after | Raw diff | Download diff | Show whitespace Increase context Show commentsHide comments
@@ -1,41 +1,42
1 #!/usr/bin/ruby
1 #!/usr/bin/ruby
2
2
3 require 'fileutils'
3 require 'fileutils'
4
4
5 ##############################
5 ##############################
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
30 C_OPTIONS = "-O2 -s -static -std=c99 -DCONTEST -lm -Wall"
31 C_OPTIONS = "-O2 -s -static -std=c99 -DCONTEST -lm -Wall"
31 CPLUSPLUS_OPTIONS = "-O2 -s -static -DCONTEST -lm -Wall"
32 CPLUSPLUS_OPTIONS = "-O2 -s -static -DCONTEST -lm -Wall"
32 PASCAL_OPTIONS = "-O1 -XS -dCONTEST"
33 PASCAL_OPTIONS = "-O1 -XS -dCONTEST"
33
34
34 # Check for the correct number of arguments. Otherwise, print usage.
35 # Check for the correct number of arguments. Otherwise, print usage.
35 if ARGV.length == 0 or ARGV.length > 4
36 if ARGV.length == 0 or ARGV.length > 4
36 puts "Usage: compile <language> [<source-file>] [<output-file>] [<message-file>]"
37 puts "Usage: compile <language> [<source-file>] [<output-file>] [<message-file>]"
37 puts
38 puts
38 puts "<source-file> is defaulted to \"source\"."
39 puts "<source-file> is defaulted to \"source\"."
39 puts "<output-file> is defaulted to \"a.out\"."
40 puts "<output-file> is defaulted to \"a.out\"."
40 puts "<message-file> is defaulted to \"compiler_message\"."
41 puts "<message-file> is defaulted to \"compiler_message\"."
41 puts
42 puts
@@ -60,54 +61,70
60 talk "#{param_name}: #{params[param_name]}"
61 talk "#{param_name}: #{params[param_name]}"
61 end
62 end
62
63
63 # Remove any remaining output files or message files.
64 # Remove any remaining output files or message files.
64 if FileTest.exists? params[:output_file]
65 if FileTest.exists? params[:output_file]
65 FileUtils.rm(params[:output_file])
66 FileUtils.rm(params[:output_file])
66 end
67 end
67 if FileTest.exists? params[:message_file]
68 if FileTest.exists? params[:message_file]
68 FileUtils.rm(params[:message_file])
69 FileUtils.rm(params[:message_file])
69 end
70 end
70
71
71 # Check if the source file exists before attempt compiling.
72 # Check if the source file exists before attempt compiling.
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 #{params[:output_file]} #{C_OPTIONS} 2> #{params[:message_file]}"
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 #{params[:output_file]} #{CPLUSPLUS_OPTIONS} 2> #{params[:message_file]}"
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
105 exit(127)
122 exit(127)
106 end
123 end
107
124
108 # Report success or failure.
125 # Report success or failure.
109 if FileTest.exists? params[:output_file]
126 if FileTest.exists? params[:output_file]
110 talk "Compilation was successful!"
127 talk "Compilation was successful!"
111 else
128 else
112 talk "ERROR: Something was wrong during the compilation!"
129 talk "ERROR: Something was wrong during the compilation!"
113 end
130 end
You need to be logged in to leave comments. Login now