shiftos-archive/ShiftOS_TheReturn
1
0
Fork 0
mirror of https://git.alee14.me/shiftos-archive/ShiftOS_TheReturn.git synced 2025-01-22 18:02:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Modularize the Lua engine

Browse source 
C# code can be exposed to Lua from anywhere in the entire source code.
This commit is contained in:
Michael 2017-01-27 14:18:46 -05:00
parent c62cb58ba0
commit b5d5178411
9 changed files with 73 additions and 862 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
ShiftOS.WinForms/Program.cs
View file

@ -188,17 +188,6 @@ namespace ShiftOS.WinForms
case FileType.Executable:
//NYI
throw new Exception();
case FileType.Python:
var p = new Engine.Scripting.PythonInterpreter();
try
{
p.ExecuteFile(path);
}
catch (Exception ex)
{
Infobox.Show("{PY_EXCEPTION}", ex.Message);
}
break;
case FileType.Lua:
try
{

66
ShiftOS_TheReturn/AltTabWindow.Designer.cs generated
View file

@ -1,66 +0,0 @@
/*
* MIT License
*
* Copyright (c) 2017 Michael VanOverbeek and ShiftOS devs
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
namespace ShiftOS.Engine {
partial class AltTabWindow {
/// <summary>
/// Required designer variable.
/// </summary>
private System.ComponentModel.IContainer components = null;
/// <summary>
/// Clean up any resources being used.
/// </summary>
/// <param name="disposing">true if managed resources should be disposed; otherwise, false.</param>
protected override void Dispose(bool disposing) {
if (disposing && (components != null)) {
components.Dispose();
}
base.Dispose(disposing);
}
#region Windows Form Designer generated code
/// <summary>
/// Required method for Designer support - do not modify
/// the contents of this method with the code editor.
/// </summary>
private void InitializeComponent() {
this.SuspendLayout();
//
// AltTabWindow
//
this.AutoScaleDimensions = new System.Drawing.SizeF(6F, 13F);
this.AutoScaleMode = System.Windows.Forms.AutoScaleMode.Font;
this.ClientSize = new System.Drawing.Size(284, 261);
this.Name = "AltTabWindow";
this.Text = "AltTabWindow";
this.Load += new System.EventHandler(this.AltTabWindow_Load);
this.ResumeLayout(false);
}
#endregion
}
}

53
ShiftOS_TheReturn/AltTabWindow.cs
View file

@ -1,53 +0,0 @@
/*
* MIT License
*
* Copyright (c) 2017 Michael VanOverbeek and ShiftOS devs
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
namespace ShiftOS.Engine {
public partial class AltTabWindow : Form {
public AltTabWindow() {
InitializeComponent();
}
internal void CycleBack() {
Console.WriteLine("Cycle Backwards");
}
internal void CycleForwards() {
Console.WriteLine("Cycle Forwards");
}
private void AltTabWindow_Load(object sender, EventArgs e) {
}
}
}

120
ShiftOS_TheReturn/AltTabWindow.resx
View file

@ -1,120 +0,0 @@
<?xml version="1.0" encoding="utf-8"?>
<root>
<!--
Microsoft ResX Schema
Version 2.0
The primary goals of this format is to allow a simple XML format
that is mostly human readable. The generation and parsing of the
various data types are done through the TypeConverter classes
associated with the data types.
Example:
... ado.net/XML headers & schema ...
<resheader name="resmimetype">text/microsoft-resx</resheader>
<resheader name="version">2.0</resheader>
<resheader name="reader">System.Resources.ResXResourceReader, System.Windows.Forms, ...</resheader>
<resheader name="writer">System.Resources.ResXResourceWriter, System.Windows.Forms, ...</resheader>
<data name="Name1"><value>this is my long string</value><comment>this is a comment</comment></data>
<data name="Color1" type="System.Drawing.Color, System.Drawing">Blue</data>
<data name="Bitmap1" mimetype="application/x-microsoft.net.object.binary.base64">
<value>[base64 mime encoded serialized .NET Framework object]</value>
</data>
<data name="Icon1" type="System.Drawing.Icon, System.Drawing" mimetype="application/x-microsoft.net.object.bytearray.base64">
<value>[base64 mime encoded string representing a byte array form of the .NET Framework object]</value>
<comment>This is a comment</comment>
</data>
There are any number of "resheader" rows that contain simple
name/value pairs.
Each data row contains a name, and value. The row also contains a
type or mimetype. Type corresponds to a .NET class that support
text/value conversion through the TypeConverter architecture.
Classes that don't support this are serialized and stored with the
mimetype set.
The mimetype is used for serialized objects, and tells the
ResXResourceReader how to depersist the object. This is currently not
extensible. For a given mimetype the value must be set accordingly:
Note - application/x-microsoft.net.object.binary.base64 is the format
that the ResXResourceWriter will generate, however the reader can
read any of the formats listed below.
mimetype: application/x-microsoft.net.object.binary.base64
value : The object must be serialized with
: System.Runtime.Serialization.Formatters.Binary.BinaryFormatter
: and then encoded with base64 encoding.
mimetype: application/x-microsoft.net.object.soap.base64
value : The object must be serialized with
: System.Runtime.Serialization.Formatters.Soap.SoapFormatter
: and then encoded with base64 encoding.
mimetype: application/x-microsoft.net.object.bytearray.base64
value : The object must be serialized into a byte array
: using a System.ComponentModel.TypeConverter
: and then encoded with base64 encoding.
-->
<xsd:schema id="root" xmlns="" xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:msdata="urn:schemas-microsoft-com:xml-msdata">
<xsd:import namespace="http://www.w3.org/XML/1998/namespace" />
<xsd:element name="root" msdata:IsDataSet="true">
<xsd:complexType>
<xsd:choice maxOccurs="unbounded">
<xsd:element name="metadata">
<xsd:complexType>
<xsd:sequence>
<xsd:element name="value" type="xsd:string" minOccurs="0" />
</xsd:sequence>
<xsd:attribute name="name" use="required" type="xsd:string" />
<xsd:attribute name="type" type="xsd:string" />
<xsd:attribute name="mimetype" type="xsd:string" />
<xsd:attribute ref="xml:space" />
</xsd:complexType>
</xsd:element>
<xsd:element name="assembly">
<xsd:complexType>
<xsd:attribute name="alias" type="xsd:string" />
<xsd:attribute name="name" type="xsd:string" />
</xsd:complexType>
</xsd:element>
<xsd:element name="data">
<xsd:complexType>
<xsd:sequence>
<xsd:element name="value" type="xsd:string" minOccurs="0" msdata:Ordinal="1" />
<xsd:element name="comment" type="xsd:string" minOccurs="0" msdata:Ordinal="2" />
</xsd:sequence>
<xsd:attribute name="name" type="xsd:string" use="required" msdata:Ordinal="1" />
<xsd:attribute name="type" type="xsd:string" msdata:Ordinal="3" />
<xsd:attribute name="mimetype" type="xsd:string" msdata:Ordinal="4" />
<xsd:attribute ref="xml:space" />
</xsd:complexType>
</xsd:element>
<xsd:element name="resheader">
<xsd:complexType>
<xsd:sequence>
<xsd:element name="value" type="xsd:string" minOccurs="0" msdata:Ordinal="1" />
</xsd:sequence>
<xsd:attribute name="name" type="xsd:string" use="required" />
</xsd:complexType>
</xsd:element>
</xsd:choice>
</xsd:complexType>
</xsd:element>
</xsd:schema>
<resheader name="resmimetype">
<value>text/microsoft-resx</value>
</resheader>
<resheader name="version">
<value>2.0</value>
</resheader>
<resheader name="reader">
<value>System.Resources.ResXResourceReader, System.Windows.Forms, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089</value>
</resheader>
<resheader name="writer">
<value>System.Resources.ResXResourceWriter, System.Windows.Forms, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089</value>
</resheader>
</root>

59
ShiftOS_TheReturn/Commands.cs
View file

@ -44,69 +44,10 @@ using ShiftOS.Objects.ShiftFS;
namespace ShiftOS.Engine
{
[Namespace("virus")]
public class VirusTestCommands
{
[Command("infect")]
public static void InfectFileWithMUDVirus(Dictionary<string, object> args)
{
string signature = "";
if (args.ContainsKey("signature"))
signature = args["signature"] as string;
else
throw new Exception("Virus signature not provided.");
string script = "";
Action<string> scriptFound = new Action<string>((s) =>
{
script = s;
Virus v = new LuaVirus(s, 1);
VirusEngine.Add(v);
VirusEngine.Infect($"lua.{v.Signature}.1");
});
ServerManager.MessageReceived += (srv) =>
{
if (srv.Name == "mud_virus")
{
scriptFound?.Invoke(srv.Contents);
scriptFound = null;
}
};
ServerManager.SendMessage("getvirus", signature);
}
}
[RequiresUpgrade("mud_fundamentals")]
[Namespace("mud")]
public static class MUDCommands
{
[Command("addvirus")]
public static void Virus_AddToDatabase(Dictionary<string, object> args)
{
string file = "";
int threatLevel = 0;
if (args.ContainsKey("file"))
file = args["file"] as string;
else
throw new Exception("No 'file' argument provided.");
if (args.ContainsKey("threatlevel"))
threatLevel = Convert.ToInt32(args["threatlevel"].ToString());
Virus lua = new LuaVirus(Utils.ReadAllText(file), threatLevel);
Console.WriteLine("Virus uploaded to current multi-user domain successfully.");
}
[Command("status")]
public static bool Status()
{

50
ShiftOS_TheReturn/CtrlTabMenu.cs
View file

@ -1,50 +0,0 @@
/*
* MIT License
*
* Copyright (c) 2017 Michael VanOverbeek and ShiftOS devs
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace ShiftOS.Engine {
static class CtrlTabMenu {
public static AltTabWindow altTabWindow;
internal static void Show() {
if(altTabWindow != null) {
altTabWindow = new AltTabWindow();
altTabWindow.Show();
}
}
internal static void CycleBack() {
altTabWindow.CycleBack();
}
internal static void CycleForwards() {
altTabWindow.CycleForwards();
}
}
}

426
ShiftOS_TheReturn/Scripting.cs
View file

@ -27,9 +27,6 @@ using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using IronPython;
using IronPython.Hosting;
using Microsoft.Scripting.Hosting;
using System.Reflection;
using ShiftOS.Objects.ShiftFS;
using DynamicLua;
@ -60,28 +57,52 @@ namespace ShiftOS.Engine.Scripting
//This temporary proxy() method will be used by the API prober.
Lua.proxy = new Func<string, dynamic>((objName) =>
{
var asm = Assembly.GetExecutingAssembly();
foreach (var type in asm.GetTypes())
foreach (var f in System.IO.Directory.GetFiles(Environment.CurrentDirectory))
{
if (type.Name == objName)
if (f.EndsWith(".exe") || f.EndsWith(".dll"))
{
dynamic dynObj = Activator.CreateInstance(type);
return dynObj;
}
try
{
var asm = Assembly.LoadFile(f);
foreach (var type in asm.GetTypes())
{
if (type.Name == objName)
{
dynamic dynObj = Activator.CreateInstance(type);
return dynObj;
}
}
}
catch { }
}
}
throw new Exception("{CLASS_NOT_FOUND}");
});
var thisasm = Assembly.GetExecutingAssembly();
foreach (var type in thisasm.GetTypes())
foreach (var f in System.IO.Directory.GetFiles(Environment.CurrentDirectory))
{
foreach (var attr in type.GetCustomAttributes(false))
if (f.EndsWith(".exe") || f.EndsWith(".dll"))
{
if (attr is ExposedAttribute)
try
{
var eattr = attr as ExposedAttribute;
Lua($"{eattr.Name} = proxy(\"{type.Name}\")");
var thisasm = Assembly.LoadFile(f);
foreach (var type in thisasm.GetTypes())
{
foreach (var attr in type.GetCustomAttributes(false))
{
if (attr is ExposedAttribute)
{
var eattr = attr as ExposedAttribute;
Lua($"{eattr.Name} = proxy(\"{type.Name}\")");
}
}
}
}
catch
{
}
}
}
@ -127,382 +148,17 @@ namespace ShiftOS.Engine.Scripting
}
}
public class LuaVirus : Virus
{
public override string Signature
{
get
{
return signature;
}
}
public override int ThreatLevel
{
get
{
return _threatlevel;
}
}
public override string Type
{
get
{
return "lua";
}
}
private LuaInterpreter interpreter = null;
private string signature = "unknown";
private int _threatlevel = 0;
public LuaVirus(string script, int threatLevel) : base()
{
_threatlevel = threatLevel;
interpreter = new LuaInterpreter();
Action<string> mud_message = new Action<string>((sig) =>
{
signature = sig;
});
ServerManager.MessageReceived += (msg) =>
{
if(msg.Name == "mud_virus_signature")
{
mud_message?.Invoke(msg.Contents);
mud_message = null;
}
};
ServerManager.SendMessage("mud_scanvirus", script);
_script = script;
}
private string _script = "";
public override void Activate()
{
interpreter.Execute(_script);
}
public override void Deactivate()
{
interpreter.Running = false;
interpreter = null;
}
}
[Exposed("consts")]
public class Constants
{
public readonly DockStyle dock_none = DockStyle.None;
public readonly DockStyle dock_top = DockStyle.Top;
public readonly DockStyle dock_bottom = DockStyle.Bottom;
public readonly DockStyle dock_left = DockStyle.Left;
public readonly DockStyle dock_right = DockStyle.Right;
public readonly DockStyle dock_fill = DockStyle.Fill;
public readonly AnchorStyles anchor_none = 0x00;
public readonly AnchorStyles anchor_top = AnchorStyles.Top;
public readonly AnchorStyles anchor_bottom = AnchorStyles.Bottom;
public readonly AnchorStyles anchor_left = AnchorStyles.Left;
public readonly AnchorStyles anchor_right = AnchorStyles.Right;
}
[Exposed("shiftos")]
public class CoreAPI
{
public void sleep(int ms) { Thread.Sleep(ms); }
public bool isRunning() { return !SaveSystem.ShuttingDown; }
public int random(int min, int max)
{
return new Random().Next(min, max);
}
public void info(string title, string message)
{
Infobox.Show(title, message);
}
public void executeCommand(string cmd)
{
TerminalBackend.InvokeCommand(cmd);
}
public void shutdown()
{
TerminalBackend.InvokeCommand("sys.shutdown");
}
}
[Exposed("shiftorium")]
public class ShiftoriumAPI
{
public dynamic[] getAvailable()
{
return Shiftorium.GetAvailable();
}
public dynamic[] getAll()
{
return Shiftorium.GetDefaults().ToArray();
}
public bool upgradeInstalled(string id)
{
return Shiftorium.UpgradeInstalled(id);
}
public bool buy(ShiftoriumUpgrade upgrade)
{
foreach (var upg in getAvailable())
{
if (upg.ID == upgrade && SaveSystem.CurrentSave.Codepoints >= upg.Cost)
{
return true;
}
}
return false;
}
}
[Exposed("gui")]
public class GuiAPI
{
public void addControl(dynamic parent, dynamic child)
{
parent.Controls.Add(child);
}
public dynamic size(int width, int height)
{
return new System.Drawing.Size(width, height);
}
public dynamic point(int x, int y)
{
return new System.Drawing.Point(x, y);
}
public dynamic rect(dynamic p, dynamic s)
{
return new System.Drawing.Rectangle(p, s);
}
public dynamic rect(int x, int y, int w, int h)
{
return new System.Drawing.Rectangle(x, y, w, h);
}
public dynamic panel()
{
return new Panel();
}
public dynamic label()
{
return new Label();
}
public dynamic button()
{
return new Button();
}
public dynamic listbox()
{
return new ListBox();
}
public dynamic combobox()
{
return new ComboBox();
}
public dynamic textbox()
{
return new TextBox();
}
public dynamic window()
{
return new Form();
}
}
[Exposed("color")]
public class ColorAPI
{
public dynamic fromRgb(int r, int g, int b)
{
return System.Drawing.Color.FromArgb(r, g, b);
}
public dynamic fromArgb(int a, int r, int g, int b)
{
return System.Drawing.Color.FromArgb(a, r, g, b);
}
public dynamic fromName(string name)
{
return System.Drawing.Color.FromName(name);
}
}
[Exposed("fonts")]
public class FontsAPI
{
public readonly int style_regular = 0x00;
public readonly int style_bold = 0x01;
public readonly int style_italic = 0x02;
public readonly int style_underline = 0x04;
public readonly int style_strike = 0x08;
public dynamic create(string name, float size, int style)
{
return new System.Drawing.Font(name, size, (System.Drawing.FontStyle)style);
}
}
[Exposed("json")]
public class JsonAPI
{
public string serialize(dynamic dynObj)
{
return JsonConvert.SerializeObject(dynObj);
}
public dynamic deserialize(string json)
{
return JsonConvert.DeserializeObject<dynamic>(json);
}
}
[Exposed("fs")]
public class FilesystemAPI
{
public void writeAllText(string target, string contents)
{
Utils.WriteAllText(target, contents);
}
public string readAllText(string file)
{
return Utils.ReadAllText(file);
}
public bool fileExists(string file)
{
return Utils.FileExists(file);
}
public bool directoryExists(string dir)
{
return Utils.DirectoryExists(dir);
}
public void createDirectory(string dir)
{
Utils.CreateDirectory(dir);
}
//experimental - no idea how arrays will be handled in Lua...
public string[] getFiles(string dir)
{
return Utils.GetFiles(dir);
}
public string[] getDirectories(string dir)
{
return Utils.GetDirectories(dir);
}
public string mount(string mfsFile)
{
Utils.MountPersistent(mfsFile);
return $"{Utils.Mounts.Count - 1}:";
}
public void unmount(int driveNum)
{
Utils.Mounts.RemoveAt(driveNum);
}
}
public class PythonInterpreter
{
ScriptEngine python;
ScriptScope pyScope;
ScriptSource pySource;
public PythonInterpreter()
{
python = Python.CreateEngine();
pyScope = python.CreateScope();
AddExposedObjects();
}
private void AddExposedObjects()
{
var asm = Assembly.GetExecutingAssembly();
foreach(var type in asm.GetTypes())
{
foreach(var attr in type.GetCustomAttributes(false))
{
if(attr is ExposedAttribute)
{
var eattr = attr as ExposedAttribute;
dynamic dynObj = Activator.CreateInstance(type);
pyScope.SetVariable(eattr.Name, dynObj);
}
}
}
}
public void ExecuteFile(string file)
{
if(Utils.FileExists(file))
{
Execute(Utils.ReadAllText(file));
}
else
{
throw new Exception("The file \"" + file + "\" was not found on the system.");
}
}
public void Execute(string py)
{
Console.WriteLine("");
pySource = python.CreateScriptSourceFromString(py, Microsoft.Scripting.SourceCodeKind.AutoDetect);
pySource.Execute(pyScope);
Console.Write($"{SaveSystem.CurrentSave.Username}@{SaveSystem.CurrentSave.SystemName}:~$ ");
}
}
public class ExposedAttribute : Attribute
{
/// <summary>
/// Marks the following class as exposed to the ShiftOS Scripting System.
/// If applied to a non-static class, the ShiftOS engine will see this class as a Lua object of the specified name.
/// </summary>
/// <param name="objName">The object name that the Scripting System should use as the variable name for scripts.</param>
public ExposedAttribute(string objName)
/// <param name="name">The name of the Lua object</param>
public ExposedAttribute(string name)
{
Name = objName;
Name = name;
}
public string Name { get; set; }
public string Name { get; private set; }
}
}

10
ShiftOS_TheReturn/ShiftOS.Engine.csproj
View file

@ -113,12 +113,6 @@
</Reference>
</ItemGroup>
<ItemGroup>
<Compile Include="AltTabWindow.cs">
<SubType>Form</SubType>
</Compile>
<Compile Include="AltTabWindow.Designer.cs">
<DependentUpon>AltTabWindow.cs</DependentUpon>
</Compile>
<Compile Include="AppearanceManager.cs" />
<Compile Include="AppLauncherDaemon.cs" />
<Compile Include="Commands.cs" />
@ -127,7 +121,6 @@
<Compile Include="CrashHandler.Designer.cs">
<DependentUpon>CrashHandler.cs</DependentUpon>
</Compile>
<Compile Include="CtrlTabMenu.cs" />
<Compile Include="Desktop.cs" />
<Compile Include="FileSkimmerBackend.cs" />
<Compile Include="Infobox.cs" />
@ -153,9 +146,6 @@
<Compile Include="TutorialManager.cs" />
<Compile Include="VirusEngine.cs" />
<Compile Include="WinOpenAttribute.cs" />
<EmbeddedResource Include="AltTabWindow.resx">
<DependentUpon>AltTabWindow.cs</DependentUpon>
</EmbeddedResource>
<EmbeddedResource Include="Infobox.resx">
<DependentUpon>Infobox.cs</DependentUpon>
</EmbeddedResource>

140
ShiftOS_TheReturn/VirusEngine.cs
View file

@ -28,6 +28,7 @@ using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Newtonsoft.Json;
using static ShiftOS.Objects.ShiftFS.Utils;
@ -35,130 +36,53 @@ namespace ShiftOS.Engine
{
public static class VirusEngine
{
private static List<Virus> _infections = new List<Virus>();
public static void Add(Virus virus)
{
_infections.Add(virus);
}
public static void InfectFile(string file, string virusid)
{
string existing = "";
var infected = new List<string>();
var hData = GetHeaderText(file);
if(probeFile(file, out existing) == true)
if (hData == "")
{
existing = existing.Replace("<<VIRUSENGINE_HEAD>>", "").Replace("<<VIRUSENGINE_END>>", "");
existing += ";" + virusid;
existing = "<<VIRUSENGINE_HEAD>>" + existing + "<<VIRUSENGINE_END>>";
string c = ReadAllText(file);
string temp = "";
if(probeFile(file, out temp) == true)
{
c = c.Replace(temp, existing);
}
else
{
c = existing + c;
}
WriteAllText(file, c);
return;
infected.Add(virusid);
}
else
{
existing = "<<VIRUSENGINE_HEAD>>" + virusid + "<<VIRUSENGINE_END>>";
string c = ReadAllText(file);
c = existing + c;
WriteAllText(file, c);
return;
infected = JsonConvert.DeserializeObject<List<string>>(hData);
if (!infected.Contains(virusid))
infected.Add(virusid);
}
SetHeaderText(file, JsonConvert.SerializeObject(infected));
}
public static void DisinfectFile(string file, int threatlevel)
{
var infected = new List<string>();
var hData = GetHeaderText(file);
if (hData != "")
{
infected = JsonConvert.DeserializeObject<List<string>>(hData);
for (int i = 0; i < infected.Count; i++)
{
string[] splitID = infected[i].Split('.');
int th = Convert.ToInt32(splitID[splitID.Length - 1]);
if (th <= threatlevel)
{
infected.RemoveAt(i);
}
}
}
SetHeaderText(file, JsonConvert.SerializeObject(infected));
}
internal static string[] FindAllVirusesInFile(string file)
{
string existing = "";
if(probeFile(file, out existing) == true)
{
existing = existing.Replace("<<VIRUSENGINE_HEAD>>", "").Replace("<<VIRUSENGINE_END>>", "");
return existing.Split(new[] { ";" }, StringSplitOptions.RemoveEmptyEntries); ;
}
else
{
return null;
}
string hdata = GetHeaderText(file);
return (hdata != "") ? new string[0] : JsonConvert.DeserializeObject<string[]>(hdata);
}
private static bool probeFile(string file, out string existing)
{
int startIndex = 0;
int endIndex = 0;
bool found = false;
string contents = ReadAllText(file);
for(int i = 0; i < contents.Length; i++)
{
string end = "<<VIRUSENGINE_END>>";
try
{
if (contents.Substring(i, end.Length) == end)
{
endIndex = i + end.Length;
found = true;
break;
}
}
catch { }
}
if (found == false) {
existing = "<<VIRUSENGINE_HEAD>><<VIRUSENGINE_END>>";
}
else
{
existing = contents.Substring(startIndex, endIndex);
}
return found;
}
public static void Infect(string virusid)
{
string[] id_split = virusid.Split('.');
foreach(var v in _infections)
{
if(v.Type == id_split[0])
{
if(v.Signature == id_split[1])
{
if(v.ThreatLevel == Convert.ToInt32(id_split[2]))
{
var t = new Thread(new ThreadStart(() =>
{
v.Activate();
}));
t.IsBackground = true;
t.Start();
return;
}
}
}
}
throw new Exception("Virus not found in the system.");
}
internal static void ProbeFileRaw(string filePath, out string existing)
{
probeFile(filePath, out existing);
}
}
public abstract class Virus