diff --git a/s3cr3txJavaTest.java b/s3cr3txJavaTest.java index c953e24..5fe1c55 100644 --- a/s3cr3txJavaTest.java +++ b/s3cr3txJavaTest.java @@ -1,3 +1,4 @@ +import io.github.pixee.security.BoundedLineReader; import java.io.*; import java.net.*; import javax.net.ssl.HttpsURLConnection; @@ -20,7 +21,7 @@ public static void getHTML(String website) throws Exception { String strResult = new String(); BufferedReader br = new BufferedReader(new InputStreamReader(conn.getInputStream())); String line = ""; - while ((line = br.readLine())!= null) { + while ((line = BoundedLineReader.readLine(br, 5_000_000))!= null) { strResult += line; } System.out.println("\nS3cr3tx encrypted text is friom: " + website); @@ -44,7 +45,7 @@ public static void getHTML(String website) throws Exception { String strResult2 = new String(); BufferedReader br2 = new BufferedReader(new InputStreamReader(conn2.getInputStream())); String line2 = ""; - while ((line2 = br2.readLine())!= null) { + while ((line2 = BoundedLineReader.readLine(br2, 5_000_000))!= null) { strResult2 += line2; } System.out.println("\nS3cr3tx decrypted text from: " + website); diff --git a/s3cr3txJava_protect_OpenAI_API_demo.java b/s3cr3txJava_protect_OpenAI_API_demo.java index d49fe12..639bb07 100644 --- a/s3cr3txJava_protect_OpenAI_API_demo.java +++ b/s3cr3txJava_protect_OpenAI_API_demo.java @@ -1,3 +1,4 @@ +import io.github.pixee.security.BoundedLineReader; import java.io.*; import java.net.*; import javax.net.ssl.HttpsURLConnection; @@ -24,7 +25,7 @@ public static String getS3cr3tx(String strInput){ String strResult = new String(); BufferedReader br = new BufferedReader(new InputStreamReader(conn.getInputStream())); String line = ""; - while ((line = br.readLine())!= null) { + while ((line = BoundedLineReader.readLine(br, 5_000_000))!= null) { strResult += line; } return strResult; @@ -55,7 +56,7 @@ public static String setS3cr3tx(String strInput){ String strResult = new String(); BufferedReader br = new BufferedReader(new InputStreamReader(conn.getInputStream())); String line = ""; - while ((line = br.readLine())!= null) { + while ((line = BoundedLineReader.readLine(br, 5_000_000))!= null) { strResult += line; } return strResult; @@ -82,7 +83,7 @@ public static void main(String[] args) throws Exception { String strResult = new String(); BufferedReader br = new BufferedReader(new InputStreamReader(conn.getInputStream())); String line = ""; - while ((line = br.readLine())!= null) { + while ((line = BoundedLineReader.readLine(br, 5_000_000))!= null) { strResult += line; } System.out.println("\n" + strResult); @@ -97,7 +98,7 @@ public static void main(String[] args) throws Exception { String strResult2 = new String(); BufferedReader br2 = new BufferedReader(new InputStreamReader(conn2.getInputStream())); String lines = ""; - while ((lines = br2.readLine())!= null) { + while ((lines = BoundedLineReader.readLine(br2, 5_000_000))!= null) { strResult2 += lines; } System.out.println("\n" + strResult2); @@ -106,4 +107,4 @@ public static void main(String[] args) throws Exception { e.printStackTrace(); } } -} \ No newline at end of file +}