From 9ae7c76542df4239a955d4ac2969b3480c47b674 Mon Sep 17 00:00:00 2001
From: Uthman Eqbal <60008549+UthmanPortSwigger@users.noreply.github.com>
Date: Fri, 27 Aug 2021 15:42:51 +0100
Subject: [PATCH] Update Add-Headers-Python.py

Remove unnecessary comments.
---
 Custom-Headers/Python/Add-Headers-Python.py | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/Custom-Headers/Python/Add-Headers-Python.py b/Custom-Headers/Python/Add-Headers-Python.py
index 2d26dbd..5000bf2 100644
--- a/Custom-Headers/Python/Add-Headers-Python.py
+++ b/Custom-Headers/Python/Add-Headers-Python.py
@@ -3,32 +3,31 @@
 from burp import IHttpRequestResponse
 from burp import IRequestInfo
 
-# BurpExtender Class contaning all functions used to interact with Burp Suite Extender API
+# implement BurpExtender Class
 class BurpExtender(IBurpExtender, IHttpListener):
 
     # define registerExtenderCallbacks: From IBurpExtender Interface
     def registerExtenderCallbacks(self, callbacks):
 
-        # keep a reference to our callbacks object (Burp Extensibility Feature)
+        # keep a reference to our callbacks object
         self._callbacks = callbacks
-        # obtain an extension helpers object (Burp Extensibility Feature)
-        # http://portswigger.net/burp/extender/api/burp/IExtensionHelpers.html
+        # obtain an extension helpers object
         self._helpers = callbacks.getHelpers()
-        # set our extension name that will display in Extender Tab
-        self._callbacks.setExtensionName("Add a custom header")
-        # register ourselves as an HTTP listener
+        # set our extension name that will display in Extender > Output Tab
+        self._callbacks.setExtensionName("Add-Headers-Python")
+        # register an HTTP listener
         callbacks.registerHttpListener(self)
 
     # define processHttpMessage: From IHttpListener Interface
     def processHttpMessage(self, toolFlag, messageIsRequest, messageInfo):
         # only work on requests:
         if messageIsRequest:
-            request = messageInfo.getRequest()                                          #get Request from IHttpRequestResponse instance
+            request = messageInfo.getRequest()                                          # get Request from IHttpRequestResponse instance
             requestStr = self._callbacks.getHelpers().bytesToString(request)
             requestParsed = self._helpers.analyzeRequest(request)
             body = requestStr[requestParsed.getBodyOffset():]
             headers = requestParsed.getHeaders()
-            headers.add('Header1: value1')                                              #change header name and value here 
+            headers.add('Header1: value1')                                              # change header name and value here 
             httpRequest = self._callbacks.getHelpers().buildHttpMessage(headers, body)
             messageInfo.setRequest(httpRequest)
-            return
\ No newline at end of file
+            return