zeldon
/
http-proxy-server
mirror of https://github.com/xzeldon/http-proxy-server.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(main.go): add validation for AUTH_USERNAME and AUTH_PASSWORD to ensure both are provided or neither is provided

This commit is contained in:
Timofey Gelazoniya 2023-10-31 05:48:14 +03:00
parent cacc311854
commit 8e0a899bf2
Signed by: zeldon
GPG Key ID: 047886915281DD2A
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.go
View File

@ -116,6 +116,10 @@ func main() {
flag.StringVar(&AUTH_PASSWORD, "password", "", "Password for proxy authentication") flag.StringVar(&AUTH_PASSWORD, "password", "", "Password for proxy authentication")
flag.Parse() flag.Parse()
if (AUTH_USERNAME == "" && AUTH_PASSWORD != "") || (AUTH_USERNAME != "" && AUTH_PASSWORD == "") {
log.Fatal("Error: Both username and password must be provided, or neither should be.")
}
server := &http.Server{ server := &http.Server{
Addr: ":" + PORT, Addr: ":" + PORT,
Handler: http.HandlerFunc(mainHandler), Handler: http.HandlerFunc(mainHandler),