Jiri Simsa | d7616c9 | 2015-03-24 23:44:30 -0700 | [diff] [blame] | 1 | // Copyright 2015 The Vanadium Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style |
| 3 | // license that can be found in the LICENSE file. |
| 4 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 5 | package rpc |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 6 | |
| 7 | import ( |
Bogdan Caprita | 9592d9f | 2015-01-08 22:15:16 -0800 | [diff] [blame] | 8 | "errors" |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 9 | "fmt" |
| 10 | "io" |
Cosmos Nicolaou | bae615a | 2014-08-27 23:32:31 -0700 | [diff] [blame] | 11 | "net" |
Asim Shankar | b54d764 | 2014-06-05 13:08:04 -0700 | [diff] [blame] | 12 | "reflect" |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 13 | "strings" |
| 14 | "sync" |
| 15 | "time" |
| 16 | |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 17 | "v.io/x/lib/netstate" |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 18 | |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 19 | "v.io/v23" |
Jiri Simsa | 6ac9522 | 2015-02-23 16:11:49 -0800 | [diff] [blame] | 20 | "v.io/v23/context" |
Matt Rosencrantz | 88be118 | 2015-04-27 13:45:43 -0700 | [diff] [blame] | 21 | "v.io/v23/i18n" |
Todd Wang | 5082a55 | 2015-04-02 10:56:11 -0700 | [diff] [blame] | 22 | "v.io/v23/namespace" |
Jiri Simsa | 6ac9522 | 2015-02-23 16:11:49 -0800 | [diff] [blame] | 23 | "v.io/v23/naming" |
| 24 | "v.io/v23/options" |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 25 | "v.io/v23/rpc" |
Jiri Simsa | 6ac9522 | 2015-02-23 16:11:49 -0800 | [diff] [blame] | 26 | "v.io/v23/security" |
Todd Wang | 387d8a4 | 2015-03-30 17:09:05 -0700 | [diff] [blame] | 27 | "v.io/v23/security/access" |
Jiri Simsa | 6ac9522 | 2015-02-23 16:11:49 -0800 | [diff] [blame] | 28 | "v.io/v23/vdl" |
| 29 | "v.io/v23/verror" |
Jiri Simsa | 6ac9522 | 2015-02-23 16:11:49 -0800 | [diff] [blame] | 30 | "v.io/v23/vom" |
| 31 | "v.io/v23/vtrace" |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 32 | |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 33 | "v.io/x/ref/lib/apilog" |
Jiri Simsa | 574ec4b | 2015-08-11 09:31:37 -0700 | [diff] [blame] | 34 | "v.io/x/ref/lib/pubsub" |
Jiri Simsa | ffceefa | 2015-02-28 11:03:34 -0800 | [diff] [blame] | 35 | "v.io/x/ref/lib/stats" |
Suharsh Sivakumar | dcc11d7 | 2015-05-11 12:19:20 -0700 | [diff] [blame] | 36 | "v.io/x/ref/runtime/internal/lib/publisher" |
| 37 | inaming "v.io/x/ref/runtime/internal/naming" |
| 38 | "v.io/x/ref/runtime/internal/rpc/stream" |
| 39 | "v.io/x/ref/runtime/internal/rpc/stream/manager" |
| 40 | "v.io/x/ref/runtime/internal/rpc/stream/vc" |
Cosmos Nicolaou | 185c0c6 | 2015-04-13 21:22:43 -0700 | [diff] [blame] | 41 | ) |
Cosmos Nicolaou | 28dabfc | 2014-12-15 22:51:07 -0800 | [diff] [blame] | 42 | |
Cosmos Nicolaou | 185c0c6 | 2015-04-13 21:22:43 -0700 | [diff] [blame] | 43 | var ( |
| 44 | // These errors are intended to be used as arguments to higher |
| 45 | // level errors and hence {1}{2} is omitted from their format |
| 46 | // strings to avoid repeating these n-times in the final error |
| 47 | // message visible to the user. |
| 48 | errResponseEncoding = reg(".errResponseEncoding", "failed to encode RPC response {3} <-> {4}{:5}") |
| 49 | errResultEncoding = reg(".errResultEncoding", "failed to encode result #{3} [{4}]{:5}") |
| 50 | errFailedToResolveToEndpoint = reg(".errFailedToResolveToEndpoint", "failed to resolve {3} to an endpoint") |
| 51 | errFailedToResolveProxy = reg(".errFailedToResolveProxy", "failed to resolve proxy {3}{:4}") |
| 52 | errFailedToListenForProxy = reg(".errFailedToListenForProxy", "failed to listen on {3}{:4}") |
Suharsh Sivakumar | dcc11d7 | 2015-05-11 12:19:20 -0700 | [diff] [blame] | 53 | errInternalTypeConversion = reg(".errInternalTypeConversion", "failed to convert {3} to v.io/x/ref/runtime/internal/naming.Endpoint") |
Cosmos Nicolaou | 185c0c6 | 2015-04-13 21:22:43 -0700 | [diff] [blame] | 54 | errFailedToParseIP = reg(".errFailedToParseIP", "failed to parse {3} as an IP host") |
Ali Ghassemi | bac3403 | 2015-04-30 18:27:57 -0700 | [diff] [blame] | 55 | errUnexpectedSuffix = reg(".errUnexpectedSuffix", "suffix {3} was not expected because either server has the option IsLeaf set to true or it served an object and not a dispatcher") |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 56 | errNoListeners = reg(".errNoListeners", "failed to ceate any listeners{:3}") |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 57 | ) |
| 58 | |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 59 | // state for each requested listen address |
| 60 | type listenState struct { |
| 61 | protocol, address string |
| 62 | ln stream.Listener |
| 63 | lep naming.Endpoint |
| 64 | lnerr, eperr error |
| 65 | roaming bool |
| 66 | // We keep track of all of the endpoints, the port and a copy of |
| 67 | // the original listen endpoint for use with roaming network changes. |
| 68 | ieps []*inaming.Endpoint // list of currently active eps |
| 69 | port string // port to use for creating new eps |
| 70 | protoIEP inaming.Endpoint // endpoint to use as template for new eps (includes rid, versions etc) |
| 71 | } |
| 72 | |
| 73 | // state for each requested proxy |
| 74 | type proxyState struct { |
| 75 | endpoint naming.Endpoint |
Mike Burrows | dc6b360 | 2015-02-05 15:52:12 -0800 | [diff] [blame] | 76 | err error |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 77 | } |
| 78 | |
| 79 | type dhcpState struct { |
| 80 | name string |
Cosmos Nicolaou | 11c0ca1 | 2015-04-23 16:23:43 -0700 | [diff] [blame] | 81 | publisher *pubsub.Publisher |
| 82 | stream *pubsub.Stream |
| 83 | ch chan pubsub.Setting // channel to receive dhcp settings over |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 84 | err error // error status. |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 85 | watchers map[chan<- rpc.NetworkChange]struct{} |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 86 | } |
| 87 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 88 | type server struct { |
| 89 | sync.Mutex |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 90 | // context used by the server to make internal RPCs, error messages etc. |
Cosmos Nicolaou | 00fe9a4 | 2015-04-24 14:18:01 -0700 | [diff] [blame] | 91 | ctx *context.T |
| 92 | cancel context.CancelFunc // function to cancel the above context. |
| 93 | state serverState // track state of the server. |
| 94 | streamMgr stream.Manager // stream manager to listen for new flows. |
| 95 | publisher publisher.Publisher // publisher to publish mounttable mounts. |
Asim Shankar | 99b18a7 | 2015-04-25 23:19:28 -0700 | [diff] [blame] | 96 | dc vc.DischargeClient // fetches discharges of blessings |
Cosmos Nicolaou | 00fe9a4 | 2015-04-24 14:18:01 -0700 | [diff] [blame] | 97 | listenerOpts []stream.ListenerOpt // listener opts for Listen. |
| 98 | settingsPublisher *pubsub.Publisher // pubsub publisher for dhcp |
| 99 | settingsName string // pubwsub stream name for dhcp |
| 100 | dhcpState *dhcpState // dhcpState, nil if not using dhcp |
| 101 | principal security.Principal |
| 102 | blessings security.Blessings |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 103 | |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 104 | // maps that contain state on listeners. |
| 105 | listenState map[*listenState]struct{} |
| 106 | listeners map[stream.Listener]struct{} |
| 107 | |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 108 | // state of proxies keyed by the name of the proxy |
| 109 | proxies map[string]proxyState |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 110 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 111 | disp rpc.Dispatcher // dispatcher to serve RPCs |
| 112 | dispReserved rpc.Dispatcher // dispatcher for reserved methods |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 113 | active sync.WaitGroup // active goroutines we've spawned. |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 114 | stoppedChan chan struct{} // closed when the server has been stopped. |
| 115 | preferredProtocols []string // protocols to use when resolving proxy name to endpoint. |
Jungho Ahn | 25545d3 | 2015-01-26 15:14:14 -0800 | [diff] [blame] | 116 | // We cache the IP networks on the device since it is not that cheap to read |
| 117 | // network interfaces through os syscall. |
| 118 | // TODO(jhahn): Add monitoring the network interface changes. |
| 119 | ipNets []*net.IPNet |
Todd Wang | 5082a55 | 2015-04-02 10:56:11 -0700 | [diff] [blame] | 120 | ns namespace.T |
Jungho Ahn | 25545d3 | 2015-01-26 15:14:14 -0800 | [diff] [blame] | 121 | servesMountTable bool |
Robin Thellend | 89e9523 | 2015-03-24 13:48:48 -0700 | [diff] [blame] | 122 | isLeaf bool |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 123 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 124 | // TODO(cnicolaou): add roaming stats to rpcStats |
| 125 | stats *rpcStats // stats for this server. |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 126 | } |
| 127 | |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 128 | type serverState int |
| 129 | |
| 130 | const ( |
| 131 | initialized serverState = iota |
| 132 | listening |
| 133 | serving |
| 134 | publishing |
| 135 | stopping |
| 136 | stopped |
| 137 | ) |
| 138 | |
| 139 | // Simple state machine for the server implementation. |
| 140 | type next map[serverState]bool |
| 141 | type transitions map[serverState]next |
| 142 | |
| 143 | var ( |
| 144 | states = transitions{ |
| 145 | initialized: next{listening: true, stopping: true}, |
| 146 | listening: next{listening: true, serving: true, stopping: true}, |
| 147 | serving: next{publishing: true, stopping: true}, |
| 148 | publishing: next{publishing: true, stopping: true}, |
| 149 | stopping: next{}, |
| 150 | stopped: next{}, |
| 151 | } |
| 152 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 153 | externalStates = map[serverState]rpc.ServerState{ |
| 154 | initialized: rpc.ServerInit, |
| 155 | listening: rpc.ServerActive, |
| 156 | serving: rpc.ServerActive, |
| 157 | publishing: rpc.ServerActive, |
| 158 | stopping: rpc.ServerStopping, |
| 159 | stopped: rpc.ServerStopped, |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 160 | } |
| 161 | ) |
| 162 | |
| 163 | func (s *server) allowed(next serverState, method string) error { |
| 164 | if states[s.state][next] { |
| 165 | s.state = next |
| 166 | return nil |
| 167 | } |
Jiri Simsa | 074bf36 | 2015-02-17 09:29:45 -0800 | [diff] [blame] | 168 | return verror.New(verror.ErrBadState, s.ctx, fmt.Sprintf("%s called out of order or more than once", method)) |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 169 | } |
| 170 | |
| 171 | func (s *server) isStopState() bool { |
| 172 | return s.state == stopping || s.state == stopped |
| 173 | } |
| 174 | |
Matt Rosencrantz | 98d6d7c | 2015-09-04 12:34:08 -0700 | [diff] [blame^] | 175 | var _ rpc.DeprecatedServer = (*server)(nil) |
Benjamin Prosnitz | fdfbf7b | 2014-10-08 09:47:21 -0700 | [diff] [blame] | 176 | |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 177 | func InternalNewServer( |
| 178 | ctx *context.T, |
| 179 | streamMgr stream.Manager, |
| 180 | ns namespace.T, |
Cosmos Nicolaou | 00fe9a4 | 2015-04-24 14:18:01 -0700 | [diff] [blame] | 181 | settingsPublisher *pubsub.Publisher, |
| 182 | settingsName string, |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 183 | client rpc.Client, |
Matt Rosencrantz | 98d6d7c | 2015-09-04 12:34:08 -0700 | [diff] [blame^] | 184 | opts ...rpc.ServerOpt) (rpc.DeprecatedServer, error) { |
Matt Rosencrantz | 1094d06 | 2015-01-30 06:43:12 -0800 | [diff] [blame] | 185 | ctx, cancel := context.WithRootCancel(ctx) |
Todd Wang | ad49204 | 2015-04-17 15:58:40 -0700 | [diff] [blame] | 186 | ctx, _ = vtrace.WithNewSpan(ctx, "NewServer") |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 187 | statsPrefix := naming.Join("rpc", "server", "routing-id", streamMgr.RoutingID().String()) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 188 | s := &server{ |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 189 | ctx: ctx, |
| 190 | cancel: cancel, |
| 191 | streamMgr: streamMgr, |
| 192 | publisher: publisher.New(ctx, ns, publishPeriod), |
| 193 | listenState: make(map[*listenState]struct{}), |
| 194 | listeners: make(map[stream.Listener]struct{}), |
| 195 | proxies: make(map[string]proxyState), |
| 196 | stoppedChan: make(chan struct{}), |
| 197 | |
Cosmos Nicolaou | 00fe9a4 | 2015-04-24 14:18:01 -0700 | [diff] [blame] | 198 | ns: ns, |
| 199 | stats: newRPCStats(statsPrefix), |
| 200 | settingsPublisher: settingsPublisher, |
| 201 | settingsName: settingsName, |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 202 | } |
Suharsh Sivakumar | 2c5d810 | 2015-03-23 08:49:12 -0700 | [diff] [blame] | 203 | var ( |
| 204 | dischargeExpiryBuffer = vc.DefaultServerDischargeExpiryBuffer |
| 205 | securityLevel options.SecurityLevel |
| 206 | ) |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 207 | ipNets, err := ipNetworks() |
| 208 | if err != nil { |
| 209 | return nil, err |
| 210 | } |
| 211 | s.ipNets = ipNets |
| 212 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 213 | for _, opt := range opts { |
Bogdan Caprita | 187269b | 2014-05-13 19:59:46 -0700 | [diff] [blame] | 214 | switch opt := opt.(type) { |
| 215 | case stream.ListenerOpt: |
| 216 | // Collect all ServerOpts that are also ListenerOpts. |
| 217 | s.listenerOpts = append(s.listenerOpts, opt) |
Bogdan Caprita | e737631 | 2014-11-10 13:13:17 -0800 | [diff] [blame] | 218 | switch opt := opt.(type) { |
Suharsh Sivakumar | 0891858 | 2015-03-03 15:16:36 -0800 | [diff] [blame] | 219 | case vc.DischargeExpiryBuffer: |
| 220 | dischargeExpiryBuffer = time.Duration(opt) |
Bogdan Caprita | e737631 | 2014-11-10 13:13:17 -0800 | [diff] [blame] | 221 | } |
Suharsh Sivakumar | e5e5dcc | 2015-03-18 14:29:31 -0700 | [diff] [blame] | 222 | case options.ServerBlessings: |
| 223 | s.blessings = opt.Blessings |
Asim Shankar | cc04421 | 2014-10-15 23:25:26 -0700 | [diff] [blame] | 224 | case options.ServesMountTable: |
Cosmos Nicolaou | e6e87f1 | 2014-06-03 14:29:10 -0700 | [diff] [blame] | 225 | s.servesMountTable = bool(opt) |
Ali Ghassemi | bac3403 | 2015-04-30 18:27:57 -0700 | [diff] [blame] | 226 | case options.IsLeaf: |
| 227 | s.isLeaf = bool(opt) |
Suharsh Sivakumar | d7a6519 | 2015-01-27 22:57:15 -0800 | [diff] [blame] | 228 | case ReservedNameDispatcher: |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 229 | s.dispReserved = opt.Dispatcher |
Nicolas LaCasse | 55a10f3 | 2014-11-26 13:25:53 -0800 | [diff] [blame] | 230 | case PreferredServerResolveProtocols: |
| 231 | s.preferredProtocols = []string(opt) |
Suharsh Sivakumar | 2c5d810 | 2015-03-23 08:49:12 -0700 | [diff] [blame] | 232 | case options.SecurityLevel: |
| 233 | securityLevel = opt |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 234 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 235 | } |
| 236 | } |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 237 | |
| 238 | authenticateVC := true |
| 239 | |
Suharsh Sivakumar | 2c5d810 | 2015-03-23 08:49:12 -0700 | [diff] [blame] | 240 | if securityLevel == options.SecurityNone { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 241 | authenticateVC = false |
Suharsh Sivakumar | 2c5d810 | 2015-03-23 08:49:12 -0700 | [diff] [blame] | 242 | s.blessings = security.Blessings{} |
| 243 | s.dispReserved = nil |
| 244 | } |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 245 | if authenticateVC { |
| 246 | s.principal = v23.GetPrincipal(ctx) |
| 247 | if s.blessings.IsZero() && s.principal != nil { |
| 248 | s.blessings = s.principal.BlessingStore().Default() |
| 249 | } |
| 250 | } |
| 251 | |
Suharsh Sivakumar | 0891858 | 2015-03-03 15:16:36 -0800 | [diff] [blame] | 252 | // Make dischargeExpiryBuffer shorter than the VC discharge buffer to ensure we have fetched |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 253 | // the discharges by the time the VC asks for them. |
Asim Shankar | 99b18a7 | 2015-04-25 23:19:28 -0700 | [diff] [blame] | 254 | s.dc = InternalNewDischargeClient(ctx, client, dischargeExpiryBuffer-(5*time.Second)) |
| 255 | s.listenerOpts = append(s.listenerOpts, s.dc) |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 256 | s.listenerOpts = append(s.listenerOpts, stream.AuthenticatedVC(authenticateVC)) |
Bogdan Caprita | e737631 | 2014-11-10 13:13:17 -0800 | [diff] [blame] | 257 | blessingsStatsName := naming.Join(statsPrefix, "security", "blessings") |
Asim Shankar | 2bf7b1e | 2015-02-27 00:45:12 -0800 | [diff] [blame] | 258 | // TODO(caprita): revist printing the blessings with %s, and |
| 259 | // instead expose them as a list. |
Suharsh Sivakumar | e5e5dcc | 2015-03-18 14:29:31 -0700 | [diff] [blame] | 260 | stats.NewString(blessingsStatsName).Set(fmt.Sprintf("%s", s.blessings)) |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 261 | if s.principal != nil { |
Bogdan Caprita | e737631 | 2014-11-10 13:13:17 -0800 | [diff] [blame] | 262 | stats.NewStringFunc(blessingsStatsName, func() string { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 263 | return fmt.Sprintf("%s (default)", s.principal.BlessingStore().Default()) |
Bogdan Caprita | e737631 | 2014-11-10 13:13:17 -0800 | [diff] [blame] | 264 | }) |
| 265 | } |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 266 | return s, nil |
| 267 | } |
| 268 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 269 | func (s *server) Status() rpc.ServerStatus { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 270 | defer apilog.LogCall(nil)(nil) // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 271 | status := rpc.ServerStatus{} |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 272 | s.Lock() |
| 273 | defer s.Unlock() |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 274 | status.State = externalStates[s.state] |
| 275 | status.ServesMountTable = s.servesMountTable |
| 276 | status.Mounts = s.publisher.Status() |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 277 | status.Endpoints = []naming.Endpoint{} |
| 278 | for ls, _ := range s.listenState { |
| 279 | if ls.eperr != nil { |
| 280 | status.Errors = append(status.Errors, ls.eperr) |
| 281 | } |
| 282 | if ls.lnerr != nil { |
| 283 | status.Errors = append(status.Errors, ls.lnerr) |
| 284 | } |
| 285 | for _, iep := range ls.ieps { |
| 286 | status.Endpoints = append(status.Endpoints, iep) |
| 287 | } |
| 288 | } |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 289 | status.Proxies = make([]rpc.ProxyStatus, 0, len(s.proxies)) |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 290 | for k, v := range s.proxies { |
Jiri Simsa | d9a7b3c | 2015-08-12 16:38:27 -0700 | [diff] [blame] | 291 | status.Proxies = append(status.Proxies, rpc.ProxyStatus{ |
| 292 | Proxy: k, |
| 293 | Endpoint: v.endpoint, |
| 294 | Error: v.err, |
| 295 | }) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 296 | } |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 297 | return status |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 298 | } |
| 299 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 300 | func (s *server) WatchNetwork(ch chan<- rpc.NetworkChange) { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 301 | defer apilog.LogCallf(nil, "ch=")(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 302 | s.Lock() |
| 303 | defer s.Unlock() |
| 304 | if s.dhcpState != nil { |
| 305 | s.dhcpState.watchers[ch] = struct{}{} |
| 306 | } |
| 307 | } |
| 308 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 309 | func (s *server) UnwatchNetwork(ch chan<- rpc.NetworkChange) { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 310 | defer apilog.LogCallf(nil, "ch=")(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 311 | s.Lock() |
| 312 | defer s.Unlock() |
| 313 | if s.dhcpState != nil { |
| 314 | delete(s.dhcpState.watchers, ch) |
| 315 | } |
| 316 | } |
| 317 | |
Robin Thellend | 92b65a4 | 2014-12-17 14:30:16 -0800 | [diff] [blame] | 318 | // resolveToEndpoint resolves an object name or address to an endpoint. |
| 319 | func (s *server) resolveToEndpoint(address string) (string, error) { |
Asim Shankar | aae3180 | 2015-01-22 11:59:42 -0800 | [diff] [blame] | 320 | var resolved *naming.MountEntry |
| 321 | var err error |
Asim Shankar | dee311d | 2014-08-01 17:41:31 -0700 | [diff] [blame] | 322 | if s.ns != nil { |
Asim Shankar | aae3180 | 2015-01-22 11:59:42 -0800 | [diff] [blame] | 323 | if resolved, err = s.ns.Resolve(s.ctx, address); err != nil { |
Asim Shankar | dee311d | 2014-08-01 17:41:31 -0700 | [diff] [blame] | 324 | return "", err |
| 325 | } |
| 326 | } else { |
Asim Shankar | aae3180 | 2015-01-22 11:59:42 -0800 | [diff] [blame] | 327 | // Fake a namespace resolution |
| 328 | resolved = &naming.MountEntry{Servers: []naming.MountedServer{ |
| 329 | {Server: address}, |
| 330 | }} |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 331 | } |
Nicolas LaCasse | 55a10f3 | 2014-11-26 13:25:53 -0800 | [diff] [blame] | 332 | // An empty set of protocols means all protocols... |
Jungho Ahn | 25545d3 | 2015-01-26 15:14:14 -0800 | [diff] [blame] | 333 | if resolved.Servers, err = filterAndOrderServers(resolved.Servers, s.preferredProtocols, s.ipNets); err != nil { |
Nicolas LaCasse | 55a10f3 | 2014-11-26 13:25:53 -0800 | [diff] [blame] | 334 | return "", err |
| 335 | } |
Asim Shankar | aae3180 | 2015-01-22 11:59:42 -0800 | [diff] [blame] | 336 | for _, n := range resolved.Names() { |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 337 | address, suffix := naming.SplitAddressName(n) |
David Why Use Two When One Will Do Presotto | adf0ca1 | 2014-11-13 10:49:01 -0800 | [diff] [blame] | 338 | if suffix != "" { |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 339 | continue |
| 340 | } |
Asim Shankar | aae3180 | 2015-01-22 11:59:42 -0800 | [diff] [blame] | 341 | if ep, err := inaming.NewEndpoint(address); err == nil { |
| 342 | return ep.String(), nil |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 343 | } |
| 344 | } |
Cosmos Nicolaou | 185c0c6 | 2015-04-13 21:22:43 -0700 | [diff] [blame] | 345 | return "", verror.New(errFailedToResolveToEndpoint, s.ctx, address) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 346 | } |
| 347 | |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 348 | // createEndpoints creates appropriate inaming.Endpoint instances for |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 349 | // all of the externally accessible network addresses that can be used |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 350 | // to reach this server. |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 351 | func (s *server) createEndpoints(lep naming.Endpoint, chooser netstate.AddressChooser) ([]*inaming.Endpoint, string, bool, error) { |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 352 | iep, ok := lep.(*inaming.Endpoint) |
| 353 | if !ok { |
Cosmos Nicolaou | 185c0c6 | 2015-04-13 21:22:43 -0700 | [diff] [blame] | 354 | return nil, "", false, verror.New(errInternalTypeConversion, nil, fmt.Sprintf("%T", lep)) |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 355 | } |
| 356 | if !strings.HasPrefix(iep.Protocol, "tcp") && |
| 357 | !strings.HasPrefix(iep.Protocol, "ws") { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 358 | // If not tcp, ws, or wsh, just return the endpoint we were given. |
| 359 | return []*inaming.Endpoint{iep}, "", false, nil |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 360 | } |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 361 | host, port, err := net.SplitHostPort(iep.Address) |
| 362 | if err != nil { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 363 | return nil, "", false, err |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 364 | } |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 365 | addrs, unspecified, err := netstate.PossibleAddresses(iep.Protocol, host, chooser) |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 366 | if err != nil { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 367 | return nil, port, false, err |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 368 | } |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 369 | |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 370 | ieps := make([]*inaming.Endpoint, 0, len(addrs)) |
| 371 | for _, addr := range addrs { |
| 372 | n, err := inaming.NewEndpoint(lep.String()) |
| 373 | if err != nil { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 374 | return nil, port, false, err |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 375 | } |
| 376 | n.IsMountTable = s.servesMountTable |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 377 | n.Address = net.JoinHostPort(addr.String(), port) |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 378 | ieps = append(ieps, n) |
| 379 | } |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 380 | return ieps, port, unspecified, nil |
Cosmos Nicolaou | 28dabfc | 2014-12-15 22:51:07 -0800 | [diff] [blame] | 381 | } |
| 382 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 383 | func (s *server) Listen(listenSpec rpc.ListenSpec) ([]naming.Endpoint, error) { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 384 | defer apilog.LogCallf(nil, "listenSpec=")(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 385 | useProxy := len(listenSpec.Proxy) > 0 |
| 386 | if !useProxy && len(listenSpec.Addrs) == 0 { |
Jiri Simsa | 074bf36 | 2015-02-17 09:29:45 -0800 | [diff] [blame] | 387 | return nil, verror.New(verror.ErrBadArg, s.ctx, "ListenSpec contains no proxy or addresses to listen on") |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 388 | } |
| 389 | |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 390 | s.Lock() |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 391 | defer s.Unlock() |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 392 | |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 393 | if err := s.allowed(listening, "Listen"); err != nil { |
| 394 | return nil, err |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 395 | } |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 396 | |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 397 | // Start the proxy as early as possible, ignore duplicate requests |
| 398 | // for the same proxy. |
| 399 | if _, inuse := s.proxies[listenSpec.Proxy]; useProxy && !inuse { |
Asim Shankar | 6b1c8b0 | 2015-04-26 22:09:45 -0700 | [diff] [blame] | 400 | // Pre-emptively fetch discharges on the blessings (they will be cached |
| 401 | // within s.dc for future calls). |
| 402 | // This shouldn't be required, but is a hack to reduce flakiness in |
| 403 | // JavaScript browser integration tests. |
| 404 | // See https://v.io/i/392 |
| 405 | s.dc.PrepareDischarges(s.ctx, s.blessings.ThirdPartyCaveats(), security.DischargeImpetus{}) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 406 | // We have a goroutine for listening on proxy connections. |
Cosmos Nicolaou | eef1fab | 2014-11-11 18:23:41 -0800 | [diff] [blame] | 407 | s.active.Add(1) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 408 | go func() { |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 409 | s.proxyListenLoop(listenSpec.Proxy) |
| 410 | s.active.Done() |
| 411 | }() |
| 412 | } |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 413 | |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 414 | roaming := false |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 415 | lnState := make([]*listenState, 0, len(listenSpec.Addrs)) |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 416 | for _, addr := range listenSpec.Addrs { |
| 417 | if len(addr.Address) > 0 { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 418 | // Listen if we have a local address to listen on. |
| 419 | ls := &listenState{ |
| 420 | protocol: addr.Protocol, |
| 421 | address: addr.Address, |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 422 | } |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 423 | ls.ln, ls.lep, ls.lnerr = s.streamMgr.Listen(s.ctx, addr.Protocol, addr.Address, s.blessings, s.listenerOpts...) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 424 | lnState = append(lnState, ls) |
| 425 | if ls.lnerr != nil { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 426 | s.ctx.VI(2).Infof("Listen(%q, %q, ...) failed: %v", addr.Protocol, addr.Address, ls.lnerr) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 427 | continue |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 428 | } |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 429 | ls.ieps, ls.port, ls.roaming, ls.eperr = s.createEndpoints(ls.lep, listenSpec.AddressChooser) |
| 430 | if ls.roaming && ls.eperr == nil { |
| 431 | ls.protoIEP = *ls.lep.(*inaming.Endpoint) |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 432 | roaming = true |
| 433 | } |
| 434 | } |
| 435 | } |
| 436 | |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 437 | found := false |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 438 | var lastErr error |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 439 | for _, ls := range lnState { |
| 440 | if ls.ln != nil { |
| 441 | found = true |
| 442 | break |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 443 | } |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 444 | if ls.lnerr != nil { |
| 445 | lastErr = ls.lnerr |
| 446 | } |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 447 | } |
| 448 | if !found && !useProxy { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 449 | return nil, verror.New(verror.ErrBadArg, s.ctx, verror.New(errNoListeners, s.ctx, lastErr)) |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 450 | } |
| 451 | |
Cosmos Nicolaou | 00fe9a4 | 2015-04-24 14:18:01 -0700 | [diff] [blame] | 452 | if roaming && s.dhcpState == nil && s.settingsPublisher != nil { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 453 | // Create a dhcp listener if we haven't already done so. |
| 454 | dhcp := &dhcpState{ |
Cosmos Nicolaou | 00fe9a4 | 2015-04-24 14:18:01 -0700 | [diff] [blame] | 455 | name: s.settingsName, |
| 456 | publisher: s.settingsPublisher, |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 457 | watchers: make(map[chan<- rpc.NetworkChange]struct{}), |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 458 | } |
| 459 | s.dhcpState = dhcp |
Cosmos Nicolaou | 11c0ca1 | 2015-04-23 16:23:43 -0700 | [diff] [blame] | 460 | dhcp.ch = make(chan pubsub.Setting, 10) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 461 | dhcp.stream, dhcp.err = dhcp.publisher.ForkStream(dhcp.name, dhcp.ch) |
| 462 | if dhcp.err == nil { |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 463 | // We have a goroutine to listen for dhcp changes. |
| 464 | s.active.Add(1) |
| 465 | go func() { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 466 | s.dhcpLoop(dhcp.ch) |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 467 | s.active.Done() |
| 468 | }() |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 469 | } |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 470 | } |
| 471 | |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 472 | eps := make([]naming.Endpoint, 0, 10) |
| 473 | for _, ls := range lnState { |
| 474 | s.listenState[ls] = struct{}{} |
| 475 | if ls.ln != nil { |
| 476 | // We have a goroutine per listener to accept new flows. |
| 477 | // Each flow is served from its own goroutine. |
| 478 | s.active.Add(1) |
| 479 | go func(ln stream.Listener, ep naming.Endpoint) { |
| 480 | s.listenLoop(ln, ep) |
| 481 | s.active.Done() |
| 482 | }(ls.ln, ls.lep) |
| 483 | } |
| 484 | |
| 485 | for _, iep := range ls.ieps { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 486 | eps = append(eps, iep) |
| 487 | } |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 488 | } |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 489 | |
Cosmos Nicolaou | 28dabfc | 2014-12-15 22:51:07 -0800 | [diff] [blame] | 490 | return eps, nil |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 491 | } |
| 492 | |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 493 | func (s *server) reconnectAndPublishProxy(proxy string) (*inaming.Endpoint, stream.Listener, error) { |
Robin Thellend | 92b65a4 | 2014-12-17 14:30:16 -0800 | [diff] [blame] | 494 | resolved, err := s.resolveToEndpoint(proxy) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 495 | if err != nil { |
Cosmos Nicolaou | 185c0c6 | 2015-04-13 21:22:43 -0700 | [diff] [blame] | 496 | return nil, nil, verror.New(errFailedToResolveProxy, s.ctx, proxy, err) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 497 | } |
Asim Shankar | 99b18a7 | 2015-04-25 23:19:28 -0700 | [diff] [blame] | 498 | opts := append([]stream.ListenerOpt{proxyAuth{s}}, s.listenerOpts...) |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 499 | ln, ep, err := s.streamMgr.Listen(s.ctx, inaming.Network, resolved, s.blessings, opts...) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 500 | if err != nil { |
Cosmos Nicolaou | 185c0c6 | 2015-04-13 21:22:43 -0700 | [diff] [blame] | 501 | return nil, nil, verror.New(errFailedToListenForProxy, s.ctx, resolved, err) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 502 | } |
| 503 | iep, ok := ep.(*inaming.Endpoint) |
| 504 | if !ok { |
| 505 | ln.Close() |
Cosmos Nicolaou | 185c0c6 | 2015-04-13 21:22:43 -0700 | [diff] [blame] | 506 | return nil, nil, verror.New(errInternalTypeConversion, s.ctx, fmt.Sprintf("%T", ep)) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 507 | } |
| 508 | s.Lock() |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 509 | s.proxies[proxy] = proxyState{iep, nil} |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 510 | s.Unlock() |
Robin Thellend | e22920e | 2015-02-05 17:15:50 -0800 | [diff] [blame] | 511 | iep.IsMountTable = s.servesMountTable |
Robin Thellend | b457df9 | 2015-03-30 09:42:15 -0700 | [diff] [blame] | 512 | iep.IsLeaf = s.isLeaf |
Robin Thellend | 89e9523 | 2015-03-24 13:48:48 -0700 | [diff] [blame] | 513 | s.publisher.AddServer(iep.String()) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 514 | return iep, ln, nil |
| 515 | } |
| 516 | |
| 517 | func (s *server) proxyListenLoop(proxy string) { |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 518 | const ( |
| 519 | min = 5 * time.Millisecond |
| 520 | max = 5 * time.Minute |
| 521 | ) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 522 | |
| 523 | iep, ln, err := s.reconnectAndPublishProxy(proxy) |
| 524 | if err != nil { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 525 | s.ctx.Errorf("Failed to connect to proxy: %s", err) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 526 | } |
| 527 | // the initial connection maybe have failed, but we enter the retry |
| 528 | // loop anyway so that we will continue to try and connect to the |
| 529 | // proxy. |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 530 | s.Lock() |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 531 | if s.isStopState() { |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 532 | s.Unlock() |
| 533 | return |
| 534 | } |
| 535 | s.Unlock() |
| 536 | |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 537 | for { |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 538 | if ln != nil && iep != nil { |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 539 | err := s.listenLoop(ln, iep) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 540 | // The listener is done, so: |
| 541 | // (1) Unpublish its name |
Cosmos Nicolaou | 8bd8e10 | 2015-01-13 21:52:53 -0800 | [diff] [blame] | 542 | s.publisher.RemoveServer(iep.String()) |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 543 | s.Lock() |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 544 | if err != nil { |
Jiri Simsa | 074bf36 | 2015-02-17 09:29:45 -0800 | [diff] [blame] | 545 | s.proxies[proxy] = proxyState{iep, verror.New(verror.ErrNoServers, s.ctx, err)} |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 546 | } else { |
Asim Shankar | 7171a25 | 2015-03-07 14:41:40 -0800 | [diff] [blame] | 547 | // err will be nil if we're stopping. |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 548 | s.proxies[proxy] = proxyState{iep, nil} |
| 549 | s.Unlock() |
| 550 | return |
| 551 | } |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 552 | s.Unlock() |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 553 | } |
| 554 | |
| 555 | s.Lock() |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 556 | if s.isStopState() { |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 557 | s.Unlock() |
| 558 | return |
| 559 | } |
| 560 | s.Unlock() |
| 561 | |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 562 | // (2) Reconnect to the proxy unless the server has been stopped |
| 563 | backoff := min |
| 564 | ln = nil |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 565 | for { |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 566 | select { |
| 567 | case <-time.After(backoff): |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 568 | if backoff = backoff * 2; backoff > max { |
| 569 | backoff = max |
| 570 | } |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 571 | case <-s.stoppedChan: |
| 572 | return |
| 573 | } |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 574 | // (3) reconnect, publish new address |
| 575 | if iep, ln, err = s.reconnectAndPublishProxy(proxy); err != nil { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 576 | s.ctx.Errorf("Failed to reconnect to proxy %q: %s", proxy, err) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 577 | } else { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 578 | s.ctx.VI(1).Infof("Reconnected to proxy %q, %s", proxy, iep) |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 579 | break |
| 580 | } |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 581 | } |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 582 | } |
| 583 | } |
| 584 | |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 585 | // addListener adds the supplied listener taking care to |
| 586 | // check to see if we're already stopping. It returns true |
| 587 | // if the listener was added. |
| 588 | func (s *server) addListener(ln stream.Listener) bool { |
| 589 | s.Lock() |
| 590 | defer s.Unlock() |
| 591 | if s.isStopState() { |
| 592 | return false |
| 593 | } |
| 594 | s.listeners[ln] = struct{}{} |
| 595 | return true |
| 596 | } |
| 597 | |
| 598 | // rmListener removes the supplied listener taking care to |
| 599 | // check if we're already stopping. It returns true if the |
| 600 | // listener was removed. |
| 601 | func (s *server) rmListener(ln stream.Listener) bool { |
| 602 | s.Lock() |
| 603 | defer s.Unlock() |
| 604 | if s.isStopState() { |
| 605 | return false |
| 606 | } |
| 607 | delete(s.listeners, ln) |
| 608 | return true |
| 609 | } |
| 610 | |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 611 | func (s *server) listenLoop(ln stream.Listener, ep naming.Endpoint) error { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 612 | defer s.ctx.VI(1).Infof("rpc: Stopped listening on %s", ep) |
Cosmos Nicolaou | eef1fab | 2014-11-11 18:23:41 -0800 | [diff] [blame] | 613 | var calls sync.WaitGroup |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 614 | |
| 615 | if !s.addListener(ln) { |
| 616 | // We're stopping. |
| 617 | return nil |
| 618 | } |
| 619 | |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 620 | defer func() { |
Cosmos Nicolaou | eef1fab | 2014-11-11 18:23:41 -0800 | [diff] [blame] | 621 | calls.Wait() |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 622 | s.rmListener(ln) |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 623 | }() |
| 624 | for { |
| 625 | flow, err := ln.Accept() |
| 626 | if err != nil { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 627 | s.ctx.VI(10).Infof("rpc: Accept on %v failed: %v", ep, err) |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 628 | return err |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 629 | } |
Cosmos Nicolaou | eef1fab | 2014-11-11 18:23:41 -0800 | [diff] [blame] | 630 | calls.Add(1) |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 631 | go func(flow stream.Flow) { |
Todd Wang | 34ed4c6 | 2014-11-26 15:15:52 -0800 | [diff] [blame] | 632 | defer calls.Done() |
| 633 | fs, err := newFlowServer(flow, s) |
| 634 | if err != nil { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 635 | s.ctx.VI(1).Infof("newFlowServer on %v failed: %v", ep, err) |
Todd Wang | 34ed4c6 | 2014-11-26 15:15:52 -0800 | [diff] [blame] | 636 | return |
| 637 | } |
| 638 | if err := fs.serve(); err != nil { |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 639 | // TODO(caprita): Logging errors here is too spammy. For example, "not |
| 640 | // authorized" errors shouldn't be logged as server errors. |
Cosmos Nicolaou | 93dd88b | 2015-02-19 15:10:53 -0800 | [diff] [blame] | 641 | // TODO(cnicolaou): revisit this when verror2 transition is |
| 642 | // done. |
Cosmos Nicolaou | 1534b3f | 2014-12-10 15:30:00 -0800 | [diff] [blame] | 643 | if err != io.EOF { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 644 | s.ctx.VI(2).Infof("Flow.serve on %v failed: %v", ep, err) |
Cosmos Nicolaou | 1534b3f | 2014-12-10 15:30:00 -0800 | [diff] [blame] | 645 | } |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 646 | } |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 647 | }(flow) |
| 648 | } |
| 649 | } |
| 650 | |
Cosmos Nicolaou | 11c0ca1 | 2015-04-23 16:23:43 -0700 | [diff] [blame] | 651 | func (s *server) dhcpLoop(ch chan pubsub.Setting) { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 652 | defer s.ctx.VI(1).Infof("rpc: Stopped listen for dhcp changes") |
| 653 | s.ctx.VI(2).Infof("rpc: dhcp loop") |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 654 | for setting := range ch { |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 655 | if setting == nil { |
| 656 | return |
| 657 | } |
| 658 | switch v := setting.Value().(type) { |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 659 | case []net.Addr: |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 660 | s.Lock() |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 661 | if s.isStopState() { |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 662 | s.Unlock() |
| 663 | return |
| 664 | } |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 665 | change := rpc.NetworkChange{ |
Cosmos Nicolaou | 00fe9a4 | 2015-04-24 14:18:01 -0700 | [diff] [blame] | 666 | Time: time.Now(), |
| 667 | State: externalStates[s.state], |
| 668 | } |
| 669 | switch setting.Name() { |
| 670 | case NewAddrsSetting: |
| 671 | change.Changed = s.addAddresses(v) |
| 672 | change.AddedAddrs = v |
| 673 | case RmAddrsSetting: |
| 674 | change.Changed, change.Error = s.removeAddresses(v) |
| 675 | change.RemovedAddrs = v |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 676 | } |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 677 | s.ctx.VI(2).Infof("rpc: dhcp: change %v", change) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 678 | for ch, _ := range s.dhcpState.watchers { |
| 679 | select { |
| 680 | case ch <- change: |
| 681 | default: |
| 682 | } |
| 683 | } |
| 684 | s.Unlock() |
| 685 | default: |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 686 | s.ctx.Errorf("rpc: dhcpLoop: unhandled setting type %T", v) |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 687 | } |
| 688 | } |
| 689 | } |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 690 | |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 691 | func getHost(address net.Addr) string { |
| 692 | host, _, err := net.SplitHostPort(address.String()) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 693 | if err == nil { |
| 694 | return host |
| 695 | } |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 696 | return address.String() |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 697 | |
| 698 | } |
| 699 | |
| 700 | // Remove all endpoints that have the same host address as the supplied |
| 701 | // address parameter. |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 702 | func (s *server) removeAddresses(addrs []net.Addr) ([]naming.Endpoint, error) { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 703 | var removed []naming.Endpoint |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 704 | for _, address := range addrs { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 705 | host := getHost(address) |
| 706 | for ls, _ := range s.listenState { |
| 707 | if ls != nil && ls.roaming && len(ls.ieps) > 0 { |
| 708 | remaining := make([]*inaming.Endpoint, 0, len(ls.ieps)) |
| 709 | for _, iep := range ls.ieps { |
| 710 | lnHost, _, err := net.SplitHostPort(iep.Address) |
| 711 | if err != nil { |
| 712 | lnHost = iep.Address |
| 713 | } |
| 714 | if lnHost == host { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 715 | s.ctx.VI(2).Infof("rpc: dhcp removing: %s", iep) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 716 | removed = append(removed, iep) |
| 717 | s.publisher.RemoveServer(iep.String()) |
| 718 | continue |
| 719 | } |
| 720 | remaining = append(remaining, iep) |
| 721 | } |
| 722 | ls.ieps = remaining |
| 723 | } |
| 724 | } |
| 725 | } |
| 726 | return removed, nil |
| 727 | } |
| 728 | |
| 729 | // Add new endpoints for the new address. There is no way to know with |
| 730 | // 100% confidence which new endpoints to publish without shutting down |
| 731 | // all network connections and reinitializing everything from scratch. |
| 732 | // Instead, we find all roaming listeners with at least one endpoint |
| 733 | // and create a new endpoint with the same port as the existing ones |
| 734 | // but with the new address supplied to us to by the dhcp code. As |
| 735 | // an additional safeguard we reject the new address if it is not |
| 736 | // externally accessible. |
| 737 | // This places the onus on the dhcp/roaming code that sends us addresses |
| 738 | // to ensure that those addresses are externally reachable. |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 739 | func (s *server) addAddresses(addrs []net.Addr) []naming.Endpoint { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 740 | var added []naming.Endpoint |
Cosmos Nicolaou | aa87e29 | 2015-04-21 22:15:50 -0700 | [diff] [blame] | 741 | for _, address := range netstate.ConvertToAddresses(addrs) { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 742 | if !netstate.IsAccessibleIP(address) { |
| 743 | return added |
| 744 | } |
| 745 | host := getHost(address) |
| 746 | for ls, _ := range s.listenState { |
| 747 | if ls != nil && ls.roaming { |
| 748 | niep := ls.protoIEP |
| 749 | niep.Address = net.JoinHostPort(host, ls.port) |
Robin Thellend | b457df9 | 2015-03-30 09:42:15 -0700 | [diff] [blame] | 750 | niep.IsMountTable = s.servesMountTable |
| 751 | niep.IsLeaf = s.isLeaf |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 752 | ls.ieps = append(ls.ieps, &niep) |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 753 | s.ctx.VI(2).Infof("rpc: dhcp adding: %s", niep) |
Robin Thellend | 89e9523 | 2015-03-24 13:48:48 -0700 | [diff] [blame] | 754 | s.publisher.AddServer(niep.String()) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 755 | added = append(added, &niep) |
| 756 | } |
| 757 | } |
| 758 | } |
| 759 | return added |
| 760 | } |
Cosmos Nicolaou | ef323db | 2014-09-07 22:13:28 -0700 | [diff] [blame] | 761 | |
Bogdan Caprita | 7590a6d | 2015-01-08 13:43:40 -0800 | [diff] [blame] | 762 | type leafDispatcher struct { |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 763 | invoker rpc.Invoker |
Bogdan Caprita | 7590a6d | 2015-01-08 13:43:40 -0800 | [diff] [blame] | 764 | auth security.Authorizer |
| 765 | } |
| 766 | |
Cosmos Nicolaou | 5a3125a | 2015-07-10 11:19:20 -0700 | [diff] [blame] | 767 | func (d leafDispatcher) Lookup(ctx *context.T, suffix string) (interface{}, security.Authorizer, error) { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 768 | defer apilog.LogCallf(nil, "suffix=%.10s...", suffix)(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Bogdan Caprita | 7590a6d | 2015-01-08 13:43:40 -0800 | [diff] [blame] | 769 | if suffix != "" { |
Cosmos Nicolaou | 185c0c6 | 2015-04-13 21:22:43 -0700 | [diff] [blame] | 770 | return nil, nil, verror.New(verror.ErrUnknownSuffix, nil, suffix) |
Bogdan Caprita | 7590a6d | 2015-01-08 13:43:40 -0800 | [diff] [blame] | 771 | } |
| 772 | return d.invoker, d.auth, nil |
| 773 | } |
| 774 | |
Cosmos Nicolaou | 92dba58 | 2014-11-05 17:24:10 -0800 | [diff] [blame] | 775 | func (s *server) Serve(name string, obj interface{}, authorizer security.Authorizer) error { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 776 | defer apilog.LogCallf(nil, "name=%.10s...,obj=,authorizer=", name)(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 777 | if obj == nil { |
Jiri Simsa | 074bf36 | 2015-02-17 09:29:45 -0800 | [diff] [blame] | 778 | return verror.New(verror.ErrBadArg, s.ctx, "nil object") |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 779 | } |
Bogdan Caprita | 9592d9f | 2015-01-08 22:15:16 -0800 | [diff] [blame] | 780 | invoker, err := objectToInvoker(obj) |
| 781 | if err != nil { |
Jiri Simsa | 074bf36 | 2015-02-17 09:29:45 -0800 | [diff] [blame] | 782 | return verror.New(verror.ErrBadArg, s.ctx, fmt.Sprintf("bad object: %v", err)) |
Cosmos Nicolaou | 61c96c7 | 2014-11-03 11:57:56 -0800 | [diff] [blame] | 783 | } |
Robin Thellend | b457df9 | 2015-03-30 09:42:15 -0700 | [diff] [blame] | 784 | s.setLeaf(true) |
Bogdan Caprita | 9592d9f | 2015-01-08 22:15:16 -0800 | [diff] [blame] | 785 | return s.ServeDispatcher(name, &leafDispatcher{invoker, authorizer}) |
Cosmos Nicolaou | 61c96c7 | 2014-11-03 11:57:56 -0800 | [diff] [blame] | 786 | } |
| 787 | |
Robin Thellend | b457df9 | 2015-03-30 09:42:15 -0700 | [diff] [blame] | 788 | func (s *server) setLeaf(value bool) { |
| 789 | s.Lock() |
| 790 | defer s.Unlock() |
| 791 | s.isLeaf = value |
| 792 | for ls, _ := range s.listenState { |
| 793 | for i := range ls.ieps { |
| 794 | ls.ieps[i].IsLeaf = s.isLeaf |
| 795 | } |
| 796 | } |
| 797 | } |
| 798 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 799 | func (s *server) ServeDispatcher(name string, disp rpc.Dispatcher) error { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 800 | defer apilog.LogCallf(nil, "name=%.10s...,disp=", name)(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Cosmos Nicolaou | 92dba58 | 2014-11-05 17:24:10 -0800 | [diff] [blame] | 801 | if disp == nil { |
Jiri Simsa | 074bf36 | 2015-02-17 09:29:45 -0800 | [diff] [blame] | 802 | return verror.New(verror.ErrBadArg, s.ctx, "nil dispatcher") |
Cosmos Nicolaou | fdc838b | 2014-06-30 21:44:27 -0700 | [diff] [blame] | 803 | } |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 804 | s.Lock() |
| 805 | defer s.Unlock() |
| 806 | if err := s.allowed(serving, "Serve or ServeDispatcher"); err != nil { |
| 807 | return err |
Cosmos Nicolaou | fdc838b | 2014-06-30 21:44:27 -0700 | [diff] [blame] | 808 | } |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 809 | vtrace.GetSpan(s.ctx).Annotate("Serving under name: " + name) |
Cosmos Nicolaou | 92dba58 | 2014-11-05 17:24:10 -0800 | [diff] [blame] | 810 | s.disp = disp |
Cosmos Nicolaou | fdc838b | 2014-06-30 21:44:27 -0700 | [diff] [blame] | 811 | if len(name) > 0 { |
Robin Thellend | b457df9 | 2015-03-30 09:42:15 -0700 | [diff] [blame] | 812 | for ls, _ := range s.listenState { |
| 813 | for _, iep := range ls.ieps { |
| 814 | s.publisher.AddServer(iep.String()) |
| 815 | } |
| 816 | } |
Robin Thellend | 89e9523 | 2015-03-24 13:48:48 -0700 | [diff] [blame] | 817 | s.publisher.AddName(name, s.servesMountTable, s.isLeaf) |
Cosmos Nicolaou | fdc838b | 2014-06-30 21:44:27 -0700 | [diff] [blame] | 818 | } |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 819 | return nil |
| 820 | } |
| 821 | |
Cosmos Nicolaou | 92dba58 | 2014-11-05 17:24:10 -0800 | [diff] [blame] | 822 | func (s *server) AddName(name string) error { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 823 | defer apilog.LogCallf(nil, "name=%.10s...", name)(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Ali Ghassemi | 3c6db7b | 2014-11-10 17:20:26 -0800 | [diff] [blame] | 824 | if len(name) == 0 { |
Jiri Simsa | 074bf36 | 2015-02-17 09:29:45 -0800 | [diff] [blame] | 825 | return verror.New(verror.ErrBadArg, s.ctx, "name is empty") |
Ali Ghassemi | 3c6db7b | 2014-11-10 17:20:26 -0800 | [diff] [blame] | 826 | } |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 827 | s.Lock() |
| 828 | defer s.Unlock() |
| 829 | if err := s.allowed(publishing, "AddName"); err != nil { |
| 830 | return err |
Cosmos Nicolaou | 92dba58 | 2014-11-05 17:24:10 -0800 | [diff] [blame] | 831 | } |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 832 | vtrace.GetSpan(s.ctx).Annotate("Serving under name: " + name) |
Robin Thellend | 89e9523 | 2015-03-24 13:48:48 -0700 | [diff] [blame] | 833 | s.publisher.AddName(name, s.servesMountTable, s.isLeaf) |
Cosmos Nicolaou | 92dba58 | 2014-11-05 17:24:10 -0800 | [diff] [blame] | 834 | return nil |
| 835 | } |
| 836 | |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 837 | func (s *server) RemoveName(name string) { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 838 | defer apilog.LogCallf(nil, "name=%.10s...", name)(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Cosmos Nicolaou | 92dba58 | 2014-11-05 17:24:10 -0800 | [diff] [blame] | 839 | s.Lock() |
| 840 | defer s.Unlock() |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 841 | if err := s.allowed(publishing, "RemoveName"); err != nil { |
| 842 | return |
| 843 | } |
Matt Rosencrantz | 5f98d94 | 2015-01-08 13:48:30 -0800 | [diff] [blame] | 844 | vtrace.GetSpan(s.ctx).Annotate("Removed name: " + name) |
Cosmos Nicolaou | 92dba58 | 2014-11-05 17:24:10 -0800 | [diff] [blame] | 845 | s.publisher.RemoveName(name) |
Cosmos Nicolaou | 92dba58 | 2014-11-05 17:24:10 -0800 | [diff] [blame] | 846 | } |
| 847 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 848 | func (s *server) Stop() error { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 849 | defer apilog.LogCall(nil)(nil) // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Bogdan Caprita | 0ca0216 | 2015-04-30 11:21:44 -0700 | [diff] [blame] | 850 | serverDebug := fmt.Sprintf("Dispatcher: %T, Status:[%v]", s.disp, s.Status()) |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 851 | s.ctx.VI(1).Infof("Stop: %s", serverDebug) |
| 852 | defer s.ctx.VI(1).Infof("Stop done: %s", serverDebug) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 853 | s.Lock() |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 854 | if s.isStopState() { |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 855 | s.Unlock() |
| 856 | return nil |
| 857 | } |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 858 | s.state = stopping |
Asim Shankar | 0ea02ab | 2014-06-09 11:39:24 -0700 | [diff] [blame] | 859 | close(s.stoppedChan) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 860 | s.Unlock() |
| 861 | |
Robin Thellend | df42823 | 2014-10-06 12:50:44 -0700 | [diff] [blame] | 862 | // Delete the stats object. |
| 863 | s.stats.stop() |
| 864 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 865 | // Note, It's safe to Stop/WaitForStop on the publisher outside of the |
| 866 | // server lock, since publisher is safe for concurrent access. |
| 867 | |
| 868 | // Stop the publisher, which triggers unmounting of published names. |
| 869 | s.publisher.Stop() |
| 870 | // Wait for the publisher to be done unmounting before we can proceed to |
| 871 | // close the listeners (to minimize the number of mounted names pointing |
| 872 | // to endpoint that are no longer serving). |
| 873 | // |
| 874 | // TODO(caprita): See if make sense to fail fast on rejecting |
| 875 | // connections once listeners are closed, and parallelize the publisher |
| 876 | // and listener shutdown. |
| 877 | s.publisher.WaitForStop() |
| 878 | |
| 879 | s.Lock() |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 880 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 881 | // Close all listeners. No new flows will be accepted, while in-flight |
| 882 | // flows will continue until they terminate naturally. |
| 883 | nListeners := len(s.listeners) |
| 884 | errCh := make(chan error, nListeners) |
Cosmos Nicolaou | bc74314 | 2014-10-06 21:27:18 -0700 | [diff] [blame] | 885 | |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 886 | for ln, _ := range s.listeners { |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 887 | go func(ln stream.Listener) { |
| 888 | errCh <- ln.Close() |
| 889 | }(ln) |
Cosmos Nicolaou | ae8dd21 | 2014-12-13 23:43:08 -0800 | [diff] [blame] | 890 | } |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 891 | |
Cosmos Nicolaou | 11c0ca1 | 2015-04-23 16:23:43 -0700 | [diff] [blame] | 892 | drain := func(ch chan pubsub.Setting) { |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 893 | for { |
| 894 | select { |
| 895 | case v := <-ch: |
| 896 | if v == nil { |
| 897 | return |
| 898 | } |
| 899 | default: |
| 900 | close(ch) |
| 901 | return |
| 902 | } |
| 903 | } |
| 904 | } |
| 905 | |
| 906 | if dhcp := s.dhcpState; dhcp != nil { |
Cosmos Nicolaou | aceb8d9 | 2015-02-05 20:44:02 -0800 | [diff] [blame] | 907 | // TODO(cnicolaou,caprita): investigate not having to close and drain |
| 908 | // the channel here. It's a little awkward right now since we have to |
| 909 | // be careful to not close the channel in two places, i.e. here and |
| 910 | // and from the publisher's Shutdown method. |
| 911 | if err := dhcp.publisher.CloseFork(dhcp.name, dhcp.ch); err == nil { |
| 912 | drain(dhcp.ch) |
| 913 | } |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 914 | } |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 915 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 916 | s.Unlock() |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 917 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 918 | var firstErr error |
| 919 | for i := 0; i < nListeners; i++ { |
| 920 | if err := <-errCh; err != nil && firstErr == nil { |
| 921 | firstErr = err |
| 922 | } |
| 923 | } |
| 924 | // At this point, we are guaranteed that no new requests are going to be |
| 925 | // accepted. |
| 926 | |
| 927 | // Wait for the publisher and active listener + flows to finish. |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 928 | done := make(chan struct{}, 1) |
| 929 | go func() { s.active.Wait(); done <- struct{}{} }() |
| 930 | |
| 931 | select { |
| 932 | case <-done: |
Bogdan Caprita | 0ca0216 | 2015-04-30 11:21:44 -0700 | [diff] [blame] | 933 | case <-time.After(5 * time.Second): |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 934 | s.ctx.Errorf("%s: Listener Close Error: %v", serverDebug, firstErr) |
| 935 | s.ctx.Errorf("%s: Timedout waiting for goroutines to stop: listeners: %d (currently: %d)", serverDebug, nListeners, len(s.listeners)) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 936 | for ln, _ := range s.listeners { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 937 | s.ctx.Errorf("%s: Listener: %v", serverDebug, ln) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 938 | } |
| 939 | for ls, _ := range s.listenState { |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 940 | s.ctx.Errorf("%s: ListenState: %v", serverDebug, ls) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 941 | } |
| 942 | <-done |
Cosmos Nicolaou | e9c622d | 2015-07-10 11:09:42 -0700 | [diff] [blame] | 943 | s.ctx.Infof("%s: Done waiting.", serverDebug) |
Cosmos Nicolaou | 1b3594d | 2015-02-01 10:05:03 -0800 | [diff] [blame] | 944 | } |
Cosmos Nicolaou | 9388ae4 | 2014-11-10 10:57:15 -0800 | [diff] [blame] | 945 | |
Cosmos Nicolaou | fdc838b | 2014-06-30 21:44:27 -0700 | [diff] [blame] | 946 | s.Lock() |
Cosmos Nicolaou | 28dabfc | 2014-12-15 22:51:07 -0800 | [diff] [blame] | 947 | defer s.Unlock() |
Cosmos Nicolaou | fdc838b | 2014-06-30 21:44:27 -0700 | [diff] [blame] | 948 | s.disp = nil |
Cosmos Nicolaou | 28dabfc | 2014-12-15 22:51:07 -0800 | [diff] [blame] | 949 | if firstErr != nil { |
Jiri Simsa | 074bf36 | 2015-02-17 09:29:45 -0800 | [diff] [blame] | 950 | return verror.New(verror.ErrInternal, s.ctx, firstErr) |
Cosmos Nicolaou | 28dabfc | 2014-12-15 22:51:07 -0800 | [diff] [blame] | 951 | } |
Cosmos Nicolaou | 9fbe7d2 | 2015-01-25 22:13:13 -0800 | [diff] [blame] | 952 | s.state = stopped |
Matt Rosencrantz | 1094d06 | 2015-01-30 06:43:12 -0800 | [diff] [blame] | 953 | s.cancel() |
Cosmos Nicolaou | 28dabfc | 2014-12-15 22:51:07 -0800 | [diff] [blame] | 954 | return nil |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 955 | } |
| 956 | |
| 957 | // flowServer implements the RPC server-side protocol for a single RPC, over a |
| 958 | // flow that's already connected to the client. |
| 959 | type flowServer struct { |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 960 | ctx *context.T // context associated with the RPC |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 961 | server *server // rpc.Server that this flow server belongs to |
| 962 | disp rpc.Dispatcher // rpc.Dispatcher that will serve RPCs on this flow |
Todd Wang | 3425a90 | 2015-01-21 18:43:59 -0800 | [diff] [blame] | 963 | dec *vom.Decoder // to decode requests and args from the client |
| 964 | enc *vom.Encoder // to encode responses and results to the client |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 965 | flow stream.Flow // underlying flow |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 966 | |
Asim Shankar | 220a015 | 2014-10-30 21:21:09 -0700 | [diff] [blame] | 967 | // Fields filled in during the server invocation. |
Suharsh Sivakumar | 380bf34 | 2015-02-27 15:38:27 -0800 | [diff] [blame] | 968 | clientBlessings security.Blessings |
| 969 | ackBlessings bool |
| 970 | grantedBlessings security.Blessings |
| 971 | method, suffix string |
| 972 | tags []*vdl.Value |
| 973 | discharges map[string]security.Discharge |
| 974 | starttime time.Time |
| 975 | endStreamArgs bool // are the stream args at EOF? |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 976 | } |
| 977 | |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 978 | var ( |
| 979 | _ rpc.StreamServerCall = (*flowServer)(nil) |
| 980 | _ security.Call = (*flowServer)(nil) |
| 981 | ) |
Benjamin Prosnitz | fdfbf7b | 2014-10-08 09:47:21 -0700 | [diff] [blame] | 982 | |
Todd Wang | 34ed4c6 | 2014-11-26 15:15:52 -0800 | [diff] [blame] | 983 | func newFlowServer(flow stream.Flow, server *server) (*flowServer, error) { |
Cosmos Nicolaou | dcba93d | 2014-07-30 11:09:26 -0700 | [diff] [blame] | 984 | server.Lock() |
| 985 | disp := server.disp |
| 986 | server.Unlock() |
Matt Rosencrantz | 9fe6082 | 2014-09-12 10:09:53 -0700 | [diff] [blame] | 987 | |
Todd Wang | 34ed4c6 | 2014-11-26 15:15:52 -0800 | [diff] [blame] | 988 | fs := &flowServer{ |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 989 | ctx: server.ctx, |
Todd Wang | 34ed4c6 | 2014-11-26 15:15:52 -0800 | [diff] [blame] | 990 | server: server, |
| 991 | disp: disp, |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 992 | flow: flow, |
| 993 | discharges: make(map[string]security.Discharge), |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 994 | } |
Jungho Ahn | cfc8962 | 2015-04-24 11:27:23 -0700 | [diff] [blame] | 995 | typeenc := flow.VCDataCache().Get(vc.TypeEncoderKey{}) |
| 996 | if typeenc == nil { |
Jungho Ahn | 5d1fe97 | 2015-04-27 17:51:32 -0700 | [diff] [blame] | 997 | fs.enc = vom.NewEncoder(flow) |
| 998 | fs.dec = vom.NewDecoder(flow) |
Jungho Ahn | cfc8962 | 2015-04-24 11:27:23 -0700 | [diff] [blame] | 999 | } else { |
Jungho Ahn | 5d1fe97 | 2015-04-27 17:51:32 -0700 | [diff] [blame] | 1000 | fs.enc = vom.NewEncoderWithTypeEncoder(flow, typeenc.(*vom.TypeEncoder)) |
Jungho Ahn | cfc8962 | 2015-04-24 11:27:23 -0700 | [diff] [blame] | 1001 | typedec := flow.VCDataCache().Get(vc.TypeDecoderKey{}) |
Jungho Ahn | 5d1fe97 | 2015-04-27 17:51:32 -0700 | [diff] [blame] | 1002 | fs.dec = vom.NewDecoderWithTypeDecoder(flow, typedec.(*vom.TypeDecoder)) |
Todd Wang | 34ed4c6 | 2014-11-26 15:15:52 -0800 | [diff] [blame] | 1003 | } |
| 1004 | return fs, nil |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1005 | } |
| 1006 | |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 1007 | // authorizeVtrace works by simulating a call to __debug/vtrace.Trace. That |
| 1008 | // rpc is essentially equivalent in power to the data we are attempting to |
| 1009 | // attach here. |
Cosmos Nicolaou | 5a3125a | 2015-07-10 11:19:20 -0700 | [diff] [blame] | 1010 | func (fs *flowServer) authorizeVtrace(ctx *context.T) error { |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 1011 | // Set up a context as though we were calling __debug/vtrace. |
| 1012 | params := &security.CallParams{} |
Todd Wang | 4264e4b | 2015-04-16 22:43:40 -0700 | [diff] [blame] | 1013 | params.Copy(fs) |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 1014 | params.Method = "Trace" |
| 1015 | params.MethodTags = []*vdl.Value{vdl.ValueOf(access.Debug)} |
| 1016 | params.Suffix = "__debug/vtrace" |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 1017 | |
| 1018 | var auth security.Authorizer |
| 1019 | if fs.server.dispReserved != nil { |
Cosmos Nicolaou | 5a3125a | 2015-07-10 11:19:20 -0700 | [diff] [blame] | 1020 | _, auth, _ = fs.server.dispReserved.Lookup(ctx, params.Suffix) |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 1021 | } |
Todd Wang | 4264e4b | 2015-04-16 22:43:40 -0700 | [diff] [blame] | 1022 | return authorize(fs.ctx, security.NewCall(params), auth) |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 1023 | } |
| 1024 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1025 | func (fs *flowServer) serve() error { |
| 1026 | defer fs.flow.Close() |
Matt Rosencrantz | 8689793 | 2014-10-02 09:34:34 -0700 | [diff] [blame] | 1027 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1028 | results, err := fs.processRequest() |
Matt Rosencrantz | 9fe6082 | 2014-09-12 10:09:53 -0700 | [diff] [blame] | 1029 | |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1030 | vtrace.GetSpan(fs.ctx).Finish() |
Matt Rosencrantz | 1fa3277 | 2014-10-28 11:31:46 -0700 | [diff] [blame] | 1031 | |
Matt Rosencrantz | 9fe6082 | 2014-09-12 10:09:53 -0700 | [diff] [blame] | 1032 | var traceResponse vtrace.Response |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 1033 | // Check if the caller is permitted to view vtrace data. |
Cosmos Nicolaou | 5a3125a | 2015-07-10 11:19:20 -0700 | [diff] [blame] | 1034 | if fs.authorizeVtrace(fs.ctx) == nil { |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1035 | traceResponse = vtrace.GetResponse(fs.ctx) |
Matt Rosencrantz | 9fe6082 | 2014-09-12 10:09:53 -0700 | [diff] [blame] | 1036 | } |
| 1037 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1038 | // Respond to the client with the response header and positional results. |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1039 | response := rpc.Response{ |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1040 | Error: err, |
| 1041 | EndStreamResults: true, |
| 1042 | NumPosResults: uint64(len(results)), |
Matt Rosencrantz | 9fe6082 | 2014-09-12 10:09:53 -0700 | [diff] [blame] | 1043 | TraceResponse: traceResponse, |
Suharsh Sivakumar | 720b704 | 2014-12-22 17:33:23 -0800 | [diff] [blame] | 1044 | AckBlessings: fs.ackBlessings, |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1045 | } |
| 1046 | if err := fs.enc.Encode(response); err != nil { |
Cosmos Nicolaou | 1534b3f | 2014-12-10 15:30:00 -0800 | [diff] [blame] | 1047 | if err == io.EOF { |
| 1048 | return err |
| 1049 | } |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1050 | return verror.New(errResponseEncoding, fs.ctx, fs.LocalEndpoint().String(), fs.RemoteEndpoint().String(), err) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1051 | } |
| 1052 | if response.Error != nil { |
| 1053 | return response.Error |
| 1054 | } |
| 1055 | for ix, res := range results { |
Todd Wang | f519f8f | 2015-01-21 10:07:41 -0800 | [diff] [blame] | 1056 | if err := fs.enc.Encode(res); err != nil { |
Cosmos Nicolaou | 1534b3f | 2014-12-10 15:30:00 -0800 | [diff] [blame] | 1057 | if err == io.EOF { |
| 1058 | return err |
| 1059 | } |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1060 | return verror.New(errResultEncoding, fs.ctx, ix, fmt.Sprintf("%T=%v", res, res), err) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1061 | } |
| 1062 | } |
| 1063 | // TODO(ashankar): Should unread data from the flow be drained? |
| 1064 | // |
| 1065 | // Reason to do so: |
Suharsh Sivakumar | dcc11d7 | 2015-05-11 12:19:20 -0700 | [diff] [blame] | 1066 | // The common stream.Flow implementation (v.io/x/ref/runtime/internal/rpc/stream/vc/reader.go) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1067 | // uses iobuf.Slices backed by an iobuf.Pool. If the stream is not drained, these |
| 1068 | // slices will not be returned to the pool leading to possibly increased memory usage. |
| 1069 | // |
| 1070 | // Reason to not do so: |
| 1071 | // Draining here will conflict with any Reads on the flow in a separate goroutine |
| 1072 | // (for example, see TestStreamReadTerminatedByServer in full_test.go). |
| 1073 | // |
| 1074 | // For now, go with the reason to not do so as having unread data in the stream |
| 1075 | // should be a rare case. |
| 1076 | return nil |
| 1077 | } |
| 1078 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1079 | func (fs *flowServer) readRPCRequest() (*rpc.Request, error) { |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1080 | // Set a default timeout before reading from the flow. Without this timeout, |
| 1081 | // a client that sends no request or a partial request will retain the flow |
| 1082 | // indefinitely (and lock up server resources). |
Matt Rosencrantz | 8689793 | 2014-10-02 09:34:34 -0700 | [diff] [blame] | 1083 | initTimer := newTimer(defaultCallTimeout) |
| 1084 | defer initTimer.Stop() |
| 1085 | fs.flow.SetDeadline(initTimer.C) |
| 1086 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1087 | // Decode the initial request. |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1088 | var req rpc.Request |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1089 | if err := fs.dec.Decode(&req); err != nil { |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1090 | return nil, verror.New(verror.ErrBadProtocol, fs.ctx, newErrBadRequest(fs.ctx, err)) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1091 | } |
Matt Rosencrantz | 8689793 | 2014-10-02 09:34:34 -0700 | [diff] [blame] | 1092 | return &req, nil |
| 1093 | } |
| 1094 | |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1095 | func (fs *flowServer) processRequest() ([]interface{}, error) { |
Asim Shankar | 0cad083 | 2014-11-04 01:27:38 -0800 | [diff] [blame] | 1096 | fs.starttime = time.Now() |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1097 | req, err := fs.readRPCRequest() |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1098 | if err != nil { |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1099 | // We don't know what the rpc call was supposed to be, but we'll create |
Matt Rosencrantz | 1fa3277 | 2014-10-28 11:31:46 -0700 | [diff] [blame] | 1100 | // a placeholder span so we can capture annotations. |
Todd Wang | ad49204 | 2015-04-17 15:58:40 -0700 | [diff] [blame] | 1101 | fs.ctx, _ = vtrace.WithNewSpan(fs.ctx, fmt.Sprintf("\"%s\".UNKNOWN", fs.suffix)) |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1102 | return nil, err |
Matt Rosencrantz | 8689793 | 2014-10-02 09:34:34 -0700 | [diff] [blame] | 1103 | } |
Suharsh Sivakumar | c095fc2 | 2015-06-29 16:00:18 -0700 | [diff] [blame] | 1104 | // We must call fs.drainDecoderArgs for any error that occurs |
| 1105 | // after this point, and before we actually decode the arguments. |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1106 | fs.method = req.Method |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1107 | fs.suffix = strings.TrimLeft(req.Suffix, "/") |
Matt Rosencrantz | 8689793 | 2014-10-02 09:34:34 -0700 | [diff] [blame] | 1108 | |
Matt Rosencrantz | 88be118 | 2015-04-27 13:45:43 -0700 | [diff] [blame] | 1109 | if req.Language != "" { |
| 1110 | fs.ctx = i18n.WithLangID(fs.ctx, i18n.LangID(req.Language)) |
| 1111 | } |
| 1112 | |
Matt Rosencrantz | 9fe6082 | 2014-09-12 10:09:53 -0700 | [diff] [blame] | 1113 | // TODO(mattr): Currently this allows users to trigger trace collection |
| 1114 | // on the server even if they will not be allowed to collect the |
Matt Rosencrantz | 3197d6c | 2014-11-06 09:53:22 -0800 | [diff] [blame] | 1115 | // results later. This might be considered a DOS vector. |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1116 | spanName := fmt.Sprintf("\"%s\".%s", fs.suffix, fs.method) |
Todd Wang | ad49204 | 2015-04-17 15:58:40 -0700 | [diff] [blame] | 1117 | fs.ctx, _ = vtrace.WithContinuedTrace(fs.ctx, spanName, req.TraceRequest) |
Matt Rosencrantz | 137b8d2 | 2014-08-18 09:56:15 -0700 | [diff] [blame] | 1118 | |
Matt Rosencrantz | 137b8d2 | 2014-08-18 09:56:15 -0700 | [diff] [blame] | 1119 | var cancel context.CancelFunc |
Todd Wang | f6a0688 | 2015-02-27 17:38:01 -0800 | [diff] [blame] | 1120 | if !req.Deadline.IsZero() { |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1121 | fs.ctx, cancel = context.WithDeadline(fs.ctx, req.Deadline.Time) |
Matt Rosencrantz | 137b8d2 | 2014-08-18 09:56:15 -0700 | [diff] [blame] | 1122 | } else { |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1123 | fs.ctx, cancel = context.WithCancel(fs.ctx) |
Matt Rosencrantz | 137b8d2 | 2014-08-18 09:56:15 -0700 | [diff] [blame] | 1124 | } |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1125 | fs.flow.SetDeadline(fs.ctx.Done()) |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1126 | go fs.cancelContextOnClose(cancel) |
Matt Rosencrantz | 137b8d2 | 2014-08-18 09:56:15 -0700 | [diff] [blame] | 1127 | |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1128 | // Initialize security: blessings, discharges, etc. |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1129 | if err := fs.initSecurity(req); err != nil { |
Suharsh Sivakumar | c095fc2 | 2015-06-29 16:00:18 -0700 | [diff] [blame] | 1130 | fs.drainDecoderArgs(int(req.NumPosArgs)) |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1131 | return nil, err |
Andres Erbsen | b7f95f3 | 2014-07-07 12:07:56 -0700 | [diff] [blame] | 1132 | } |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1133 | // Lookup the invoker. |
Matt Rosencrantz | 311378b | 2015-03-25 15:26:12 -0700 | [diff] [blame] | 1134 | invoker, auth, err := fs.lookup(fs.suffix, fs.method) |
Todd Wang | ebb3b01 | 2015-02-09 21:59:05 -0800 | [diff] [blame] | 1135 | if err != nil { |
Suharsh Sivakumar | c095fc2 | 2015-06-29 16:00:18 -0700 | [diff] [blame] | 1136 | fs.drainDecoderArgs(int(req.NumPosArgs)) |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1137 | return nil, err |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1138 | } |
Matt Rosencrantz | 311378b | 2015-03-25 15:26:12 -0700 | [diff] [blame] | 1139 | |
| 1140 | // Note that we strip the reserved prefix when calling the invoker so |
| 1141 | // that __Glob will call Glob. Note that we've already assigned a |
| 1142 | // special invoker so that we never call the wrong method by mistake. |
| 1143 | strippedMethod := naming.StripReserved(fs.method) |
| 1144 | |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1145 | // Prepare invoker and decode args. |
| 1146 | numArgs := int(req.NumPosArgs) |
Cosmos Nicolaou | 5a3125a | 2015-07-10 11:19:20 -0700 | [diff] [blame] | 1147 | argptrs, tags, err := invoker.Prepare(fs.ctx, strippedMethod, numArgs) |
Asim Shankar | 0cad083 | 2014-11-04 01:27:38 -0800 | [diff] [blame] | 1148 | fs.tags = tags |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1149 | if err != nil { |
Suharsh Sivakumar | c095fc2 | 2015-06-29 16:00:18 -0700 | [diff] [blame] | 1150 | fs.drainDecoderArgs(numArgs) |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1151 | return nil, err |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1152 | } |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1153 | if called, want := req.NumPosArgs, uint64(len(argptrs)); called != want { |
Suharsh Sivakumar | c095fc2 | 2015-06-29 16:00:18 -0700 | [diff] [blame] | 1154 | fs.drainDecoderArgs(numArgs) |
| 1155 | return nil, newErrBadNumInputArgs(fs.ctx, fs.suffix, fs.method, called, want) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1156 | } |
| 1157 | for ix, argptr := range argptrs { |
| 1158 | if err := fs.dec.Decode(argptr); err != nil { |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1159 | return nil, newErrBadInputArg(fs.ctx, fs.suffix, fs.method, uint64(ix), err) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1160 | } |
| 1161 | } |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 1162 | |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1163 | // Check application's authorization policy. |
Todd Wang | 4264e4b | 2015-04-16 22:43:40 -0700 | [diff] [blame] | 1164 | if err := authorize(fs.ctx, fs, auth); err != nil { |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1165 | return nil, err |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1166 | } |
Matt Rosencrantz | bf0d9d9 | 2015-04-08 12:43:14 -0700 | [diff] [blame] | 1167 | |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1168 | // Invoke the method. |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1169 | results, err := invoker.Invoke(fs.ctx, fs, strippedMethod, argptrs) |
Robin Thellend | b16d716 | 2014-11-07 13:47:26 -0800 | [diff] [blame] | 1170 | fs.server.stats.record(fs.method, time.Since(fs.starttime)) |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1171 | return results, err |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1172 | } |
| 1173 | |
Suharsh Sivakumar | c095fc2 | 2015-06-29 16:00:18 -0700 | [diff] [blame] | 1174 | // drainDecoderArgs drains the next n arguments encoded onto the flows decoder. |
| 1175 | // This is needed to ensure that the client is able to encode all of its args |
| 1176 | // before the server closes its flow. This guarantees that the client will |
| 1177 | // consistently get the server's error response. |
| 1178 | // TODO(suharshs): Figure out a better way to solve this race condition without |
| 1179 | // unnecessarily reading all arguments. |
| 1180 | func (fs *flowServer) drainDecoderArgs(n int) error { |
| 1181 | for i := 0; i < n; i++ { |
| 1182 | if err := fs.dec.Ignore(); err != nil { |
| 1183 | return err |
| 1184 | } |
| 1185 | } |
| 1186 | return nil |
| 1187 | } |
| 1188 | |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1189 | func (fs *flowServer) cancelContextOnClose(cancel context.CancelFunc) { |
| 1190 | // Ensure that the context gets cancelled if the flow is closed |
| 1191 | // due to a network error, or client cancellation. |
| 1192 | select { |
| 1193 | case <-fs.flow.Closed(): |
| 1194 | // Here we remove the contexts channel as a deadline to the flow. |
| 1195 | // We do this to ensure clients get a consistent error when they read/write |
| 1196 | // after the flow is closed. Since the flow is already closed, it doesn't |
| 1197 | // matter that the context is also cancelled. |
| 1198 | fs.flow.SetDeadline(nil) |
| 1199 | cancel() |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1200 | case <-fs.ctx.Done(): |
Robin Thellend | c26c32e | 2014-10-06 17:44:04 -0700 | [diff] [blame] | 1201 | } |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1202 | } |
| 1203 | |
| 1204 | // lookup returns the invoker and authorizer responsible for serving the given |
| 1205 | // name and method. The suffix is stripped of any leading slashes. If it begins |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1206 | // with rpc.DebugKeyword, we use the internal debug dispatcher to look up the |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1207 | // invoker. Otherwise, and we use the server's dispatcher. The suffix and method |
| 1208 | // value may be modified to match the actual suffix and method to use. |
Matt Rosencrantz | 311378b | 2015-03-25 15:26:12 -0700 | [diff] [blame] | 1209 | func (fs *flowServer) lookup(suffix string, method string) (rpc.Invoker, security.Authorizer, error) { |
| 1210 | if naming.IsReserved(method) { |
Asim Shankar | 149b497 | 2015-04-23 13:29:58 -0700 | [diff] [blame] | 1211 | return reservedInvoker(fs.disp, fs.server.dispReserved), security.AllowEveryone(), nil |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1212 | } |
| 1213 | disp := fs.disp |
| 1214 | if naming.IsReserved(suffix) { |
| 1215 | disp = fs.server.dispReserved |
Ali Ghassemi | bac3403 | 2015-04-30 18:27:57 -0700 | [diff] [blame] | 1216 | } else if fs.server.isLeaf && suffix != "" { |
| 1217 | innerErr := verror.New(errUnexpectedSuffix, fs.ctx, suffix) |
| 1218 | return nil, nil, verror.New(verror.ErrUnknownSuffix, fs.ctx, suffix, innerErr) |
Robin Thellend | d24f084 | 2014-09-23 10:27:29 -0700 | [diff] [blame] | 1219 | } |
| 1220 | if disp != nil { |
Cosmos Nicolaou | 5a3125a | 2015-07-10 11:19:20 -0700 | [diff] [blame] | 1221 | obj, auth, err := disp.Lookup(fs.ctx, suffix) |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1222 | switch { |
| 1223 | case err != nil: |
Todd Wang | 9548d85 | 2015-02-10 16:15:59 -0800 | [diff] [blame] | 1224 | return nil, nil, err |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1225 | case obj != nil: |
Bogdan Caprita | 9592d9f | 2015-01-08 22:15:16 -0800 | [diff] [blame] | 1226 | invoker, err := objectToInvoker(obj) |
| 1227 | if err != nil { |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1228 | return nil, nil, verror.New(verror.ErrInternal, fs.ctx, "invalid received object", err) |
Bogdan Caprita | 9592d9f | 2015-01-08 22:15:16 -0800 | [diff] [blame] | 1229 | } |
| 1230 | return invoker, auth, nil |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1231 | } |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1232 | } |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1233 | return nil, nil, verror.New(verror.ErrUnknownSuffix, fs.ctx, suffix) |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1234 | } |
| 1235 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1236 | func objectToInvoker(obj interface{}) (rpc.Invoker, error) { |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1237 | if obj == nil { |
Bogdan Caprita | 9592d9f | 2015-01-08 22:15:16 -0800 | [diff] [blame] | 1238 | return nil, errors.New("nil object") |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1239 | } |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1240 | if invoker, ok := obj.(rpc.Invoker); ok { |
Bogdan Caprita | 9592d9f | 2015-01-08 22:15:16 -0800 | [diff] [blame] | 1241 | return invoker, nil |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1242 | } |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1243 | return rpc.ReflectInvoker(obj) |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1244 | } |
| 1245 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1246 | func (fs *flowServer) initSecurity(req *rpc.Request) error { |
Ankur | b905dae | 2015-03-04 12:38:20 -0800 | [diff] [blame] | 1247 | // LocalPrincipal is nil which means we are operating under |
Suharsh Sivakumar | 2c5d810 | 2015-03-23 08:49:12 -0700 | [diff] [blame] | 1248 | // SecurityNone. |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1249 | if fs.LocalPrincipal() == nil { |
Ankur | b905dae | 2015-03-04 12:38:20 -0800 | [diff] [blame] | 1250 | return nil |
| 1251 | } |
| 1252 | |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1253 | // If additional credentials are provided, make them available in the context |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1254 | // Detect unusable blessings now, rather then discovering they are unusable on |
| 1255 | // first use. |
| 1256 | // |
| 1257 | // TODO(ashankar,ataly): Potential confused deputy attack: The client provides |
| 1258 | // the server's identity as the blessing. Figure out what we want to do about |
| 1259 | // this - should servers be able to assume that a blessing is something that |
| 1260 | // does not have the authorizations that the server's own identity has? |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1261 | if got, want := req.GrantedBlessings.PublicKey(), fs.LocalPrincipal().PublicKey(); got != nil && !reflect.DeepEqual(got, want) { |
| 1262 | return verror.New(verror.ErrNoAccess, fs.ctx, fmt.Sprintf("blessing granted not bound to this server(%v vs %v)", got, want)) |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1263 | } |
Asim Shankar | b07ec69 | 2015-02-27 23:40:44 -0800 | [diff] [blame] | 1264 | fs.grantedBlessings = req.GrantedBlessings |
Ankur | b905dae | 2015-03-04 12:38:20 -0800 | [diff] [blame] | 1265 | |
Asim Shankar | b07ec69 | 2015-02-27 23:40:44 -0800 | [diff] [blame] | 1266 | var err error |
| 1267 | if fs.clientBlessings, err = serverDecodeBlessings(fs.flow.VCDataCache(), req.Blessings, fs.server.stats); err != nil { |
Suharsh Sivakumar | 720b704 | 2014-12-22 17:33:23 -0800 | [diff] [blame] | 1268 | // When the server can't access the blessings cache, the client is not following |
| 1269 | // protocol, so the server closes the VCs corresponding to the client endpoint. |
| 1270 | // TODO(suharshs,toddw): Figure out a way to only shutdown the current VC, instead |
| 1271 | // of all VCs connected to the RemoteEndpoint. |
| 1272 | fs.server.streamMgr.ShutdownEndpoint(fs.RemoteEndpoint()) |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1273 | return verror.New(verror.ErrBadProtocol, fs.ctx, newErrBadBlessingsCache(fs.ctx, err)) |
Suharsh Sivakumar | 720b704 | 2014-12-22 17:33:23 -0800 | [diff] [blame] | 1274 | } |
Ankur | b905dae | 2015-03-04 12:38:20 -0800 | [diff] [blame] | 1275 | // Verify that the blessings sent by the client in the request have the same public |
| 1276 | // key as those sent by the client during VC establishment. |
| 1277 | if got, want := fs.clientBlessings.PublicKey(), fs.flow.RemoteBlessings().PublicKey(); got != nil && !reflect.DeepEqual(got, want) { |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1278 | return verror.New(verror.ErrNoAccess, fs.ctx, fmt.Sprintf("blessings sent with the request are bound to a different public key (%v) from the blessing used during VC establishment (%v)", got, want)) |
Ankur | b905dae | 2015-03-04 12:38:20 -0800 | [diff] [blame] | 1279 | } |
Asim Shankar | 2bf7b1e | 2015-02-27 00:45:12 -0800 | [diff] [blame] | 1280 | fs.ackBlessings = true |
Suharsh Sivakumar | 720b704 | 2014-12-22 17:33:23 -0800 | [diff] [blame] | 1281 | |
Asim Shankar | 3ad0b8a | 2015-02-25 00:37:21 -0800 | [diff] [blame] | 1282 | for _, d := range req.Discharges { |
Asim Shankar | 0864282 | 2015-03-02 21:21:09 -0800 | [diff] [blame] | 1283 | fs.discharges[d.ID()] = d |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1284 | } |
| 1285 | return nil |
Robin Thellend | c26c32e | 2014-10-06 17:44:04 -0700 | [diff] [blame] | 1286 | } |
| 1287 | |
Todd Wang | 4264e4b | 2015-04-16 22:43:40 -0700 | [diff] [blame] | 1288 | func authorize(ctx *context.T, call security.Call, auth security.Authorizer) error { |
Matt Rosencrantz | 9dce9b2 | 2015-03-02 10:48:37 -0800 | [diff] [blame] | 1289 | if call.LocalPrincipal() == nil { |
Todd Wang | 5739dda | 2014-11-16 22:44:02 -0800 | [diff] [blame] | 1290 | // LocalPrincipal is nil means that the server wanted to avoid |
| 1291 | // authentication, and thus wanted to skip authorization as well. |
| 1292 | return nil |
| 1293 | } |
Asim Shankar | 8f05c22 | 2014-10-06 22:08:19 -0700 | [diff] [blame] | 1294 | if auth == nil { |
Asim Shankar | e4a8c09 | 2015-04-01 18:43:39 -0700 | [diff] [blame] | 1295 | auth = security.DefaultAuthorizer() |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1296 | } |
Todd Wang | 4264e4b | 2015-04-16 22:43:40 -0700 | [diff] [blame] | 1297 | if err := auth.Authorize(ctx, call); err != nil { |
Matt Rosencrantz | 250558f | 2015-03-17 11:37:31 -0700 | [diff] [blame] | 1298 | return verror.New(verror.ErrNoAccess, ctx, newErrBadAuth(ctx, call.Suffix(), call.Method(), err)) |
Asim Shankar | a5457f0 | 2014-10-24 23:23:07 -0700 | [diff] [blame] | 1299 | } |
| 1300 | return nil |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1301 | } |
| 1302 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1303 | // Send implements the rpc.Stream method. |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1304 | func (fs *flowServer) Send(item interface{}) error { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 1305 | defer apilog.LogCallf(nil, "item=")(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1306 | // The empty response header indicates what follows is a streaming result. |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1307 | if err := fs.enc.Encode(rpc.Response{}); err != nil { |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1308 | return err |
| 1309 | } |
| 1310 | return fs.enc.Encode(item) |
| 1311 | } |
| 1312 | |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1313 | // Recv implements the rpc.Stream method. |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1314 | func (fs *flowServer) Recv(itemptr interface{}) error { |
Cosmos Nicolaou | f3c1909 | 2015-05-27 17:53:37 -0700 | [diff] [blame] | 1315 | defer apilog.LogCallf(nil, "itemptr=")(nil, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
Matt Rosencrantz | 94502cf | 2015-03-18 09:43:44 -0700 | [diff] [blame] | 1316 | var req rpc.Request |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1317 | if err := fs.dec.Decode(&req); err != nil { |
| 1318 | return err |
| 1319 | } |
| 1320 | if req.EndStreamArgs { |
| 1321 | fs.endStreamArgs = true |
| 1322 | return io.EOF |
| 1323 | } |
| 1324 | return fs.dec.Decode(itemptr) |
| 1325 | } |
| 1326 | |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1327 | // Implementations of rpc.ServerCall and security.Call methods. |
Jiri Simsa | 5293dcb | 2014-05-10 09:56:38 -0700 | [diff] [blame] | 1328 | |
Todd Wang | 54feabe | 2015-04-15 23:38:26 -0700 | [diff] [blame] | 1329 | func (fs *flowServer) Security() security.Call { |
| 1330 | //nologcall |
| 1331 | return fs |
| 1332 | } |
Ankur | edd74ee | 2015-03-04 16:38:45 -0800 | [diff] [blame] | 1333 | func (fs *flowServer) LocalDischarges() map[string]security.Discharge { |
| 1334 | //nologcall |
| 1335 | return fs.flow.LocalDischarges() |
| 1336 | } |
Asim Shankar | 2519cc1 | 2014-11-10 21:16:53 -0800 | [diff] [blame] | 1337 | func (fs *flowServer) RemoteDischarges() map[string]security.Discharge { |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1338 | //nologcall |
| 1339 | return fs.discharges |
| 1340 | } |
Matt Rosencrantz | 98d6d7c | 2015-09-04 12:34:08 -0700 | [diff] [blame^] | 1341 | func (fs *flowServer) Server() rpc.Server { |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1342 | //nologcall |
| 1343 | return fs.server |
| 1344 | } |
Asim Shankar | 0cad083 | 2014-11-04 01:27:38 -0800 | [diff] [blame] | 1345 | func (fs *flowServer) Timestamp() time.Time { |
| 1346 | //nologcall |
| 1347 | return fs.starttime |
| 1348 | } |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1349 | func (fs *flowServer) Method() string { |
| 1350 | //nologcall |
| 1351 | return fs.method |
| 1352 | } |
Todd Wang | b31da59 | 2015-02-20 12:50:39 -0800 | [diff] [blame] | 1353 | func (fs *flowServer) MethodTags() []*vdl.Value { |
Asim Shankar | 0cad083 | 2014-11-04 01:27:38 -0800 | [diff] [blame] | 1354 | //nologcall |
| 1355 | return fs.tags |
| 1356 | } |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1357 | func (fs *flowServer) Suffix() string { |
| 1358 | //nologcall |
| 1359 | return fs.suffix |
| 1360 | } |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1361 | func (fs *flowServer) LocalPrincipal() security.Principal { |
| 1362 | //nologcall |
Asim Shankar | 8f05c22 | 2014-10-06 22:08:19 -0700 | [diff] [blame] | 1363 | return fs.flow.LocalPrincipal() |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1364 | } |
| 1365 | func (fs *flowServer) LocalBlessings() security.Blessings { |
| 1366 | //nologcall |
Asim Shankar | 8f05c22 | 2014-10-06 22:08:19 -0700 | [diff] [blame] | 1367 | return fs.flow.LocalBlessings() |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1368 | } |
| 1369 | func (fs *flowServer) RemoteBlessings() security.Blessings { |
| 1370 | //nologcall |
Asim Shankar | 2bf7b1e | 2015-02-27 00:45:12 -0800 | [diff] [blame] | 1371 | if !fs.clientBlessings.IsZero() { |
Suharsh Sivakumar | 720b704 | 2014-12-22 17:33:23 -0800 | [diff] [blame] | 1372 | return fs.clientBlessings |
| 1373 | } |
Asim Shankar | 8f05c22 | 2014-10-06 22:08:19 -0700 | [diff] [blame] | 1374 | return fs.flow.RemoteBlessings() |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1375 | } |
Suharsh Sivakumar | 380bf34 | 2015-02-27 15:38:27 -0800 | [diff] [blame] | 1376 | func (fs *flowServer) GrantedBlessings() security.Blessings { |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1377 | //nologcall |
Suharsh Sivakumar | 380bf34 | 2015-02-27 15:38:27 -0800 | [diff] [blame] | 1378 | return fs.grantedBlessings |
Mehrdad Afshari | cd9852b | 2014-09-26 11:07:35 -0700 | [diff] [blame] | 1379 | } |
| 1380 | func (fs *flowServer) LocalEndpoint() naming.Endpoint { |
| 1381 | //nologcall |
| 1382 | return fs.flow.LocalEndpoint() |
| 1383 | } |
| 1384 | func (fs *flowServer) RemoteEndpoint() naming.Endpoint { |
| 1385 | //nologcall |
| 1386 | return fs.flow.RemoteEndpoint() |
| 1387 | } |
Asim Shankar | 99b18a7 | 2015-04-25 23:19:28 -0700 | [diff] [blame] | 1388 | |
| 1389 | type proxyAuth struct { |
| 1390 | s *server |
| 1391 | } |
| 1392 | |
| 1393 | func (a proxyAuth) RPCStreamListenerOpt() {} |
| 1394 | |
| 1395 | func (a proxyAuth) Login(proxy stream.Flow) (security.Blessings, []security.Discharge, error) { |
Asim Shankar | 6b1c8b0 | 2015-04-26 22:09:45 -0700 | [diff] [blame] | 1396 | var ( |
| 1397 | principal = a.s.principal |
| 1398 | dc = a.s.dc |
| 1399 | ctx = a.s.ctx |
| 1400 | ) |
| 1401 | if principal == nil { |
| 1402 | return security.Blessings{}, nil, nil |
| 1403 | } |
| 1404 | proxyNames, _ := security.RemoteBlessingNames(ctx, security.NewCall(&security.CallParams{ |
| 1405 | LocalPrincipal: principal, |
| 1406 | RemoteBlessings: proxy.RemoteBlessings(), |
| 1407 | RemoteDischarges: proxy.RemoteDischarges(), |
| 1408 | RemoteEndpoint: proxy.RemoteEndpoint(), |
| 1409 | LocalEndpoint: proxy.LocalEndpoint(), |
| 1410 | })) |
| 1411 | blessings := principal.BlessingStore().ForPeer(proxyNames...) |
| 1412 | tpc := blessings.ThirdPartyCaveats() |
| 1413 | if len(tpc) == 0 { |
| 1414 | return blessings, nil, nil |
| 1415 | } |
| 1416 | // Set DischargeImpetus.Server = proxyNames. |
| 1417 | // See https://v.io/i/392 |
| 1418 | discharges := dc.PrepareDischarges(ctx, tpc, security.DischargeImpetus{}) |
| 1419 | return blessings, discharges, nil |
Asim Shankar | 99b18a7 | 2015-04-25 23:19:28 -0700 | [diff] [blame] | 1420 | } |
| 1421 | |
| 1422 | var _ manager.ProxyAuthenticator = proxyAuth{} |