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