diff --git a/cli/src/main/java/ysomap/cli/Session.java b/cli/src/main/java/ysomap/cli/Session.java index 8af8bf0..4ba7846 100644 --- a/cli/src/main/java/ysomap/cli/Session.java +++ b/cli/src/main/java/ysomap/cli/Session.java @@ -122,22 +122,28 @@ public HashMap update(Class clazz){ return settings; } - public void setValue(String key, String value) throws ArgumentsMissMatchException { - if("serializeType".equals(key) && payload != null){ - payload.setSerializeType(value); - return; - }else if("encoder".equals(key) && payload != null){ - payload.setEncoder(value); - return; - }else if("output".equals(key) && payload != null){ - payload.setOutputType(value); - return; - }else if("serialVersionUID".equals(key) && payload != null){ - payload.setSerialVersionUID(value); - return; - }else if("checkRunning".equals(key)){ - isCheckRunning = Boolean.parseBoolean(value); - return; + public void setValue(String key, String value) throws Exception { + if(payload != null && payload.has(key)){ + if("serializeType".equals(key) && payload != null){ + payload.setSerializeType(value); + return; + }else if("encoder".equals(key) && payload != null){ + payload.setEncoder(value); + return; + }else if("output".equals(key) && payload != null){ + payload.setOutputType(value); + return; + }else if("serialVersionUID".equals(key) && payload != null){ + payload.setSerialVersionUID(value); + return; + }else if("checkRunning".equals(key)){ + isCheckRunning = Boolean.parseBoolean(value); + return; + } + else if("wrapped".equals(key)){ + payload.set("wrapped",Boolean.parseBoolean(value)); + return; + } } if(exploit != null && exploit.has(key)){ @@ -204,6 +210,7 @@ public void printSessionDetails(){ Logger.normal("Current Serializer Encoder: "+ColorStyle.makeWordRed(payload.getEncoder())); Logger.normal("Current Serializer Output Type: "+ColorStyle.makeWordRed(payload.getOutputType())); Logger.normal("Current Serializer serialVersionUID: "+ColorStyle.makeWordRed(payload.getSerialVersionUID())); + if(payload.has("wrapped")){Logger.normal("Current wrapped(to bypass blacklist): "+ColorStyle.makeWordRed(payload.get("wrapped")));} if(bullet == null){ Printer.printCandidates("bullets", clazz, false, null); }