diff --git a/src/main/java/com/jamesquinley/DelegationServices/delegationServices.java b/src/main/java/com/jamesquinley/DelegationServices/delegationServices.java index 674334a..8fb2194 100644 --- a/src/main/java/com/jamesquinley/DelegationServices/delegationServices.java +++ b/src/main/java/com/jamesquinley/DelegationServices/delegationServices.java @@ -1,6 +1,4 @@ package com.jamesquinley.DelegationServices; - -import com.jamesquinley.Asecure.electronicMailHandler; import com.jamesquinley.Asecure.secureCalls; import com.mongodb.BasicDBObject; import com.mongodb.ConnectionString; @@ -10,10 +8,7 @@ import com.mongodb.client.MongoClient; import com.mongodb.client.MongoClients; import com.mongodb.client.MongoCollection; import com.mongodb.client.MongoDatabase; -import org.apache.logging.log4j.core.util.JsonUtils; import org.bson.Document; -import org.bson.types.ObjectId; - import java.io.File; import java.io.FileWriter; import java.io.IOException; @@ -1830,7 +1825,7 @@ public class delegationServices { } else { System.out.println("Running in limited mode (Web Request)"); - if (requesttype == 73 || requesttype == 74 || requesttype == 75 || requesttype ==78) { + if (requesttype == 73 || requesttype == 74 || requesttype == 75 || requesttype ==78 || requesttype == 79) { if (requesttype == 73) { System.out.println("Login Request (USR&PASS)"); String username = (String) Lrequest.get("user"); @@ -2337,6 +2332,32 @@ public class delegationServices { dCollection.updateOne(searchQuery, updateQuery); delegationTimer.DeleteTargetRequestByID(Lrequest.get("_id")); } + if (requesttype == 79) { + BasicDBObject searchQuery = new BasicDBObject(); + searchQuery.append("_id", ClientID); + System.out.println("Init ClientTraceback"); + String policyDocument; + try { + policyDocument = Files.readString(Path.of(System.getProperty("user.home") + "/ocs/policy/launcher/legal/eula.txt")); + } catch (Exception EXR) { + System.out.println("Unable to pull file, fallback."); + policyDocument = "Policy file does not exist on server. This is an issue on our end. Error ID: " + EXR; + } + + BasicDBObject updateQuery = new BasicDBObject(); + BasicDBObject updateQuery2 = new BasicDBObject(); + BasicDBObject updateQuery3 = new BasicDBObject(); + updateQuery.append("$set", + new BasicDBObject().append("type", 80)); + updateQuery2.append("$set", + new BasicDBObject().append("policyDocument", policyDocument)); + updateQuery3.append("$set", + new BasicDBObject().append("status", true)); + dCollection.updateOne(searchQuery, updateQuery); + dCollection.updateOne(searchQuery, updateQuery2); + dCollection.updateOne(searchQuery, updateQuery3); + delegationTimer.DeleteTargetRequestByID(Lrequest.get("_id")); + } } else { System.out.println("Type does not exist (ERR4)"); BasicDBObject searchQuery = new BasicDBObject();