TheArchive/tailscale
1
0
Fork 0
mirror of https://github.com/tailscale/tailscale.git synced 2025-12-03 10:31:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

ipn/ipnlocal: add some peerapi tests

Browse Source 
Updates tailscale/corp#1594

Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
This commit is contained in:
Brad Fitzpatrick
2021-04-19 20:21:48 -07:00
parent f72a120016
commit 0d1550898e
2 changed files with 155 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ipn/ipnlocal/peerapi.go
View File

@@ -290,7 +290,6 @@ func (pln *peerAPIListener) serve() {
remoteAddr: ipp,
peerNode: peerNode,
peerUser: peerUser,
lb: pln.lb,
}
httpServer := &http.Server{
Handler: h,
@@ -324,7 +323,6 @@ type peerAPIHandler struct {
isSelf bool // whether peerNode is owned by same user as this node
peerNode *tailcfg.Node // peerNode is who's making the request
peerUser tailcfg.UserProfile // profile of peerNode
lb *LocalBackend
}
func (h *peerAPIHandler) logf(format string, a ...interface{}) {
@@ -333,7 +331,7 @@ func (h *peerAPIHandler) logf(format string, a ...interface{}) {
func (h *peerAPIHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if strings.HasPrefix(r.URL.Path, "/v0/put/") {
h.put(w, r)
h.handlePeerPut(w, r)
return
}
who := h.peerUser.DisplayName
@@ -405,7 +403,7 @@ func (f *incomingFile) PartialFile() ipn.PartialFile {
}
}
func (h *peerAPIHandler) put(w http.ResponseWriter, r *http.Request) {
func (h *peerAPIHandler) handlePeerPut(w http.ResponseWriter, r *http.Request) {
if !h.isSelf {
http.Error(w, "not owner", http.StatusForbidden)
return