summaryrefslogtreecommitdiff
path: root/yaksh/cpp_code_evaluator.py
blob: becf37189b2b7eaa28c5a82ba67078ae50f13984 (plain)
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
127
128
129
130
131
132
133
#!/usr/bin/env python
import traceback
import pwd
import os
from os.path import join, isfile
import subprocess
import importlib

# local imports
from code_evaluator import CodeEvaluator


class CppCodeEvaluator(CodeEvaluator):
    """Tests the C code obtained from Code Server"""
    def setup(self):
        super(CppCodeEvaluator, self).setup()
        self.submit_code_path = self.create_submit_code_file('submit.c')
        self.compiled_user_answer = None
        self.compiled_test_code = None

    def teardown(self):
        super(CppCodeEvaluator, self).teardown()
        # Delete the created file.
        os.remove(self.submit_code_path)

    def set_file_paths(self):
        user_output_path = os.getcwd() + '/output'
        ref_output_path = os.getcwd() + '/executable'

        return user_output_path, ref_output_path

    def get_commands(self, clean_ref_code_path, user_output_path,
                                 ref_output_path):
        compile_command = 'g++  {0} -c -o {1}'.format(self.submit_code_path,
                                                 user_output_path)
        compile_main = 'g++ {0} {1} -o {2}'.format(clean_ref_code_path,
                                                user_output_path,
                                                ref_output_path)
        return compile_command, compile_main

    def compile_code(self,  user_answer, test_case):
        if self.compiled_user_answer and self.compiled_test_code:
            return None
        else:
            ref_code_path = test_case
            clean_ref_code_path, clean_test_case_path = self._set_test_code_file_path(ref_code_path)

            if not isfile(clean_ref_code_path):
                return False, "No file at %s or Incorrect path" % clean_ref_code_path
            if not isfile(self.submit_code_path):
                return False, 'No file at %s or Incorrect path' % self.submit_code_path

            self.write_to_submit_code_file(self.submit_code_path, user_answer)
            self.user_output_path, self.ref_output_path = self.set_file_paths()
            self.compile_command, self.compile_main = self.get_commands(clean_ref_code_path, self.user_output_path, self.ref_output_path)
            self.compiled_user_answer = self._run_command(self.compile_command,
                                     shell=True,
                                     stdout=subprocess.PIPE,
                                     stderr=subprocess.PIPE)

            self.compiled_test_code = self._run_command(self.compile_main,
                                     shell=True,
                                     stdout=subprocess.PIPE,
                                     stderr=subprocess.PIPE)


            return self.compiled_user_answer, self.compiled_test_code

    def check_code(self, user_answer, test_case):
            """ Function validates student code using instructor code as
            reference.The first argument ref_code_path, is the path to
            instructor code, it is assumed to have executable permission.
            The second argument submit_code_path, is the path to the student
            code, it is assumed to have executable permission.

            Returns
            --------

            returns (True, "Correct answer") : If the student function returns
            expected output when called by reference code.

            returns (False, error_msg): If the student function fails to return
            expected output when called by reference code.

            Returns (False, error_msg): If mandatory arguments are not files or
            if the required permissions are not given to the file(s).

            """
            success = False
            proc, stdnt_out, stdnt_stderr = self.compiled_user_answer
            stdnt_stderr = self._remove_null_substitute_char(stdnt_stderr)

            # Only if compilation is successful, the program is executed
            # And tested with testcases
            if stdnt_stderr == '':
                proc, main_out, main_err = self.compiled_test_code
                main_err = self._remove_null_substitute_char(main_err)

                if main_err == '':
                    ret = self._run_command([self.ref_output_path], stdin=None,
                                            stdout=subprocess.PIPE,
                                            stderr=subprocess.PIPE)
                    proc, stdout, stderr = ret
                    if proc.returncode == 0:
                        success, err = True, "Correct answer"
                    else:
                        err = stdout + "\n" + stderr
                    os.remove(self.ref_output_path)
                else:
                    err = "Error:"
                    try:
                        error_lines = main_err.splitlines()
                        for e in error_lines:
                            if ':' in e:
                                err = err + "\n" + e.split(":", 1)[1]
                            else:
                                err = err + "\n" + e
                    except:
                            err = err + "\n" + main_err
                os.remove(self.user_output_path)
            else:
                err = "Compilation Error:"
                try:
                    error_lines = stdnt_stderr.splitlines()
                    for e in error_lines:
                        if ':' in e:
                            err = err + "\n" + e.split(":", 1)[1]
                        else:
                            err = err + "\n" + e
                except:
                    err = err + "\n" + stdnt_stderr

            return success, err