34 lines
803 B
Go
34 lines
803 B
Go
|
package geolocate
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net/http"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/ooni/probe-cli/v3/internal/engine/httpheader"
|
||
|
|
"github.com/ooni/probe-cli/v3/internal/httpx"
|
||
|
|
"github.com/ooni/probe-cli/v3/internal/model"
|
||
|
|
)
|
||
|
|
|
||
|
|
func cloudflareIPLookup(
|
||
|
|
ctx context.Context,
|
||
|
|
httpClient *http.Client,
|
||
|
|
logger model.Logger,
|
||
|
|
userAgent string,
|
||
|
|
) (string, error) {
|
||
|
|
data, err := (&httpx.APIClientTemplate{
|
||
|
|
BaseURL: "https://www.cloudflare.com",
|
||
|
|
HTTPClient: httpClient,
|
||
|
|
Logger: logger,
|
||
|
|
UserAgent: httpheader.CLIUserAgent(),
|
||
|
|
}).WithBodyLogging().Build().FetchResource(ctx, "/cdn-cgi/trace")
|
||
|
|
if err != nil {
|
||
|
|
return DefaultProbeIP, err
|
||
|
|
}
|
||
|
|
r := regexp.MustCompile("(?:ip)=(.*)")
|
||
|
|
ip := strings.Trim(string(r.Find(data)), "ip=")
|
||
|
|
logger.Debugf("cloudflare: body: %s", ip)
|
||
|
|
return ip, nil
|
||
|
|
}
|