github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-14 17:13:39 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refact: some refactoring to proxy modules

Browse source
This commit is contained in:
evilsocket 2018-01-11 19:53:14 +01:00
parent 195650777f
commit 291e87de39
11 changed files with 332 additions and 315 deletions
Download patch file Download diff file Expand all files Collapse all files

39
modules/http_server.go
View file

@ -88,33 +88,30 @@ func (httpd *HttpServer) Configure() error {
}
func (httpd *HttpServer) Start() error {
if err := httpd.Configure(); err != nil {
if httpd.Running() == true {
return session.ErrAlreadyStarted
} else if err := httpd.Configure(); err != nil {
return err
}
if httpd.Running() == false {
httpd.SetRunning(true)
go func() {
log.Info("httpd server starting on http://%s", httpd.server.Addr)
err := httpd.server.ListenAndServe()
if err != nil && err != http.ErrServerClosed {
panic(err)
}
}()
httpd.SetRunning(true)
go func() {
log.Info("httpd server starting on http://%s", httpd.server.Addr)
err := httpd.server.ListenAndServe()
if err != nil && err != http.ErrServerClosed {
panic(err)
}
}()
return nil
}
return fmt.Errorf("httpd server already started.")
return nil
}
func (httpd *HttpServer) Stop() error {
if httpd.Running() == true {
httpd.SetRunning(false)
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
defer cancel()
return httpd.server.Shutdown(ctx)
} else {
return fmt.Errorf("httpd server already stopped.")
if httpd.Running() == false {
return session.ErrAlreadyStopped
}
httpd.SetRunning(false)
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
defer cancel()
return httpd.server.Shutdown(ctx)
}