54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package proxy
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/fsouza/go-dockerclient"
|
|
. "github.com/weaveworks/weave/common"
|
|
)
|
|
|
|
type createExecInterceptor struct{ proxy *Proxy }
|
|
|
|
func (i *createExecInterceptor) InterceptRequest(r *http.Request) error {
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
r.Body.Close()
|
|
r.Body = ioutil.NopCloser(bytes.NewReader(body))
|
|
|
|
options := docker.CreateExecOptions{}
|
|
if err := json.Unmarshal(body, &options); err != nil {
|
|
return err
|
|
}
|
|
|
|
container, err := inspectContainerInPath(i.proxy.client, r.URL.Path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, hasWeaveWait := container.Volumes["/w"]; !hasWeaveWait {
|
|
return nil
|
|
}
|
|
|
|
cidrs, err := i.proxy.weaveCIDRsFromConfig(container.Config, container.HostConfig)
|
|
if err != nil {
|
|
Info.Printf("Ignoring container %s due to %s", container.ID, err)
|
|
return nil
|
|
}
|
|
|
|
Info.Printf("Exec in container %s with WEAVE_CIDR \"%s\"", container.ID, strings.Join(cidrs, " "))
|
|
cmd := append(weaveWaitEntrypoint, "-s")
|
|
options.Cmd = append(cmd, options.Cmd...)
|
|
|
|
return marshalRequestBody(r, options)
|
|
}
|
|
|
|
func (i *createExecInterceptor) InterceptResponse(r *http.Response) error {
|
|
return nil
|
|
}
|