Bugfix (se corrige bien el bug del from).

[software/sercom-old.git] / src / sc_test

diff --git a/src/sc_test b/src/sc_test
index 24d23698e7b75aac0318c8c2e880c5404bc697ba..3ed7b6c4751d9d51b27fa0a7f41ec9877e97e573 100755 (executable)
--- a/src/sc_test
+++ b/src/sc_test
@@ -1,298 +1,454 @@
-#!/usr/bin/php
-<?php // vim: set binary noeol et sw=4 sts=4:
+#!/usr/bin/env python2.4
+# -*- encoding: iso-8859-1 -*-
+# vim: set et sw=4 sts=4 :

 
 

-require_once 'T/general.php';
+# Módulos estándar
+import os
+import pwd
+import time
+import signal
+import glob
+import shutil
+import datetime
+import subprocess
+import email.MIMEMultipart
+# Módulos locales
+import sercom
+import sercom.sqlo
+from sercom.sqlo import *

 
 

-define('R_ERR', 0);
-define('R_OK',  1);
+class secure_process:
+    def __init__(self, chroot, uid, gid, cpu):
+        self.chroot = chroot
+        self.uid = uid
+        self.gid = gid
+        self.cpu = cpu
+    def __call__(self):
+        from os import chroot, setuid, setgid
+        import resource
+        chroot(self.chroot)
+        setgid(self.gid)
+        setuid(self.uid)
+        x2 = lambda val: (val, val) # Devuelve una tupla con val 2 veces
+        resource.setrlimit(resource.RLIMIT_AS, x2(20*1024*1024))
+        resource.setrlimit(resource.RLIMIT_CORE, x2(0))
+        resource.setrlimit(resource.RLIMIT_CPU, x2(self.cpu))
+        resource.setrlimit(resource.RLIMIT_DATA, x2(20*1024*1024))
+        resource.setrlimit(resource.RLIMIT_FSIZE, x2(20*1024*1024)) #XXX Obtener de archivos esperados?
+        #resource.setrlimit(resource.RLIMIT_LOCKS, x2(100)) XXX NO EXISTE EN python
+        resource.setrlimit(resource.RLIMIT_MEMLOCK, x2(100))
+        resource.setrlimit(resource.RLIMIT_NOFILE, x2(100))
+        resource.setrlimit(resource.RLIMIT_NPROC, x2(0))
+        # Tratamos de forzar un sync para que entre al sleep del padre
+        time.sleep(0)

 
 

-$LOGLEVEL = DEBUG;
+def sigchld(signum, frame):
+    """Signal handler para SIGCHILD."""
+    global hijo_muerto, log
+    log.debug('Murió el hijo')
+    hijo_muerto = True

 
 

-$gconf = $CONF['general'];
+def sigterm(signum, frame):
+    """Signal handler para SIGTERM y SIGINT."""
+    global continuar, log
+    continuar = False
+    log.debug('Señal %d recibida', signum)
+    log.info('Finalizado')

 
 

-// Obtengo id de usuario con el cual ejecutar las pruebas.
-if (!($usrinfo = @posix_getpwnam($gconf['user']))) {
-    logserr("No existe el usuario '{$gconf['user']}'");
-    exit(10); // salgo con error
-}
-$UID = $usrinfo['uid'];
-$GID = $usrinfo['gid'];
-/*if (!@posix_seteuid($usrinfo['uid'])) {
-    logserr("No se puede cambiar el uid efectivo a '{$usrinfo['uid']}'");
-    exit(11); // salgo con error
-}*/
+def compilar(intento, mail):
+    global log
+    # Busco makefile
+    makefile = os.path.join(intento.entrega.ejercicio.path, 'Makefile')
+    # Compilo
+    log.debug('Ejecutando: make -f %s', makefile)
+    intento.inicioCompila = datetime.datetime.now()
+    stderr = file(os.path.join(intento.path, 'make.out'), 'w')
+    make = subprocess.Popen(('make', '-f', makefile), stderr=stderr,
+        cwd=intento.path)
+    make.wait()
+    intento.finCompila = datetime.datetime.now()
+    log.debug('Fin del comando: make -f %s', makefile)
+    # Verifico compilación
+    stderr = file(os.path.join(intento.path, 'make.out')).read()
+    intento.compila = not make.returncode
+    msg = 'Compilación: '
+    if intento.compila and not stderr:
+        msg += 'BIEN'
+    elif intento.compila:
+        msg += 'CON ADVERTENCIAS'
+    else:
+        msg += 'ERROR (código de retorno: %d)' % make.returncode
+    mail.body += msg
+    if stderr:
+        msg += '''
+Salida:
+------------------------------------------------------------------------
+%s
+------------------------------------------------------------------------

 
 

-$intento = new T_Intento;
+''' % stderr
+        mail.body += "\n(se adjunta salida, archivo 'make.stderr')\n"
+        mail.attachText(stderr, 'make.stderr')
+    mail.body += '\n\n'
+    intento.observaciones = msg + '\n\n'
+    log.debug(msg)

 
 

-// Sin cesar.
-while (1) {
-    if (!($mail = $intento->proximo_a_probar())) {
-        logs('No hay intento para probar', DEBUG);
-        sleep($gconf['intervalo']);
-        continue;
-    }
-    logs('Nuevo intento a probar (' . $intento->to_line() . ')');
+def preparar(intento, caso_de_prueba):
+    # Creo chroot - TODO copiarlo de algún lado donde ande el valgrind?
+    os.makedirs(intento.chrootPath(caso_de_prueba))
+    shutil.copy(os.path.join(intento.path, 'tp'),
+        os.path.join(intento.chrootPath(caso_de_prueba), 'tp'))
+    for f in caso_de_prueba.archivosEntrada:
+        shutil.copy(os.path.join(caso_de_prueba.pathEntradas, f),
+            os.path.join(intento.chrootPath(caso_de_prueba), f))

 
 

-    $intento_dir = "{$gconf['data_dir']}/" . $intento->path();
-    $entrega_dir = "{$gconf['data_dir']}/" . $intento->base_path('entregas');
+def limpiar(intento, caso_de_prueba):
+    # Borro chroot entero
+    #shutil.rmtree(intento.chrootPath)
+    os.unlink(os.path.join(intento.chrootPath(caso_de_prueba), 'tp'))

 
 

-    if (is_readable("$entrega_dir/Makefile")) {
-        $makefile = "$entrega_dir/Makefile";
-//XXX        @copy("$entrega_dir/Makefile", "$intento_dir/Makefile")
-//            or logserr("Error al copiar Makefile (de '$entrega_dir/Makefile' a '$intento_dir/Makefile'");
-    // Fallback
-    } elseif (is_readable("{$gconf['data_dir']}/Makefile")) {
-        $makefile = "{$gconf['data_dir']}/Makefile";
-//XXX    if (!@copy("{$gconf['data_dir']}/Makefile", "$intento_dir/Makefile")) {
-    } else {
-        enviar_respuesta_error_log($mail, 'No hay un Makefile disponible', $intento);
-        continue;
-    }
+def probar(intento, caso_de_prueba, mail):
+    def diff(prueba, mail, orig, new, name, origname='alumno', newname='catedra', longname=None):
+        from difflib import unified_diff, HtmlDiff
+        if longname is None: longname = name
+        diff = ''.join(list(unified_diff(orig, new, fromfile=name+'.'+origname, tofile=name+'.'+newname)))
+        nota = ''
+        if diff:
+            prueba.pasada = False
+            nota += "* %s no coincide con lo esperado (archivo '%s.diff').\n" % (longname, name)
+            nota += ('-' * 72) + '\n' + diff + ('-' * 72) + '\n'
+            if not prueba.casoDePrueba.privado:
+                mail.attachText(diff, name + '.diff')
+                diff = HtmlDiff().make_file(orig, new, fromdesc=name+'.'+origname, todesc=name+'.'+newname, context=True, numlines=3)
+                mail.attachText(diff, name + '.diff.html', 'html')
+        return nota
+    def usa_stdin():
+        return os.path.exists(os.path.join(caso_de_prueba.path, 'stdin'))
+    def usa_stdout():
+        return os.path.exists(os.path.join(caso_de_prueba.path, 'stdout'))
+    def usa_stderr():
+        return os.path.exists(os.path.join(caso_de_prueba.path, 'stderr'))

 
 

-    $currdir = getcwd();
-    if (!@chdir($intento_dir)) {
-        enviar_respuesta_error_log($mail, 'Error al cambiar al directorio del tp', $intento);
-        continue;
-    }
-    logs("Cambio de directorio '$currdir' -> '$intento_dir'", DEBUG);
-    logs('Ejecutando el comando: make -f '.escapeshellarg($makefile), DEBUG);
-    if (exec_get_info('make -f '.escapeshellarg($makefile), $ret, $err, $out)) {
-        if ($ret) {
-            logs('Error al compilar');
-            logs("Código de retorno: $ret, mensaje: $err)", DEBUG);
-            //XXX $intento->informar_compilacion(false);
-            enviar_respuesta(R_ERR, $mail, "ERROR AL COMPILAR!\n\n$err\n\nCódigo de retorno: $ret\n", $intento);
-            continue;
-        } else {
-            logs('Compilado OK');
-            //XXX $intento->informar_compilacion(true);
-            // TODO mail acumulativo
-            // enviar_respuesta(R_OK, $mail, 'El intento fue compilado con éxito!', $intento);
-        }
-    } else {
-        //XXX $intento->informar_compilacion(false);
-        enviar_respuesta_error_log($mail, 'No se pudo ejecutar make', $intento);
-        continue;
-    }
-    if (!@mkdir('chroot')) {
-        enviar_respuesta_error_log($mail, 'Error al crear directorio para chroot', $intento);
-        continue;
-    }
-    if (!@rename('tp', 'chroot/tp')) {
-        enviar_respuesta_error_log($mail, 'Error al mover el tp al chroot', $intento);
-        continue;
-    }
-    if (!@copy("{$gconf['data_dir']}/redir", 'chroot/redir')) {
-        enviar_respuesta_error_log($mail, 'Error al copiar redireccionador al chroot', $intento);
-        continue;
-    }
-    if ($err = chmod_own_grp('chroot', 02770, $UID, $GID)) {
-        enviar_respuesta_error_log($mail, $err, $intento);
-        continue;
-    }
-    if ($err = chmod_own_grp('chroot/tp', 0550, $UID, $GID)) {
-        enviar_respuesta_error_log($mail, $err, $intento);
-        continue;
-    }
-    if ($err = chmod_own_grp('chroot/redir', 0550, 0, 0)) {
-        enviar_respuesta_error_log($mail, $err, $intento);
-        continue;
+    # Cosas útiles
+    global log, conn, conf, uid, gid
+    # Para manejo de SIGCHLD
+    global hijo_muerto
+    hijo_muerto = False
+    # Obtengo datos útiles del caso de prueba
+    tiempo_cpu = caso_de_prueba.tiempoCpu
+    if tiempo_cpu is None:
+        tiempo_cpu = conf.get('general', 'tiempo_cpu')
+    tiempo_cpu = int(tiempo_cpu)
+    # Creo prueba nueva
+    prueba = Prueba(intento=intento, casoDePrueba=caso_de_prueba,
+        inicio=datetime.datetime.now(), connection=conn)
+    log.debug('Iniciando prueba: %s', prueba)
+    log.debug('Caso de prueba: %s', prueba.casoDePrueba)
+    # Abro archivos para fds básicos
+    options = {
+        'cwd': intento.chrootPath(caso_de_prueba),
+        'close_fds': True,
+        'preexec_fn': secure_process(intento.chrootPath(caso_de_prueba), uid, gid, tiempo_cpu),
+        'stdin': None,
+        'stdout': None,
+        'stderr': None,

     }
     }

+    if usa_stdin():
+        options['stdin'] = file(os.path.join(caso_de_prueba.path, 'stdin'), 'r')
+    if usa_stdout():
+        stdout_fn = intento.chrootPath(caso_de_prueba) + '.stdout'
+        options['stdout'] = file(stdout_fn,  'w')
+    if usa_stderr():
+        stderr_fn = intento.chrootPath(caso_de_prueba) + '.stderr'
+        options['stderr'] = file(stderr_fn,  'w')
+    # Ejecuto programa
+    params = ['/tp']
+    if caso_de_prueba.parametros:
+        params += params2seq(caso_de_prueba.parametros)
+    log.debug('Ejecutando %s', ' '.join(params))
+    os.seteuid(0) # Dios! (para chroot)
+    try:
+        proc = subprocess.Popen(params, **options)
+    except Exception, e: # FIXME poner en el manejo de exceptiones estandar
+        try:
+            log.debug('ERROR! Trace del hijo: %s', e.child_traceback)
+            raise
+        except:
+            raise
+    os.seteuid(uid) # Mortal de nuevo
+    if not hijo_muerto: # Recibido por el sigchld, para saber si murió
+        time.sleep(tiempo_cpu) # Controlo que no tarde mucho
+    # Si el proceso sigue andando lo tenemos que matar
+    if not hijo_muerto:
+        log.debug('La prueba tardó más del tiempo permitido (%d segundos)',
+            tiempo_cpu)
+        os.seteuid(0) # Dios! (corre como Dios, hay que matarlo como Dios)
+        os.kill(proc.pid, signal.SIGKILL)
+        os.seteuid(uid) # Mortal de nuevo
+        log.debug('Prueba cancelada (kill)')
+        proc.wait()
+        prueba.observaciones = 'Excedió el límite de tiempo de ejecución ' \
+            '(%d seg)' % tiempo_cpu
+        prueba.fin = datetime.datetime.now()
+        prueba.pasada = False
+        mail.agregarResultado(prueba)
+        return prueba
+    proc.wait() # Para que no queden zombies
+    prueba.fin = datetime.datetime.now()
+    # Salió con una señal?
+    if proc.returncode < 0:
+        sig = -proc.returncode
+        # Conversor de nro de señal a string
+        sigs = {}
+        for s in [s for s in dir(signal) if s.startswith('SIG') and s.isalpha()]:
+            sigs[getattr(signal, s)] = s
+        log.debug('El programa salió con la señal %s', sigs[sig])
+        prueba.pasada = False
+        #TODO otras señales conocidas
+        if sig == signal.SIGXCPU:
+            prueba.observaciones = 'Excedió el límite de tiempo de CPU ' \
+                '(%d seg)' % tiempo_cpu
+        else:
+            prueba.observaciones = 'Salió con la señal %s' % sigs[sig]
+        mail.agregarResultado(prueba)
+        return prueba
+    # Si tenemos que verificar el código de retorno
+    if caso_de_prueba.codigoRetorno is not None:
+        #FIXME trucho lo de 256
+        # Si el código de error esperado es 256 => el código de error debe ser != 0
+        # Si no el código de error esperado debe ser igual al obtenido
+        if caso_de_prueba.codigoRetorno != proc.returncode \
+                or caso_de_prueba.codigoRetorno == 256 \
+                    and proc.returncode != 0:
+            log.debug('Código de retorno incorrecto (debía ser %d y se obtuvo %d)',
+                caso_de_prueba.codigoRetorno, proc.returncode)
+            prueba.pasada = False
+            prueba.observaciones = 'Código de retorno incorrecto (debía ' \
+                'ser %d y se obtuvo %d)' % (caso_de_prueba.codigoRetorno,
+                proc.returncode)
+            mail.agregarResultado(prueba)
+            return prueba
+    # Verifico salidas estándar/de error
+    prueba.pasada = True # Asumo que está bien, ya habrá tiempo para cambiarlo
+    obs = ''
+    if usa_stdout():
+        obs += diff(prueba, mail, file(stdout_fn).readlines(),
+            file(os.path.join(caso_de_prueba.path, 'stdout')).readlines(),
+            caso_de_prueba.nombre + '.stdout', longname='La salida estándar')
+    if usa_stderr():
+        obs += diff(prueba, mail, file(stderr_fn).readlines(),
+            file(os.path.join(caso_de_prueba.path, 'stderr')).readlines(),
+        caso_de_prueba.nombre + '.stderr', longname='La salida de error')
+    for f in caso_de_prueba.archivosSalida:
+        if f not in prueba.archivosSalida:
+            log.debug('El programa no generó el archivo de salida "%s"', f)
+            prueba.pasada = False
+            obs += "* El programa no generó el archivo de salida '%s'.\n" % f
+            continue
+        obs += diff(prueba, mail,
+            file(os.path.join(intento.chrootPath(caso_de_prueba), f)).readlines(),
+            file(os.path.join(caso_de_prueba.pathSalidas, f)).readlines(),
+            caso_de_prueba.nombre + '.' + f)
+    for f in prueba.archivosSalida - (caso_de_prueba.archivosSalida | caso_de_prueba.archivosEntrada):
+        log.debug('Se generó un archivo "%s" que no se esperaba', f)
+        prueba.pasada = False
+        obs += "* Se generó un archivo ('%s') que no se esperaba.\n" % f
+    log.debug('Fin de ejecución de caso de prueba (hijo: %d, ret: %d)',
+        proc.pid, proc.returncode)
+    if obs:
+        prueba.observaciones = obs.replace('\0','\\0')
+    mail.agregarResultado(prueba)
+    log.debug('Resultado de la prueba: %s', prueba)
+    return prueba

 
 

-    // comienza la ejecución de casos de prueba
-    while ($prueba = $intento->pedir_caso_de_prueba()) {
-        logs('Prueba: ' . $prueba->to_line(), DEBUG);
-        // ejecuta con fork
-        $pid = pcntl_fork();
-        if ($pid == -1) {
-            // Error al forkear
-            $intento->resultado_de_prueba($prueba, false, 'Error al forkear proceso');
-            enviar_respuesta_error_log($mail, 'Error al forkear proceso', $intento);
-            continue 2;
-        } elseif ($pid) {
-            // Estamos en el padre, controlamos el tiempo.
-            logs("En el padre (hijo: $pid)", DEBUG);
-            // TODO controlar tiempo.
-            pcntl_waitpid($pid, $exitcode);
-            logs("Fin de ejecución de caso de prueba (hijo: $pid, ret: $exitcode)", DEBUG);
-            $stderr = false; // FIXME ver si salida de error es vacia.
-            if ($exitcode or $stderr) {
-                logs('Comando ejecutado ERROR', DEBUG);
-                if ($ret) {
-                    $msg = "El programa salió con código de error $ret";
-                    $msgs[] = $msg;
-                    logs($msg, DEBUG);
-                }
-                if ($stderr) {
-                    $msg = "El programa imprimió mensajes en la salida de error: '$stderr'";
-                    $msgs[] = $msg;
-                    logs($msg, DEBUG);
-                }
-                $msg = join("\n", $msgs);
-                $intento->resultado_de_prueba($prueba, false, $msg);
-                enviar_respuesta(R_ERR, $mail, $msg, $intento);
-                exit(0); // salgo ok del hijo
-            } else { // Sin errores en la ejecución.
-                logs('Comando ejecutado OK', DEBUG);
-                // TODO
-                $salidas = array();
-                foreach ($prueba->salidas as $salida) {
-                    // TODO hacer diffs.
-                    if ($salida == 'stdout') {
-                        $salidas[$salida] = $stdout;
-                    } else {
-                        $salidas[$salida] = file_get_contents($salida);
-                    }
-                }
-                logs('Salidas: ' . var_export($salidas, true), DEBUG);
-                $intento->resultado_de_prueba($prueba, true, 'Salidas: ' . var_export($salidas, true));
-                enviar_respuesta(R_ERR, $mail, 'Salidas: ' . var_export($salidas, true), $intento);
-                exit(0); // salgo ok del hijo
-            }
-        } else {
-            // Estamos en el hijo, corremos en chroot.
-            logs('En el hijo', DEBUG);
-            // Hago chroot.
-            if (!@chroot('chroot')) {
-                $intento->resultado_de_prueba($prueba, false, 'Error al hacer chroot');
-                enviar_respuesta_error_log($mail, 'Error al hacer chroot', $intento);
-                exit(1); // salgo del hijo
-            }
-            logs('Chrooteado, cwd = '.getcwd(), DEBUG);
-            if (!@posix_setuid($usrinfo['uid'])) {
-                $intento->resultado_de_prueba($prueba, false, "Error al cambiar al uid '{$usrinfo['uid']}'");
-                enviar_respuesta_error_log($mail, "Error al cambiar al uid '{$usrinfo['uid']}'", $intento);
-                exit(2); // salgo del hijo
-            }
-            // TODO poner stdout y stderr en archivos para hacer diff o hacer diffs en memoria
-            $stdout = null;
-            if (in_array('stdout', $prueba->salidas)) $stdout = '';
-            logs('Se ejecutará: chroot chroot /tp ' . escapeshellarg($prueba->params), DEBUG);
-            if (!pcntl_exec('/redir', array('tp', 'stdin', 'stdout', 'stderr', '/tp', '0', $prueba->params))) {
-                logserr('No se pudo ejecutar el comando');
-                $intento->resultado_de_prueba($prueba, false, 'No se pudo ejecutar el tp');
-                enviar_respuesta_error_log($mail, 'No se pudo ejecutar el tp', $intento);
-                exit(4); // salgo del hijo
-            }
-            exit(100); // salgo del hijo (no deberia llegar nunca aca).
-        }
-        break; // FIXME
-    }
+def params2seq(params):
+    r"""Parsea un string de forma similar al bash, separando por espacios y
+    teniendo en cuenta comillas simples y dobles para agrupar. Para poner
+    comillas se puede usar el \ como caracter de escape (\' y \") y también
+    interpreta \n y \t. Devuelve una lista con los parámetros encontrados."""
+    # Constantes
+    SEP, TOKEN, DQUOTE, SQUOTE = ' ', None, '"', "'"
+    seq = []
+    buff = ''
+    escape = False
+    state = SEP
+    for c in params:
+        # Es un caracter escapado
+        if escape:
+            if c == 'n':
+                buff += '\n'
+            elif c == 't':
+                buff += '\t'
+            else:
+                buff += c
+            escape = False
+            continue
+        # Es una secuencia de escape
+        if c == '\\':
+            escape = True
+            continue
+        # Si está buscando espacios
+        if state == SEP:
+            if c == SEP:
+                continue
+            else:
+                state = TOKEN # Encontró
+        if state == TOKEN:
+            if c == DQUOTE:
+                state = DQUOTE
+                continue
+            if c == SQUOTE:
+                state = SQUOTE
+                continue
+            if c == SEP:
+                state = SEP
+                seq.append(buff)
+                buff = ''
+                continue
+            buff += c
+            continue
+        if state == DQUOTE:
+            if c == DQUOTE:
+                state = TOKEN
+                continue
+            buff += c
+            continue
+        if state == SQUOTE:
+            if c == SQUOTE:
+                state = TOKEN
+                continue
+            buff += c
+            continue
+        raise Exception, 'No tiene sentido'
+    if state == DQUOTE or state == SQUOTE:
+        raise Exception, 'Parse error, falta cerrar comilla (%s)' % state
+    if buff:
+        seq.append(buff)
+    return seq

 
 

-    // TODO make clean
-}
+class MailIntento(email.MIMEMultipart.MIMEMultipart, object):
+    def __init__(self, intento):
+        global conf
+        from email.MIMEMultipart import MIMEMultipart
+        from email.MIMEMessage import MIMEMessage
+        from email.MIMEText import MIMEText
+        MIMEMultipart.__init__(self)
+        self.subject = '[%s] Resultado del intento %d (ejercicio %d.%d)' % \
+            (conf.get('mail', 'prefijo'), intento.numero,
+                intento.entrega.nroEjercicio, intento.entrega.entrega)
+        self['From'] = conf.get('mail', 'from')
+        self['To'] = intento.mailRespuesta
+        self['Reply-To'] = conf.get('mail', 'admin')
+        self['Return-Path'] = conf.get('mail', 'admin')
+        self['X-Mailer'] = 'sercom ' + sercom.VERSION
+        self['X-Priority'] = '5'
+        self.epilogue = 'Para ver correctamente este e-mail su cliente debe ' \
+            'soportar MIME.\n\n'
+        self.prologue = '' # Garantiza que termine en \n el mensaje
+        self.attach(MIMEMessage(MIMEText('', 'plain', 'iso-8859-1')))
+        self.resultado = None
+    def __set_body(self, body):
+        self.get_payload(0).get_payload(0).set_payload(body)
+    def __get_body(self):
+        return self.get_payload(0).get_payload(0).get_payload()
+    body = property(__get_body, __set_body, doc='Cuerpo del mensaje.')
+    def attachText(self, text, nombre=None, subtype='plain'):
+        from email.MIMEText import MIMEText
+        attach = MIMEText(text, subtype, 'iso-8859-1')
+        if nombre:
+            attach.add_header('Content-Disposition', 'attachment', filename=nombre)
+        self.attach(attach)
+    def send(self, resultado=None):
+        import smtplib
+        global conf
+        smtp = smtplib.SMTP(conf.get('mail', 'smtp'))
+        if resultado:
+            self.subject += ': ' + resultado
+        self['Subject'] = self.subject
+        smtp.sendmail(self['From'], self['To'], self.as_string())
+        smtp.close()
+    def agregarResultado(self, prueba):
+        if not prueba.casoDePrueba.privado:
+            if prueba.pasada:
+                result = 'BIEN'
+            else:
+                result = 'ERROR'
+            self.body += '''
+Prueba '%s': %s
+%s
+''' % (prueba.casoDePrueba.nombre, result, prueba.observaciones or '')
+        pass

 
 
 
 

-/**
- * Ejecuta un comando devolviendo el código de error y las salidas.
- *
- * @param cmd Comando a ejecutar.
- * @param ret Código de retorno.
- * @param err Salida de error (se obtiene sólo si es != null).
- * @param out Salida estándar (se obtiene sólo si es != null).
- * @param in  Entrada a enviarle al comando (null si no se le envía nada).
- * @return true si se ejecutó el comando, false si no.
- */
-function exec_fds($cmd, &$ret, $in = false, $out = false, $in = false) {
-    $descriptors = array();
-    if ($in)  $descriptors[0] = get_fd_type($in,  'r');
-    if ($out) $descriptors[1] = get_fd_type($out, 'w');
-    if ($err) $descriptors[2] = get_fd_type($err, 'w');
-var_dump($descriptors);
-    $proc = proc_open($cmd, $descriptors, $pipes);
-    if (is_resource($proc)) {
-        $ret = proc_close($proc);
-        return true;
-    }
-    return false;
-}
+# Manejadores de señales
+signal.signal(signal.SIGTERM, sigterm)
+signal.signal(signal.SIGINT, sigterm)
+signal.signal(signal.SIGCHLD, sigchld)
+hijo_muerto = False # Cambia con SIGCHLD
+continuar = True # Cambia con SIGTERM o SIGINT

 
 

-/**
- * Obtiene un tipo de descriptor para proc_open().
- *
- * @param fd Tipo de descriptor: si es true, se usa un pipe, si es un array
- *           se usa el array y si es otra cosa se usa un archivo con este
- *           nombre.
- * @param mode Modo de apertura, sólo se usa si es true o string.
- * @return Array a pasar a proc_open().
- */
-function get_fd_type($fd, $mode = false) {
-    if ($fd === true) {
-        return array('pipe', $mode);
-    } elseif (is_array($fd)) {
-        return $fd;
-    }
-    return array('file', $fd, $mode);
-}
+# Inicializo
+conf, conn, log = sercom.init('test')
+log.info('Iniciado')

 
 

-/**
- * Ejecuta un comando devolviendo el código de error y las salidas.
- *
- * @param cmd Comando a ejecutar.
- * @param ret Código de retorno.
- * @param err Salida de error (se obtiene sólo si es != null).
- * @param out Salida estándar (se obtiene sólo si es != null).
- * @param in  Entrada a enviarle al comando (null si no se le envía nada).
- * @return true si se ejecutó el comando, false si no.
- */
-function exec_get_info($cmd, &$ret, &$err, &$out, $in = null) {
-    $descriptors = array(array('pipe', 'r'), array('pipe', 'w'), array('pipe', 'w'));
-    $proc = proc_open($cmd, $descriptors, $pipes);
-    if (is_resource($proc)) {
-        if (is_null($in)) {
-            fputs($pipes[0], $in);
-        }
-        if (is_null($out)) {
-            $out = stream_get_contents($pipes[1]);
-        }
-        if (is_null($err)) {
-            $err = stream_get_contents($pipes[2]);
-        }
-        foreach (array(0,1,2) as $i) {
-            fclose($pipes[$i]);
-        }
-        $ret = proc_close($proc);
-        return true;
-    }
-    return false;
-}
+# Obtengo id de usuario con el cual ejecutar las pruebas
+uid, gid = pwd.getpwnam(conf.get('general', 'user'))[2:4]
+
+# Cambio UID efectivo
+os.seteuid(uid)

 
 

-function enviar_respuesta($tipo, $to, $mensaje = '', $intento = null) {
-    $mconf = $GLOBALS['CONF']['mail'];
-    $subject = '[' . $NAME . '] Prueba ';
-    if ($tipo == R_OK) $estado = 'OK';
-    else               $estado = 'ERROR';
-    $subject .= $estado;
-    $body .= "Estado: $estado\n";
-    if ($mensaje) $body .= "\n$mensaje\n";
-    if ($intento) $body .= "\n" . $intento->__toString() . "\n";
-    logs("Envío de mail '$subject' a '$to'\n$body\n", DEBUG);
-    $headers = <<<EOT
-From: {$mconf['from']}
-Reply-To: {$mconf['admin']}
-X-Mailer: $NAME $VERSION
-X-Priority: 5
-EOT;
-    mail($to, $subject, $body, $headers);
-    //mail($mconf['admin'], $subject, $body, $headers);
-    return true;
-}
+# Atajo
+intervalo = float(conf.get('general', 'intervalo'))

 
 

-/**
- * Cambia permisos, dueño y grupo a un archivo, devuelve string con error o
- * false si no hay error.
- */
-function chmod_own_grp($file, $mod, $own, $grp) {
-    if (!@chmod($file, $mod)) return "Error al cambiar permisos [$mod] a '$file'";
-    if (!@chown($file, $own)) return "Error al cambiar dueño [$own] a '$file'";
-    if (!@chgrp($file, $grp)) return "Error al cambiar grupo [$grp] a '$file'";
-    return false;
-}
+# Utilizo el directorio de datos como base para todos los SQLObjects
+sercom.sqlo.dir_base = conf.get('general', 'data_dir')

 
 

-function enviar_respuesta_error_log($to, $msg = '', $intento = null) {
-    logserr($msg);
-    enviar_respuesta(R_ERR, $to, "ERROR: $msg\n\nSe envió un mail al administrador para revisar el problema.\n", $intento);
-    enviar_respuesta(R_ERR, $GLOBALS['CONF']['mail']['admin'], $msg, $intento);
-}
+# Hasta que nos maten
+while continuar:
+    # Busco intento a probar
+    intento = Intento.getProximoAProbar(conn)
+    if not intento:
+        log.debug('No hay intento para probar')
+        time.sleep(intervalo)
+        continue
+    log.info('Nuevo intento a probar (%d %d.%d.%d %d%d%d [%d %d %d])',
+        intento.inscripto.padron, intento.entrega.nroEjercicio,
+        intento.entrega.entrega, intento.numero, intento.entrega.curso.anio,
+        intento.entrega.curso.cuatrimestre, intento.entrega.curso.curso,
+        intento.inscripto.id, intento.entrega.id, intento.id)
+    mail = MailIntento(intento)
+    # Compila
+    compilar(intento, mail)
+    if not intento.compila:
+        log.info('Intento[%d] no compila', intento.id)
+        mail.send('NO COMPILA')
+        continue
+    # Ejecución de casos de prueba
+    intento.inicioPruebas = datetime.datetime.now()
+    resultado = True
+    for caso_de_prueba in intento.entrega.ejercicio.casosDePrueba:
+        # Preparo chroot
+        preparar(intento, caso_de_prueba)
+        # Pruebo y agrego prueba a la lista
+        prueba = probar(intento, caso_de_prueba, mail)
+        if not prueba.casoDePrueba.privado and not prueba.pasada:
+            log.info('Prueba %s[%d %d] no pasada', prueba.casoDePrueba.nombre,
+                prueba.casoDePrueba.id, prueba.id)
+            resultado = False
+        # Limpio chroot
+        limpiar(intento, caso_de_prueba)
+    intento.finPruebas = datetime.datetime.now()
+    # Envío mail con resultado al alumno
+    if resultado:
+        mail.send('ACEPTADO')
+        log.info('Pruebas finalizadas (intento[%d] aceptado)', intento.id)
+    else:
+        mail.send('FALLA PRUEBAS')
+        log.info('Pruebas finalizadas (intento[%d] no aceptado)', intento.id)
+    # Limpio directorio
+    log.debug('Borrando ejecutable y código objeto (*.o)')
+    os.remove(os.path.join(intento.path, 'tp'))
+    for obj in glob.glob(os.path.join(intento.path, '*.o')):
+        os.remove(obj)
+    # time.sleep(intervalo) #XXX Puede servir para enlentecer el server

 
 

-?>
\ No newline at end of file