Skip to content

Commit

Permalink
Release v0.12.1 (#323)
Browse files Browse the repository at this point in the history
Retry dns calls
Log endpoint calls
Improve message
  • Loading branch information
varunsh-coder authored Oct 25, 2022
1 parent dba4849 commit 6b2df4f
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 4 deletions.
2 changes: 1 addition & 1 deletion agent.go
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ func Run(ctx context.Context, configFilePath string, hostDNSServer DNSServer,
config.DisableSudo = false
apiclient.DisableTelemetry = true
config.DisableFileMonitoring = true
WriteAnnotation("StepSecurity Harden Runner disabled. A subscription is required for private repositories. Please start a free trial at https://stepsecurity.io")
WriteAnnotation("StepSecurity Harden Runner is disabled. A subscription is required for private repositories. Please start a free trial at https://www.stepsecurity.io")
}
}

Expand Down
19 changes: 16 additions & 3 deletions dnsproxy.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ import (
"fmt"
"io/ioutil"
"math"
"net/http"
"strings"
"sync"

Expand Down Expand Up @@ -113,9 +114,21 @@ func (proxy *DNSProxy) isAllowedDomain(domain string) bool {

func (proxy *DNSProxy) ResolveDomain(domain string) (*Answer, error) {
url := fmt.Sprintf("https://dns.google/resolve?name=%s&type=a", domain)
resp, err := proxy.ApiClient.Client.Get(url)
if err != nil {
return nil, fmt.Errorf("error in response from dns.google %v", err)

retryCounter := 0
var httpError error
var resp *http.Response
for retryCounter < 2 {
resp, httpError = proxy.ApiClient.Client.Get(url)
if httpError != nil {
retryCounter++
} else {
break
}
}

if httpError != nil {
return nil, fmt.Errorf("error in response from dns.google %v", httpError)
}

defer resp.Body.Close()
Expand Down
1 change: 1 addition & 0 deletions eventhandler.go
Original file line number Diff line number Diff line change
Expand Up @@ -189,6 +189,7 @@ func (eventHandler *EventHandler) handleNetworkEvent(event *Event) {
}
reverseLookUp := eventHandler.DNSProxy.GetReverseIPLookup(event.IPAddress)
eventHandler.ApiClient.sendNetConnection(eventHandler.CorrelationId, eventHandler.Repo, event.IPAddress, event.Port, reverseLookUp, "", event.Timestamp, tool)
WriteLog(fmt.Sprintf("endpoint called ip address:port %s:%s, domain: %s", event.IPAddress, event.Port, reverseLookUp))
eventHandler.ProcessConnectionMap[cacheKey] = true
}
}
Expand Down

0 comments on commit 6b2df4f

Please sign in to comment.