1. import java.io.IOException;
  2. import java.io.PrintWriter;
  3. import java.io.StringWriter;
  4. import java.util.HashMap;
  5. import java.util.Map;
  7. public class MetadataStripper {
  9. /**
  10. * Strips metadata from response headers.
  11. *
  12. * @param responseHeaders A map of response headers.
  13. * @param stripFlags A string containing flags to control stripping behavior.
  14. * Valid flags: "user-agent", "content-length", "date", "server", "cache-control", "transfer-encoding", "content-type", "location", "etag", "last-modified"
  15. * @return A map of stripped response headers. Returns an empty map if input is null.
  16. */
  17. public static Map<String, String> stripMetadata(Map<String, String> responseHeaders, String stripFlags) {
  18. if (responseHeaders == null) {
  19. return new HashMap<>();
  20. }
  22. Map<String, String> strippedHeaders = new HashMap<>();
  24. if (stripFlags == null || stripFlags.isEmpty()) {
  25. return responseHeaders; // No flags, return original headers
  26. }
  28. String[] flags = stripFlags.split(",");
  29. for (String flag : flags) {
  30. String trimmedFlag = flag.trim(); // remove leading/trailing whitespace
  31. if (trimmedFlag.equalsIgnoreCase("user-agent") || trimmedFlag.equalsIgnoreCase("content-length") || trimmedFlag.equalsIgnoreCase("date") || trimmedFlag.equalsIgnoreCase("server") || trimmedFlag.equalsIgnoreCase("cache-control") || trimmedFlag.equalsIgnoreCase("transfer-encoding") || trimmedFlag.equalsIgnoreCase("content-type") || trimmedFlag.equalsIgnoreCase("location") || trimmedFlag.equalsIgnoreCase("etag") || trimmedFlag.equalsIgnoreCase("last-modified")) {
  32. strippedHeaders.put(trimmedFlag, null); // Set value to null to effectively remove the header
  33. }
  34. }
  36. return strippedHeaders;
  37. }
  39. public static void main(String[] args) throws IOException {
  40. // Example Usage
  41. Map<String, String> headers = new HashMap<>();
  42. headers.put("User-Agent", "MyUserAgent");
  43. headers.put("Content-Length", "1234");
  44. headers.put("Date", "Thu, 26 Oct 2023 10:00:00 GMT");
  45. headers.put("Server", "Apache/2.4.41");
  46. headers.put("Cache-Control", "no-cache");
  47. headers.put("Content-Type", "text/html");
  49. String stripFlags = "user-agent,content-length,date";
  51. Map<String, String> stripped = stripMetadata(headers, stripFlags);
  53. StringWriter sw = new StringWriter();
  54. PrintWriter pw = new PrintWriter(sw);
  55. for (Map.Entry<String, String> entry : stripped.entrySet()) {
  56. pw.println(entry.getKey() + ": " + entry.getValue());
  57. }
  58. System.out.println(sw.toString());
  59. }
  60. }

Add your comment