由于近来很忙,加之此类文章少有人问津.所以少有时间写这一系列的文章的续篇
所以先把核心在线编译的代码发出来,细的部分大家可以查资料,或者留言问我^^
部分类库可以查我以前的文章
using System;
using System.Text;
using System.Data;
using System.Diagnostics;
using System.Threading;
using System.IO;
namespace ChswordOJ {
/// <summary>
/// DoFile 进行在线编译与测试
/// Builder:邹健 2007 4 28
/// V1.0 2007 6 12
/// </summary>
public class DoFile {
StringBuilder _output = new StringBuilder();
StringBuilder _TestOutput = new StringBuilder();
StringBuilder _TestFile_Output = new StringBuilder();
String _Test;
String _CompilerName;
StringBuilder _CompilerText = new StringBuilder();
private Int64 _Memory, _myMemory;
private Int32 _Time;//, _myTime;
private Option.AnswerStatus _TestResult;
private String _ExePath;
private String _CodePath;
private String _TextPath;
private String _TestPath;
private Int64 _AnswerId;
private String _Sign;
private Boolean _TestEnd;
Process _process;
/// <summary>
/// 构造函数，对DoFile类进行初始化
/// </summary>
public DoFile() {
_Memory = 23768;
_Time = 3;
}
/// <summary>
/// 进行编译及测试代码。
/// </summary>
/// <param name="code">要编译的代码。</param>
/// <param name="UserName">编译的用户名。</param>
/// <param name="PassWord">用户密码。</param>
/// <param name="QNum">代码的问题ID。</param>
/// <param name="CompilerName">编译器。</param>
public void GetResult(String code, String UserName, String PassWord, String QNum, String CompilerName) {
Option cp = new Option();
DataSet ds = cp.GetLimit(QNum);
User u = new User(UserName, CompilerName.Split(',')[0], CompilerName.Split(',')[1]);
_Memory = int.Parse(ds.Tables[0].Rows[0][0].ToString());//从数据库读取当前的程序 要求的Memory
_Time = int.Parse(ds.Tables[0].Rows[0][1].ToString());//从数据库读取当前的程序 要求的Time limit
_Test = ds.Tables[0].Rows[0]["test"].ToString();//读取测试文件
_CompilerName = CompilerName;//编译器名称
_Sign = u.Sign;
_CodePath = u.CodePath;//源代码文件保存的路径
_TextPath = u.TextPath;//生成文本文件保存的路径
_ExePath = u.ExePath;//EXE文件保存的路径
_myMemory = 0;//对我当前耗费的内存进行初始化
_TestPath = u.GetTestPath(QNum);//得到当前测试路径
_AnswerId = cp.AddAnswer(UserName, Int64.Parse(QNum), CompilerName);//将当前数据作为一条回答存入数据库(创建一条数据库记录)
Code cc = new Code(CompilerName, code);//创建Code实例
if (cc.Check()) {//调用配置文件中的正则表达式,测试代码中有无危险代码,有,则此回答为危险代码
SaveTextFile(u.CodePath, code);//将源代码保存为相应格式文件
_CompilerText.AppendLine(u.CompilerString);//
DoCmd(u.CompilerString);//用u生成编译字符串,执行编译操作
IsTempExists(u.TextPath, u.CompilerString);//是否编译完成
if (0 == _AnswerId)
return;
cp.SetAnswerStatus(_AnswerId, Option.AnswerStatus.编译中);
//编译完成
if (IsCompilerSucess(u.TextPath, u.CodePath, u.ExePath, 0)) {//编译产生EXE则编译成功
//编译成功
cp.SetAnswerStatus(_AnswerId, Option.AnswerStatus.测试中);
ThreadStart thr_start_func = new ThreadStart(Test);//异步调用Test进行测试
Thread fThread = new Thread(thr_start_func);
fThread.Name = "Test";
fThread.Start();
Thread.Sleep(_Time * 2000);//初始化最大超时时间(规定时间的2倍)
if (Option.AnswerStatus.测试通过 == _TestResult) {
//cp.SetAnswerStatus(_AnswerId, Compiler.AnswerStatus.测试通过);
}
else {
if (_TestResult == 0) {
try {
//对进程查看是否结束,未结束则进行操作
if (!_process.HasExited) {
if (!_process.CloseMainWindow())
_process.Kill();
if (_TestFile_Output.ToString().StartsWith(_TestOutput.ToString()))
_TestResult = Option.AnswerStatus.超时;
else
_TestResult = Option.AnswerStatus.测试失败;
}
}
catch {
_TestResult = Option.AnswerStatus.测试失败;
}
}
if (fThread.IsAlive) fThread.Abort();//最终的强制结束进程
}
}
else {
_TestResult = Option.AnswerStatus.编译失败;
cp.SetAnswerText(_AnswerId, ReplaceFileName(_CompilerText.ToString()));
}
try {
if (_TestResult == Option.AnswerStatus.测试中 || _TestResult == Option.AnswerStatus.测试失败) {
Process[] p = Process.GetProcessesByName(_Sign);
if (p.Length > 0) {
if (!p[0].HasExited) {
if (p[0].Responding) {
p[0].CloseMainWindow();
p[0].Kill();
}
else {
p[0].Kill();
}
}
if (_TestFile_Output.ToString().StartsWith(_TestOutput.ToString()))
_TestResult = Option.AnswerStatus.超时;
else
_TestResult = Option.AnswerStatus.测试失败;
}
}
}
catch { }
}
else {
_TestResult = Option.AnswerStatus.危险代码;
}
cp.SetAnswerStatus(_AnswerId, _TestResult);//写入测试结果
cp.SaveAnswerCode(_AnswerId, code);//写入提交的源代码
/*try {
if (_TestResult == Option.AnswerStatus.测试通过) {
using (StreamReader sr = File.OpenText(_CodePath)) {
cp.SaveAnswerCode(_AnswerId, sr.ReadToEnd());
}
}
}
catch { }*/
DeleteTempFile();//删除临时文件
}
private void DeleteTempFile() {//删除临时文件
Byte b = 0;
try {
Thread.Sleep(1000);//每一秒试删除一次
if (File.Exists(_ExePath)) File.Delete(_ExePath); else b++;
if (File.Exists(_CodePath)) File.Delete(_CodePath); else b++;
if (File.Exists(_TextPath)) File.Delete(_TextPath); else b++;
}
catch { }
if (b != 3) DeleteTempFile();//递归调用
return;
}
private bool IsCompilerSucess(string TextPath, string CppPath, String ExePath, int count) {
try {//看是否编译成功的函数
using (StreamReader sr = new StreamReader(TextPath, Encoding.Default)) {
String input;
bool f = true;
while ((input = sr.ReadLine()) != null) {
_CompilerText.AppendLine(input);
if (input.Contains(": error:")) {
_output.AppendLine(input);
}
else {
if (f && !input.StartsWith(CppPath))
_output.AppendLine(input);
f = false;
}
}
sr.Close();
if (File.Exists(ExePath)) {
return true;
}
else {
_output.Insert(0, "编译出错:\r\n");
return false;
}
}
}
catch {
}
Thread.Sleep(1000);
if (count >= 300) {
_CompilerText.Insert(0, "服务器响应超时或您选择的编译器不正确");
return false;
}
return IsCompilerSucess(TextPath, CppPath, ExePath, count + 1);
}
private string ReplaceFileName(string input) {
input = System.Text.RegularExpressions.Regex.Replace(input, @"([A-Z]:[\\|\/][^:\*\?<>\|]+\.(txt|cs|cpp|c|vb|jsl|js|exe))|(\\{2}[^/:\*\?<>\|]+\.(txt|cs|cpp|c|vb|jsl|js|exe))", ">", System.Text.RegularExpressions.RegexOptions.IgnoreCase);
return input;
}
private void Process_Exited(object sender, EventArgs e) {//测试程序是否超时的判断
if (_process.TotalProcessorTime.Milliseconds > _Time*1000) {
if (_TestFile_Output.ToString().StartsWith(_TestOutput.ToString()))
_TestResult = Option.AnswerStatus.超时;
else
_TestResult = Option.AnswerStatus.测试失败;
}
}
private void Test() {//进行测试
string ExePath = _ExePath;
string TestPath = _TestPath;
String input;
bool f = true;
_TestEnd = false;
_process = new Process();
_process.StartInfo.WindowStyle = ProcessWindowStyle.Hidden;
_process.StartInfo.UseShellExecute = false;
_process.StartInfo.CreateNoWindow = true;
_process.StartInfo.RedirectStandardInput = true;
_process.StartInfo.RedirectStandardOutput = true;
_process.StartInfo.RedirectStandardError = true;
_process.StartInfo.FileName = ExePath;
_process.Exited += new EventHandler(Process_Exited);
_process.OutputDataReceived += new DataReceivedEventHandler(TestOutputHandler);
_process.Start();
_process.BeginOutputReadLine();
_myMemory = (_process.PeakWorkingSet64 >> 10);//获取内存大小
StreamWriter sortStreamWriter = _process.StandardInput;//设置异步读取流
try {
using (StringReader sr = new StringReader(_Test)) {
while (((input = sr.ReadLine()) != null)) {
if (input.StartsWith("EOF"))//C/C++程序的结束(CTRL+Z)方式
if (_CompilerName.ToLower() == "c" || _CompilerName.ToLower() == "cpp")
sortStreamWriter.WriteLine(((char)26).ToString());
else//.net程序的结束(CTRL+Z)方式
_process.CloseMainWindow();
if (input.StartsWith(">"))//测试的写操作
sortStreamWriter.WriteLine(input.Substring(1));
if (input.StartsWith("<"))//测试的读操作
_TestFile_Output.AppendLine(input.Substring(1));
}
}
sortStreamWriter.Close();
_process.WaitForExit();
_TestEnd = true;
}
catch {
_TestEnd = true;
}
if (_TestOutput.ToString() == _TestFile_Output.ToString())
f = true;
else
f = false;
if (_TestEnd && f)
_TestResult = Option.AnswerStatus.测试通过;
else {
}
if (_TestEnd && (_myMemory > _Memory))//内存消耗太大
{
_TestResult = Option.AnswerStatus.内存超量;
}
_process.Close();
_process.Dispose();
if (f && _TestEnd) {
_TestResult = Option.AnswerStatus.测试通过;
}
return;
}
void SaveTextFile(string Path, string text) {
using (StreamWriter sw = new StreamWriter(Path)) { sw.Write(text); }
}
void DoCmd(string cmd) {//执行一条CMD命令的函数
Process process = new Process();
process.StartInfo.WindowStyle = ProcessWindowStyle.Hidden;
process.StartInfo.UseShellExecute = false;
process.StartInfo.CreateNoWindow = true;
process.StartInfo.RedirectStandardInput = true;
process.StartInfo.RedirectStandardError = true;
process.StartInfo.FileName = "cmd.exe";
process.Start();
process.StandardInput.WriteLine(cmd);
process.StandardInput.WriteLine("\r\nexit");
process.Close();
process.Dispose();
}
bool IsTempExists(String TextPath, String CompilerString) {
int i = 0;
while (!File.Exists(TextPath)) {
Thread.Sleep(1000);
i++;
if (i > 178) return false;
if (i % 30 == 0 && i > 10) {
DoCmd(CompilerString);
}
}
return true;
}
private void TestOutputHandler(object sendingProcess,
DataReceivedEventArgs outLine) {
int numOutputLines = 0;
if (!String.IsNullOrEmpty(outLine.Data)) {
numOutputLines++;
//Environment.NewLine +"[" + numOutputLines.ToString() + "] - " +
_TestOutput.AppendLine(outLine.Data);
}
}
}
}
上篇:由成幻OnlineJudge学习如何做自己的Acm-Icpc在线评判系统－4.建立基本的网站类库