Description:
added redefine to protect against forbidded functions
Commit status:
[Not Reviewed]
References:
Comments:
0 Commit comments 0 Inline Comments
Unresolved TODOs:
There are no unresolved TODOs
Add another comment

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 #{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
You need to be logged in to leave comments. Login now