Add support for AWS X-Forwarded-For overwrite (#21)

RyanJarv · May 28, 2020 · d561b5f · d561b5f
1 parent 9cf7aee
commit d561b5f
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/IPRotate.py b/IPRotate.py
@@ -99,7 +99,11 @@ def startAPIGateway(self):
 				restApiId=self.create_api_response['id'],
 				resourceId=get_resource_response['items'][0]['id'],
 				httpMethod='ANY',
-				authorizationType='NONE'
+				authorizationType='NONE',
+				requestParameters={
+					'method.request.path.proxy':True,
+					'method.request.header.X-My-X-Forwarded-For':True
+                                }
 			)
 
 			self.awsclient.put_integration(
@@ -109,7 +113,11 @@ def startAPIGateway(self):
 				httpMethod='ANY',
 				integrationHttpMethod='ANY',
 				uri=self.getTargetProtocol()+'://'+self.target_host.text + '/',
-				connectionType='INTERNET'
+				connectionType='INTERNET',
+				requestParameters={
+					'integration.request.path.proxy':'method.request.path.proxy',
+                                        'integration.request.header.X-Forwarded-For': 'method.request.header.X-My-X-Forwarded-For'
+				}
 			)
 
 			self.awsclient.put_method(
@@ -118,7 +126,8 @@ def startAPIGateway(self):
 				httpMethod='ANY',
 				authorizationType='NONE',
 				requestParameters={
-					'method.request.path.proxy':True
+					'method.request.path.proxy':True,
+					'method.request.header.X-My-X-Forwarded-For':True
 				}
 			)
 
@@ -131,7 +140,8 @@ def startAPIGateway(self):
 				uri= self.getTargetProtocol()+'://'+self.target_host.text+'/{proxy}',
 				connectionType= 'INTERNET',
 				requestParameters={
-					'integration.request.path.proxy':'method.request.path.proxy'
+					'integration.request.path.proxy':'method.request.path.proxy',
+                                        'integration.request.header.X-Forwarded-For': 'method.request.header.X-My-X-Forwarded-For'
 				}
 			)