| // Copyright 2019 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| //go:build linux || darwin |
| |
| package main |
| |
| import ( |
| "io" |
| "log" |
| "net/http" |
| |
| "golang.org/x/build/internal/migration" |
| ) |
| |
| func listenAndServeInternalModuleProxy() { |
| if migration.StopInternalModuleProxy { |
| log.Println("not running internal module proxy") |
| return |
| } |
| err := http.ListenAndServe(":8123", http.HandlerFunc(proxyModuleCache)) |
| log.Fatalf("error running internal module proxy: %v", err) |
| } |
| |
| // proxyModuleCache proxies requests to https://proxy.golang.org/ |
| func proxyModuleCache(w http.ResponseWriter, r *http.Request) { |
| proxyURL(w, r, "https://proxy.golang.org") |
| } |
| |
| func proxyURL(w http.ResponseWriter, r *http.Request, baseURL string) { |
| outReq, err := http.NewRequest("GET", baseURL+r.RequestURI, nil) |
| if err != nil { |
| http.Error(w, "invalid URL", http.StatusBadRequest) |
| return |
| } |
| outReq = outReq.WithContext(r.Context()) |
| outReq.Header = r.Header.Clone() |
| res, err := http.DefaultClient.Do(outReq) |
| if err != nil { |
| http.Error(w, err.Error(), http.StatusInternalServerError) |
| return |
| } |
| defer res.Body.Close() |
| for k, vv := range res.Header { |
| for _, v := range vv { |
| w.Header().Add(k, v) |
| } |
| } |
| if res.StatusCode/100 != 2 && res.StatusCode != 410 { |
| log.Printf("modproxy: proxying HTTP %s response from backend for %s, %s %s", res.Status, r.RemoteAddr, r.Method, r.RequestURI) |
| } |
| |
| w.WriteHeader(res.StatusCode) |
| io.Copy(w, res.Body) |
| } |