File tree 1 file changed +10
-7
lines changed
1 file changed +10
-7
lines changed Original file line number Diff line number Diff line change @@ -383,6 +383,14 @@ def generate_page(request: Request) -> HTMLResponse:
383
383
return templates .TemplateResponse ("generate_new.html" , generate_template_params )
384
384
385
385
386
+ def _get_src_ip (request ):
387
+ return request .headers .get (switchboard_settings .REAL_IP_HEADER ) or (
388
+ request .client .host
389
+ if (request .client and request .client .host != "testclient" )
390
+ else ""
391
+ )
392
+
393
+
386
394
@app .post (
387
395
"/generate" ,
388
396
tags = ["Create Canarytokens" ],
@@ -449,10 +457,7 @@ async def generate(request: Request) -> AnyTokenResponse: # noqa: C901 # gen i
449
457
else :
450
458
kube_config = None
451
459
canarytoken = Canarytoken ()
452
- print (request .client )
453
- src_ip = request .headers .get (switchboard_settings .REAL_IP_HEADER ) or (
454
- request .client .host if request .client else ""
455
- )
460
+ src_ip = _get_src_ip (request )
456
461
x_forwarded_for = request .headers .get ("x-forwarded-for" ) or ""
457
462
458
463
canarydrop = Canarydrop (
@@ -776,9 +781,7 @@ async def api_generate( # noqa: C901 # gen is large
776
781
kube_config = None
777
782
canarytoken = Canarytoken ()
778
783
779
- src_ip = request .headers .get (switchboard_settings .REAL_IP_HEADER ) or (
780
- request .client .host if request .client else ""
781
- )
784
+ src_ip = _get_src_ip (request )
782
785
x_forwarded_for = request .headers .get ("x-forwarded-for" ) or ""
783
786
784
787
canarydrop = Canarydrop (
You can’t perform that action at this time.
0 commit comments