Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 1 | // The app package contains the struct that keeps per javascript app state and handles translating |
| 2 | // javascript requests to veyron requests and vice versa. |
| 3 | package app |
| 4 | |
| 5 | import ( |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 6 | "bytes" |
| 7 | "encoding/hex" |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 8 | "flag" |
| 9 | "fmt" |
| 10 | "io" |
| 11 | "sync" |
| 12 | "time" |
| 13 | |
Jiri Simsa | 22b87ac | 2014-12-25 20:59:13 -0800 | [diff] [blame] | 14 | vsecurity "v.io/core/veyron/security" |
| 15 | "v.io/core/veyron2" |
| 16 | "v.io/core/veyron2/context" |
| 17 | "v.io/core/veyron2/ipc" |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 18 | "v.io/core/veyron2/naming" |
Jiri Simsa | 22b87ac | 2014-12-25 20:59:13 -0800 | [diff] [blame] | 19 | "v.io/core/veyron2/options" |
Jiri Simsa | 22b87ac | 2014-12-25 20:59:13 -0800 | [diff] [blame] | 20 | "v.io/core/veyron2/security" |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 21 | "v.io/core/veyron2/vdl" |
Jiri Simsa | 22b87ac | 2014-12-25 20:59:13 -0800 | [diff] [blame] | 22 | "v.io/core/veyron2/vdl/vdlroot/src/signature" |
Todd Wang | b63e9eb | 2015-02-10 19:57:39 -0800 | [diff] [blame] | 23 | "v.io/core/veyron2/verror" |
Jiri Simsa | 22b87ac | 2014-12-25 20:59:13 -0800 | [diff] [blame] | 24 | "v.io/core/veyron2/vlog" |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 25 | "v.io/core/veyron2/vom" |
Suharsh Sivakumar | 94d0066 | 2015-01-21 14:31:30 -0800 | [diff] [blame] | 26 | "v.io/core/veyron2/vtrace" |
Jiri Simsa | 1127716 | 2014-12-25 15:50:27 -0800 | [diff] [blame] | 27 | "v.io/wspr/veyron/services/wsprd/ipc/server" |
| 28 | "v.io/wspr/veyron/services/wsprd/lib" |
| 29 | "v.io/wspr/veyron/services/wsprd/namespace" |
| 30 | "v.io/wspr/veyron/services/wsprd/principal" |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 31 | ) |
| 32 | |
Mike Burrows | b6689c2 | 2014-10-08 11:14:15 -0700 | [diff] [blame] | 33 | // pkgPath is the prefix os errors in this package. |
Jiri Simsa | 22b87ac | 2014-12-25 20:59:13 -0800 | [diff] [blame] | 34 | const pkgPath = "v.io/core/veyron/services/wsprd/app" |
Mike Burrows | b6689c2 | 2014-10-08 11:14:15 -0700 | [diff] [blame] | 35 | |
| 36 | // Errors |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 37 | var ( |
Todd Wang | b63e9eb | 2015-02-10 19:57:39 -0800 | [diff] [blame] | 38 | marshallingError = verror.Register(pkgPath+".marshallingError", verror.NoRetry, "{1} {2} marshalling error {_}") |
| 39 | noResults = verror.Register(pkgPath+".noResults", verror.NoRetry, "{1} {2} no results from call {_}") |
| 40 | badCaveatType = verror.Register(pkgPath+".badCaveatType", verror.NoRetry, "{1} {2} bad caveat type {_}") |
| 41 | unknownBlessings = verror.Register(pkgPath+".unknownBlessings", verror.NoRetry, "{1} {2} unknown public id {_}") |
| 42 | invalidBlessingsHandle = verror.Register(pkgPath+".invalidBlessingsHandle", verror.NoRetry, "{1} {2} invalid blessings handle {_}") |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 43 | ) |
Mike Burrows | b6689c2 | 2014-10-08 11:14:15 -0700 | [diff] [blame] | 44 | |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 45 | // TODO(bjornick,nlacasse): Remove the retryTimeout flag once we able |
| 46 | // to pass it in from javascript. For now all RPCs have the same |
| 47 | // retryTimeout, set by command line flag. |
| 48 | var retryTimeout *int |
| 49 | |
| 50 | func init() { |
| 51 | // TODO(bjornick,nlacasse): Remove the retryTimeout flag once we able |
| 52 | // to pass it in from javascript. For now all RPCs have the same |
| 53 | // retryTimeout, set by command line flag. |
Nicolas LaCasse | 238aeb3 | 2014-11-26 11:32:10 -0800 | [diff] [blame] | 54 | retryTimeout = flag.Int("retry-timeout", 2, "Duration in seconds to retry starting an RPC call. 0 means never retry.") |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 55 | } |
| 56 | |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 57 | type outstandingRequest struct { |
| 58 | stream *outstandingStream |
| 59 | cancel context.CancelFunc |
| 60 | } |
| 61 | |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 62 | // Controller represents all the state of a Veyron Web App. This is the struct |
| 63 | // that is in charge performing all the veyron options. |
| 64 | type Controller struct { |
| 65 | // Protects everything. |
| 66 | // TODO(bjornick): We need to split this up. |
| 67 | sync.Mutex |
| 68 | |
Matt Rosencrantz | c90eb7b | 2015-01-09 08:32:01 -0800 | [diff] [blame] | 69 | // The context of this controller. |
| 70 | ctx *context.T |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 71 | |
Matt Rosencrantz | c90eb7b | 2015-01-09 08:32:01 -0800 | [diff] [blame] | 72 | // The cleanup function for this controller. |
| 73 | cancel context.CancelFunc |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 74 | |
| 75 | // The ipc.ListenSpec to use with server.Listen |
| 76 | listenSpec *ipc.ListenSpec |
| 77 | |
| 78 | // Used to generate unique ids for requests initiated by the proxy. |
| 79 | // These ids will be even so they don't collide with the ids generated |
| 80 | // by the client. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 81 | lastGeneratedId int32 |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 82 | |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 83 | // Used to keep track of data (streams and cancellation functions) for |
| 84 | // outstanding requests. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 85 | outstandingRequests map[int32]*outstandingRequest |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 86 | |
| 87 | // Maps flowids to the server that owns them. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 88 | flowMap map[int32]*server.Server |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 89 | |
| 90 | // A manager that Handles fetching and caching signature of remote services |
| 91 | signatureManager lib.SignatureManager |
| 92 | |
Nicolas LaCasse | 4f409ce | 2014-11-25 11:44:03 -0800 | [diff] [blame] | 93 | // We maintain multiple Veyron server per pipe for serving JavaScript |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 94 | // services. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 95 | servers map[uint32]*server.Server |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 96 | |
| 97 | // Creates a client writer for a given flow. This is a member so that tests can override |
| 98 | // the default implementation. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 99 | writerCreator func(id int32) lib.ClientWriter |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 100 | |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 101 | veyronProxyEP string |
| 102 | |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 103 | // Store for all the Blessings that javascript has a handle to. |
| 104 | blessingsStore *principal.JSBlessingsHandles |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 105 | } |
| 106 | |
| 107 | // NewController creates a new Controller. writerCreator will be used to create a new flow for rpcs to |
| 108 | // javascript server. veyronProxyEP is an endpoint for the veyron proxy to serve through. It can't be empty. |
Suharsh Sivakumar | 94d0066 | 2015-01-21 14:31:30 -0800 | [diff] [blame] | 109 | func NewController(ctx *context.T, writerCreator func(id int32) lib.ClientWriter, listenSpec *ipc.ListenSpec, namespaceRoots []string, p security.Principal) (*Controller, error) { |
| 110 | ctx, cancel := context.WithCancel(ctx) |
| 111 | |
| 112 | ctx, _ = vtrace.SetNewTrace(ctx) |
Matt Rosencrantz | 306d990 | 2015-01-10 17:46:07 -0800 | [diff] [blame] | 113 | |
Benjamin Prosnitz | 3c73850 | 2014-11-04 14:51:38 -0800 | [diff] [blame] | 114 | if namespaceRoots != nil { |
Matt Rosencrantz | 306d990 | 2015-01-10 17:46:07 -0800 | [diff] [blame] | 115 | veyron2.GetNamespace(ctx).SetRoots(namespaceRoots...) |
Benjamin Prosnitz | 3c73850 | 2014-11-04 14:51:38 -0800 | [diff] [blame] | 116 | } |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 117 | |
Suharsh Sivakumar | 94d0066 | 2015-01-21 14:31:30 -0800 | [diff] [blame] | 118 | ctx, err := veyron2.SetPrincipal(ctx, p) |
| 119 | if err != nil { |
| 120 | return nil, err |
| 121 | } |
| 122 | |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 123 | controller := &Controller{ |
Matt Rosencrantz | 306d990 | 2015-01-10 17:46:07 -0800 | [diff] [blame] | 124 | ctx: ctx, |
Suharsh Sivakumar | 94d0066 | 2015-01-21 14:31:30 -0800 | [diff] [blame] | 125 | cancel: cancel, |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 126 | writerCreator: writerCreator, |
| 127 | listenSpec: listenSpec, |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 128 | blessingsStore: principal.NewJSBlessingsHandles(), |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 129 | } |
| 130 | |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 131 | controller.setup() |
| 132 | return controller, nil |
| 133 | } |
| 134 | |
| 135 | // finishCall waits for the call to finish and write out the response to w. |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 136 | func (c *Controller) finishCall(ctx *context.T, w lib.ClientWriter, clientCall ipc.Call, msg *VeyronRPCRequest, span vtrace.Span) { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 137 | if msg.IsStreaming { |
| 138 | for { |
| 139 | var item interface{} |
| 140 | if err := clientCall.Recv(&item); err != nil { |
| 141 | if err == io.EOF { |
| 142 | break |
| 143 | } |
| 144 | w.Error(err) // Send streaming error as is |
| 145 | return |
| 146 | } |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 147 | vomItem, err := lib.VomEncode(item) |
Shyam Jayaraman | c96e1aa | 2014-11-12 16:42:39 -0800 | [diff] [blame] | 148 | if err != nil { |
Todd Wang | b63e9eb | 2015-02-10 19:57:39 -0800 | [diff] [blame] | 149 | w.Error(verror.New(marshallingError, ctx, item, err)) |
Shyam Jayaraman | c96e1aa | 2014-11-12 16:42:39 -0800 | [diff] [blame] | 150 | continue |
| 151 | } |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 152 | if err := w.Send(lib.ResponseStream, vomItem); err != nil { |
Todd Wang | b63e9eb | 2015-02-10 19:57:39 -0800 | [diff] [blame] | 153 | w.Error(verror.New(marshallingError, ctx, item)) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 154 | } |
| 155 | } |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 156 | if err := w.Send(lib.ResponseStreamClose, nil); err != nil { |
Todd Wang | b63e9eb | 2015-02-10 19:57:39 -0800 | [diff] [blame] | 157 | w.Error(verror.New(marshallingError, ctx, "ResponseStreamClose")) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 158 | } |
| 159 | } |
Todd Wang | f21e155 | 2015-02-18 13:21:52 -0800 | [diff] [blame] | 160 | results := make([]interface{}, msg.NumOutArgs) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 161 | // This array will have pointers to the values in result. |
Todd Wang | f21e155 | 2015-02-18 13:21:52 -0800 | [diff] [blame] | 162 | resultptrs := make([]interface{}, msg.NumOutArgs) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 163 | for ax := range results { |
| 164 | resultptrs[ax] = &results[ax] |
| 165 | } |
| 166 | if err := clientCall.Finish(resultptrs...); err != nil { |
| 167 | // return the call system error as is |
| 168 | w.Error(err) |
| 169 | return |
| 170 | } |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 171 | c.sendRPCResponse(ctx, w, span, results) |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 172 | } |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 173 | |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 174 | func (c *Controller) sendRPCResponse(ctx *context.T, w lib.ClientWriter, span vtrace.Span, results []interface{}) { |
Todd Wang | f21e155 | 2015-02-18 13:21:52 -0800 | [diff] [blame] | 175 | outargs := make([]vdl.AnyRep, len(results)) |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 176 | for i := range outargs { |
| 177 | outargs[i] = results[i] |
| 178 | } |
| 179 | |
| 180 | span.Finish() |
| 181 | traceRecord := vtrace.GetStore(ctx).TraceRecord(span.Trace()) |
| 182 | |
| 183 | response := VeyronRPCResponse{ |
| 184 | OutArgs: outargs, |
| 185 | TraceResponse: vtrace.Response{ |
Matt Rosencrantz | ce14861 | 2015-02-13 16:10:22 -0800 | [diff] [blame] | 186 | Flags: vtrace.CollectInMemory, |
| 187 | Trace: *traceRecord, |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 188 | }, |
| 189 | } |
| 190 | encoded, err := lib.VomEncode(response) |
Shyam Jayaraman | c96e1aa | 2014-11-12 16:42:39 -0800 | [diff] [blame] | 191 | if err != nil { |
| 192 | w.Error(err) |
| 193 | return |
| 194 | } |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 195 | if err := w.Send(lib.ResponseFinal, encoded); err != nil { |
Todd Wang | b63e9eb | 2015-02-10 19:57:39 -0800 | [diff] [blame] | 196 | w.Error(verror.Convert(marshallingError, ctx, err)) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 197 | } |
| 198 | } |
| 199 | |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 200 | func (c *Controller) startCall(ctx *context.T, w lib.ClientWriter, msg *VeyronRPCRequest, inArgs []interface{}) (ipc.Call, error) { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 201 | methodName := lib.UppercaseFirstCharacter(msg.Method) |
Asim Shankar | f3c61a3 | 2014-10-15 17:34:11 -0700 | [diff] [blame] | 202 | retryTimeoutOpt := options.RetryTimeout(time.Duration(*retryTimeout) * time.Second) |
Alex Fandrianto | aa5af7c | 2015-01-29 15:20:02 -0800 | [diff] [blame] | 203 | clientCall, err := veyron2.GetClient(ctx).StartCall(ctx, msg.Name, methodName, inArgs, retryTimeoutOpt) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 204 | if err != nil { |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 205 | return nil, fmt.Errorf("error starting call (name: %v, method: %v, args: %v): %v", msg.Name, methodName, inArgs, err) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 206 | } |
| 207 | |
| 208 | return clientCall, nil |
| 209 | } |
| 210 | |
| 211 | // Implements the serverHelper interface |
| 212 | |
| 213 | // CreateNewFlow creats a new server flow that will be used to write out |
| 214 | // streaming messages to Javascript. |
| 215 | func (c *Controller) CreateNewFlow(s *server.Server, stream ipc.Stream) *server.Flow { |
| 216 | c.Lock() |
| 217 | defer c.Unlock() |
| 218 | id := c.lastGeneratedId |
| 219 | c.lastGeneratedId += 2 |
| 220 | c.flowMap[id] = s |
| 221 | os := newStream() |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 222 | os.init(stream) |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 223 | c.outstandingRequests[id] = &outstandingRequest{ |
| 224 | stream: os, |
| 225 | } |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 226 | return &server.Flow{ID: id, Writer: c.writerCreator(id)} |
| 227 | } |
| 228 | |
| 229 | // CleanupFlow removes the bookkeping for a previously created flow. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 230 | func (c *Controller) CleanupFlow(id int32) { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 231 | c.Lock() |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 232 | request := c.outstandingRequests[id] |
| 233 | delete(c.outstandingRequests, id) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 234 | delete(c.flowMap, id) |
| 235 | c.Unlock() |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 236 | if request != nil && request.stream != nil { |
| 237 | request.stream.end() |
| 238 | request.stream.waitUntilDone() |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 239 | } |
| 240 | } |
| 241 | |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 242 | // RT returns the runtime of the app. |
Matt Rosencrantz | c90eb7b | 2015-01-09 08:32:01 -0800 | [diff] [blame] | 243 | func (c *Controller) Context() *context.T { |
| 244 | return c.ctx |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 245 | } |
| 246 | |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 247 | // AddBlessings adds the Blessings to the local blessings store and returns |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 248 | // the handle to it. This function exists because JS only has |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 249 | // a handle to the blessings to avoid shipping the certificate forest |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 250 | // to JS and back. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 251 | func (c *Controller) AddBlessings(blessings security.Blessings) int32 { |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 252 | return c.blessingsStore.Add(blessings) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 253 | } |
| 254 | |
| 255 | // Cleanup cleans up any outstanding rpcs. |
| 256 | func (c *Controller) Cleanup() { |
Matt Rosencrantz | 99ed678 | 2015-01-27 19:58:34 -0800 | [diff] [blame] | 257 | vlog.VI(0).Info("Cleaning up controller") |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 258 | c.Lock() |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 259 | |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 260 | for _, request := range c.outstandingRequests { |
| 261 | if request.cancel != nil { |
| 262 | request.cancel() |
| 263 | } |
| 264 | if request.stream != nil { |
| 265 | request.stream.end() |
| 266 | } |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 267 | } |
| 268 | |
Nicolas LaCasse | 60f423e | 2015-02-04 17:53:14 -0800 | [diff] [blame] | 269 | servers := []*server.Server{} |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 270 | for _, server := range c.servers { |
Nicolas LaCasse | 60f423e | 2015-02-04 17:53:14 -0800 | [diff] [blame] | 271 | servers = append(servers, server) |
| 272 | } |
| 273 | |
| 274 | c.Unlock() |
| 275 | |
| 276 | // We must unlock before calling server.Stop otherwise it can deadlock. |
| 277 | for _, server := range servers { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 278 | server.Stop() |
| 279 | } |
Benjamin Prosnitz | 8a51fe8 | 2014-10-16 13:05:14 -0700 | [diff] [blame] | 280 | |
Matt Rosencrantz | c90eb7b | 2015-01-09 08:32:01 -0800 | [diff] [blame] | 281 | c.cancel() |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 282 | } |
| 283 | |
| 284 | func (c *Controller) setup() { |
| 285 | c.signatureManager = lib.NewSignatureManager() |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 286 | c.outstandingRequests = make(map[int32]*outstandingRequest) |
| 287 | c.flowMap = make(map[int32]*server.Server) |
| 288 | c.servers = make(map[uint32]*server.Server) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 289 | } |
| 290 | |
| 291 | // SendOnStream writes data on id's stream. The actual network write will be |
| 292 | // done asynchronously. If there is an error, it will be sent to w. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 293 | func (c *Controller) SendOnStream(id int32, data string, w lib.ClientWriter) { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 294 | c.Lock() |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 295 | request := c.outstandingRequests[id] |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 296 | if request == nil || request.stream == nil { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 297 | vlog.Errorf("unknown stream: %d", id) |
| 298 | return |
| 299 | } |
Matt Rosencrantz | d608b37 | 2014-10-22 13:06:52 -0700 | [diff] [blame] | 300 | stream := request.stream |
| 301 | c.Unlock() |
| 302 | stream.send(data, w) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 303 | } |
| 304 | |
| 305 | // SendVeyronRequest makes a veyron request for the given flowId. If signal is non-nil, it will receive |
| 306 | // the call object after it has been constructed. |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 307 | func (c *Controller) sendVeyronRequest(ctx *context.T, id int32, msg *VeyronRPCRequest, inArgs []interface{}, w lib.ClientWriter, stream *outstandingStream, span vtrace.Span) { |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 308 | sig, err := c.getSignature(ctx, msg.Name) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 309 | if err != nil { |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 310 | w.Error(err) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 311 | return |
| 312 | } |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 313 | methName := lib.UppercaseFirstCharacter(msg.Method) |
| 314 | methSig, ok := signature.FirstMethod(sig, methName) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 315 | if !ok { |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 316 | w.Error(fmt.Errorf("method %q not found in signature: %#v", methName, sig)) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 317 | return |
| 318 | } |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 319 | if len(methSig.InArgs) != len(inArgs) { |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 320 | w.Error(fmt.Errorf("invalid number of arguments, expected: %v, got:%v", methSig, *msg)) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 321 | return |
| 322 | } |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 323 | |
| 324 | // We have to make the start call synchronous so we can make sure that we populate |
| 325 | // the call map before we can Handle a recieve call. |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 326 | call, err := c.startCall(ctx, w, msg, inArgs) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 327 | if err != nil { |
Jiri Simsa | 94f68d0 | 2015-02-17 10:22:08 -0800 | [diff] [blame] | 328 | w.Error(verror.Convert(verror.ErrInternal, ctx, err)) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 329 | return |
| 330 | } |
| 331 | |
| 332 | if stream != nil { |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 333 | stream.init(call) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 334 | } |
| 335 | |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 336 | c.finishCall(ctx, w, call, msg, span) |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 337 | c.Lock() |
| 338 | if request, ok := c.outstandingRequests[id]; ok { |
| 339 | delete(c.outstandingRequests, id) |
| 340 | if request.cancel != nil { |
| 341 | request.cancel() |
| 342 | } |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 343 | } |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 344 | c.Unlock() |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 345 | } |
| 346 | |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 347 | // TODO(mattr): This is a very limited implementation of ServerCall, |
| 348 | // but currently none of the methods the controller exports require |
| 349 | // any of this context information. |
| 350 | type localCall struct { |
| 351 | ctx *context.T |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 352 | vrpc *VeyronRPCRequest |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 353 | tags []interface{} |
| 354 | } |
| 355 | |
| 356 | func (l *localCall) Send(interface{}) error { return nil } |
| 357 | func (l *localCall) Recv(interface{}) error { return nil } |
| 358 | func (l *localCall) Blessings() security.Blessings { return nil } |
| 359 | func (l *localCall) Server() ipc.Server { return nil } |
| 360 | func (l *localCall) Context() *context.T { return l.ctx } |
| 361 | func (l *localCall) Timestamp() (t time.Time) { return } |
| 362 | func (l *localCall) Method() string { return l.vrpc.Method } |
| 363 | func (l *localCall) MethodTags() []interface{} { return l.tags } |
| 364 | func (l *localCall) Name() string { return l.vrpc.Name } |
| 365 | func (l *localCall) Suffix() string { return "" } |
| 366 | func (l *localCall) RemoteDischarges() map[string]security.Discharge { return nil } |
| 367 | func (l *localCall) LocalPrincipal() security.Principal { return nil } |
| 368 | func (l *localCall) LocalBlessings() security.Blessings { return nil } |
| 369 | func (l *localCall) RemoteBlessings() security.Blessings { return nil } |
| 370 | func (l *localCall) LocalEndpoint() naming.Endpoint { return nil } |
| 371 | func (l *localCall) RemoteEndpoint() naming.Endpoint { return nil } |
| 372 | |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 373 | func (c *Controller) handleInternalCall(ctx *context.T, msg *VeyronRPCRequest, decoder *vom.Decoder, w lib.ClientWriter, span vtrace.Span) { |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 374 | invoker, err := ipc.ReflectInvoker(ControllerServer(c)) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 375 | if err != nil { |
Jiri Simsa | 94f68d0 | 2015-02-17 10:22:08 -0800 | [diff] [blame] | 376 | w.Error(verror.Convert(verror.ErrInternal, ctx, err)) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 377 | return |
| 378 | } |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 379 | argptrs, tags, err := invoker.Prepare(msg.Method, int(msg.NumInArgs)) |
| 380 | if err != nil { |
Jiri Simsa | 94f68d0 | 2015-02-17 10:22:08 -0800 | [diff] [blame] | 381 | w.Error(verror.Convert(verror.ErrInternal, ctx, err)) |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 382 | return |
| 383 | } |
| 384 | for _, argptr := range argptrs { |
| 385 | if err := decoder.Decode(argptr); err != nil { |
Jiri Simsa | 94f68d0 | 2015-02-17 10:22:08 -0800 | [diff] [blame] | 386 | w.Error(verror.Convert(verror.ErrInternal, ctx, err)) |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 387 | return |
| 388 | } |
| 389 | } |
| 390 | results, err := invoker.Invoke(msg.Method, &localCall{ctx, msg, tags}, argptrs) |
| 391 | if err != nil { |
Jiri Simsa | 94f68d0 | 2015-02-17 10:22:08 -0800 | [diff] [blame] | 392 | w.Error(verror.Convert(verror.ErrInternal, ctx, err)) |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 393 | return |
| 394 | } |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 395 | c.sendRPCResponse(ctx, w, span, results) |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 396 | } |
| 397 | |
| 398 | // HandleVeyronRequest starts a veyron rpc and returns before the rpc has been completed. |
| 399 | func (c *Controller) HandleVeyronRequest(ctx *context.T, id int32, data string, w lib.ClientWriter) { |
| 400 | binbytes, err := hex.DecodeString(data) |
| 401 | if err != nil { |
Jiri Simsa | 94f68d0 | 2015-02-17 10:22:08 -0800 | [diff] [blame] | 402 | w.Error(verror.Convert(verror.ErrInternal, ctx, err)) |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 403 | return |
| 404 | } |
| 405 | decoder, err := vom.NewDecoder(bytes.NewReader(binbytes)) |
| 406 | if err != nil { |
Jiri Simsa | 94f68d0 | 2015-02-17 10:22:08 -0800 | [diff] [blame] | 407 | w.Error(verror.Convert(verror.ErrInternal, ctx, err)) |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 408 | return |
| 409 | } |
| 410 | |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 411 | var msg VeyronRPCRequest |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 412 | if err := decoder.Decode(&msg); err != nil { |
Jiri Simsa | 94f68d0 | 2015-02-17 10:22:08 -0800 | [diff] [blame] | 413 | w.Error(verror.Convert(verror.ErrInternal, ctx, err)) |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 414 | return |
| 415 | } |
| 416 | vlog.VI(2).Infof("VeyronRPC: %s.%s(..., streaming=%v)", msg.Name, msg.Method, msg.IsStreaming) |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 417 | spanName := fmt.Sprintf("<wspr>%q.%s", msg.Name, msg.Method) |
| 418 | ctx, span := vtrace.SetContinuedTrace(ctx, spanName, msg.TraceRequest) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 419 | |
Matt Rosencrantz | d2ecc8e | 2014-12-29 11:30:55 -0800 | [diff] [blame] | 420 | var cctx *context.T |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 421 | var cancel context.CancelFunc |
| 422 | |
| 423 | // TODO(mattr): To be consistent with go, we should not ignore 0 timeouts. |
| 424 | // However as a rollout strategy we must, otherwise there is a circular |
| 425 | // dependency between the WSPR change and the JS change that will follow. |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 426 | if msg.Timeout == lib.JSIPCNoTimeout || msg.Timeout == 0 { |
Matt Rosencrantz | b6f4b92 | 2015-01-05 13:31:55 -0800 | [diff] [blame] | 427 | cctx, cancel = context.WithCancel(ctx) |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 428 | } else { |
Matt Rosencrantz | b6f4b92 | 2015-01-05 13:31:55 -0800 | [diff] [blame] | 429 | cctx, cancel = context.WithTimeout(ctx, lib.JSToGoDuration(msg.Timeout)) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 430 | } |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 431 | |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 432 | // If this message is for an internal service, do a short-circuit dispatch here. |
| 433 | if msg.Name == "controller" { |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 434 | go c.handleInternalCall(ctx, &msg, decoder, w, span) |
Matt Rosencrantz | 786f727 | 2015-02-11 15:18:07 -0800 | [diff] [blame] | 435 | return |
| 436 | } |
| 437 | |
| 438 | inArgs := make([]interface{}, msg.NumInArgs) |
| 439 | for i := range inArgs { |
| 440 | if err := decoder.Decode(&inArgs[i]); err != nil { |
| 441 | w.Error(err) |
| 442 | return |
| 443 | } |
| 444 | } |
| 445 | |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 446 | request := &outstandingRequest{ |
| 447 | cancel: cancel, |
| 448 | } |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 449 | if msg.IsStreaming { |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 450 | // If this rpc is streaming, we would expect that the client would try to send |
| 451 | // on this stream. Since the initial handshake is done asynchronously, we have |
| 452 | // to put the outstanding stream in the map before we make the async call so that |
| 453 | // the future send know which queue to write to, even if the client call isn't |
| 454 | // actually ready yet. |
| 455 | request.stream = newStream() |
| 456 | } |
| 457 | c.Lock() |
| 458 | c.outstandingRequests[id] = request |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 459 | go c.sendVeyronRequest(cctx, id, &msg, inArgs, w, request.stream, span) |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 460 | c.Unlock() |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 461 | } |
| 462 | |
| 463 | // HandleVeyronCancellation cancels the request corresponding to the |
| 464 | // given id if it is still outstanding. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 465 | func (c *Controller) HandleVeyronCancellation(id int32) { |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 466 | c.Lock() |
| 467 | defer c.Unlock() |
| 468 | if request, ok := c.outstandingRequests[id]; ok && request.cancel != nil { |
| 469 | request.cancel() |
| 470 | } |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 471 | } |
| 472 | |
| 473 | // CloseStream closes the stream for a given id. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 474 | func (c *Controller) CloseStream(id int32) { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 475 | c.Lock() |
| 476 | defer c.Unlock() |
Matt Rosencrantz | 4aabe57 | 2014-10-22 09:25:50 -0700 | [diff] [blame] | 477 | if request, ok := c.outstandingRequests[id]; ok && request.stream != nil { |
| 478 | request.stream.end() |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 479 | return |
| 480 | } |
Matt Rosencrantz | 99ed678 | 2015-01-27 19:58:34 -0800 | [diff] [blame] | 481 | vlog.Errorf("close called on non-existent call: %v", id) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 482 | } |
| 483 | |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 484 | func (c *Controller) maybeCreateServer(serverId uint32) (*server.Server, error) { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 485 | c.Lock() |
| 486 | defer c.Unlock() |
| 487 | if server, ok := c.servers[serverId]; ok { |
| 488 | return server, nil |
| 489 | } |
| 490 | server, err := server.NewServer(serverId, c.listenSpec, c) |
| 491 | if err != nil { |
| 492 | return nil, err |
| 493 | } |
| 494 | c.servers[serverId] = server |
| 495 | return server, nil |
| 496 | } |
| 497 | |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 498 | // HandleLookupResponse handles the result of a Dispatcher.Lookup call that was |
| 499 | // run by the Javascript server. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 500 | func (c *Controller) HandleLookupResponse(id int32, data string) { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 501 | c.Lock() |
| 502 | server := c.flowMap[id] |
| 503 | c.Unlock() |
| 504 | if server == nil { |
Matt Rosencrantz | 99ed678 | 2015-01-27 19:58:34 -0800 | [diff] [blame] | 505 | vlog.Errorf("unexpected result from JavaScript. No channel "+ |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 506 | "for MessageId: %d exists. Ignoring the results.", id) |
| 507 | //Ignore unknown responses that don't belong to any channel |
| 508 | return |
| 509 | } |
| 510 | server.HandleLookupResponse(id, data) |
| 511 | } |
| 512 | |
| 513 | // HandleAuthResponse handles the result of a Authorizer.Authorize call that was |
| 514 | // run by the Javascript server. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 515 | func (c *Controller) HandleAuthResponse(id int32, data string) { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 516 | c.Lock() |
| 517 | server := c.flowMap[id] |
| 518 | c.Unlock() |
| 519 | if server == nil { |
Matt Rosencrantz | 99ed678 | 2015-01-27 19:58:34 -0800 | [diff] [blame] | 520 | vlog.Errorf("unexpected result from JavaScript. No channel "+ |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 521 | "for MessageId: %d exists. Ignoring the results.", id) |
| 522 | //Ignore unknown responses that don't belong to any channel |
| 523 | return |
| 524 | } |
| 525 | server.HandleAuthResponse(id, data) |
| 526 | } |
| 527 | |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 528 | // Serve instructs WSPR to start listening for calls on behalf |
| 529 | // of a javascript server. |
| 530 | func (c *Controller) Serve(_ ipc.ServerContext, name string, serverId uint32) error { |
| 531 | server, err := c.maybeCreateServer(serverId) |
| 532 | if err != nil { |
| 533 | return verror.Convert(verror.ErrInternal, nil, err) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 534 | } |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 535 | vlog.VI(2).Infof("serving under name: %q", name) |
| 536 | if err := server.Serve(name); err != nil { |
| 537 | return verror.Convert(verror.ErrInternal, nil, err) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 538 | } |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 539 | return nil |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 540 | } |
| 541 | |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 542 | // Stop instructs WSPR to stop listening for calls for the |
| 543 | // given javascript server. |
| 544 | func (c *Controller) Stop(_ ipc.ServerContext, serverId uint32) error { |
| 545 | c.Lock() |
| 546 | server := c.servers[serverId] |
| 547 | if server == nil { |
| 548 | c.Unlock() |
| 549 | return nil |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 550 | } |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 551 | delete(c.servers, serverId) |
| 552 | c.Unlock() |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 553 | |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 554 | server.Stop() |
| 555 | return nil |
| 556 | } |
| 557 | |
| 558 | // AddName adds a published name to an existing server. |
| 559 | func (c *Controller) AddName(_ ipc.ServerContext, serverId uint32, name string) error { |
Nicolas LaCasse | 4f409ce | 2014-11-25 11:44:03 -0800 | [diff] [blame] | 560 | // Create a server for the pipe, if it does not exist already |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 561 | server, err := c.maybeCreateServer(serverId) |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 562 | if err != nil { |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 563 | return verror.Convert(verror.ErrInternal, nil, err) |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 564 | } |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 565 | // Add name |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 566 | if err := server.AddName(name); err != nil { |
| 567 | return verror.Convert(verror.ErrInternal, nil, err) |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 568 | } |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 569 | return nil |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 570 | } |
| 571 | |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 572 | // RemoveName removes a published name from an existing server. |
| 573 | func (c *Controller) RemoveName(_ ipc.ServerContext, serverId uint32, name string) error { |
Nicolas LaCasse | 4f409ce | 2014-11-25 11:44:03 -0800 | [diff] [blame] | 574 | // Create a server for the pipe, if it does not exist already |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 575 | server, err := c.maybeCreateServer(serverId) |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 576 | if err != nil { |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 577 | return verror.Convert(verror.ErrInternal, nil, err) |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 578 | } |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 579 | // Remove name |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 580 | server.RemoveName(name) |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 581 | // Remove name from signature cache as well |
Matt Rosencrantz | d4bcec6 | 2015-02-18 14:05:49 -0800 | [diff] [blame] | 582 | c.signatureManager.FlushCacheEntry(name) |
| 583 | return nil |
Ali Ghassemi | 1008bbe | 2014-11-07 16:36:08 -0800 | [diff] [blame] | 584 | } |
| 585 | |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 586 | // HandleServerResponse handles the completion of outstanding calls to JavaScript services |
| 587 | // by filling the corresponding channel with the result from JavaScript. |
Benjamin Prosnitz | 86d5228 | 2014-12-19 15:48:38 -0800 | [diff] [blame] | 588 | func (c *Controller) HandleServerResponse(id int32, data string) { |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 589 | c.Lock() |
| 590 | server := c.flowMap[id] |
| 591 | c.Unlock() |
| 592 | if server == nil { |
Matt Rosencrantz | 99ed678 | 2015-01-27 19:58:34 -0800 | [diff] [blame] | 593 | vlog.Errorf("unexpected result from JavaScript. No channel "+ |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 594 | "for MessageId: %d exists. Ignoring the results.", id) |
| 595 | //Ignore unknown responses that don't belong to any channel |
| 596 | return |
| 597 | } |
| 598 | server.HandleServerResponse(id, data) |
| 599 | } |
| 600 | |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 601 | // parseVeyronRequest parses a json rpc request into a VeyronRPCRequest object. |
| 602 | func (c *Controller) parseVeyronRequest(data string) (*VeyronRPCRequest, error) { |
| 603 | var msg VeyronRPCRequest |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 604 | if err := lib.VomDecode(data, &msg); err != nil { |
| 605 | return nil, err |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 606 | } |
Matt Rosencrantz | ac1e3a8 | 2015-02-12 16:04:28 -0800 | [diff] [blame] | 607 | vlog.VI(2).Infof("VeyronRPCRequest: %s.%s(..., streaming=%v)", msg.Name, msg.Method, msg.IsStreaming) |
Benjamin Prosnitz | a2ac3b3 | 2014-12-12 11:40:31 -0800 | [diff] [blame] | 608 | return &msg, nil |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 609 | } |
| 610 | |
Matt Rosencrantz | ec3b950 | 2015-02-19 12:53:13 -0800 | [diff] [blame^] | 611 | // getSignature uses the signature manager to get and cache the signature of a remote server. |
Matt Rosencrantz | d2ecc8e | 2014-12-29 11:30:55 -0800 | [diff] [blame] | 612 | func (c *Controller) getSignature(ctx *context.T, name string) ([]signature.Interface, error) { |
Asim Shankar | f3c61a3 | 2014-10-15 17:34:11 -0700 | [diff] [blame] | 613 | retryTimeoutOpt := options.RetryTimeout(time.Duration(*retryTimeout) * time.Second) |
Todd Wang | f83dbfc | 2015-01-09 11:54:46 -0800 | [diff] [blame] | 614 | return c.signatureManager.Signature(ctx, name, retryTimeoutOpt) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 615 | } |
| 616 | |
Matt Rosencrantz | ec3b950 | 2015-02-19 12:53:13 -0800 | [diff] [blame^] | 617 | // Signature uses the signature manager to get and cache the signature of a remote server. |
| 618 | func (c *Controller) Signature(ctx ipc.ServerContext, name string) ([]signature.Interface, error) { |
| 619 | return c.getSignature(ctx.Context(), name) |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 620 | } |
| 621 | |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 622 | // UnlinkJSBlessings removes the given blessings from the blessings store. |
| 623 | func (c *Controller) UnlinkJSBlessings(_ ipc.ServerContext, handle int32) error { |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 624 | c.blessingsStore.Remove(handle) |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 625 | return nil |
Jiri Simsa | 78b646f | 2014-10-08 10:23:05 -0700 | [diff] [blame] | 626 | } |
| 627 | |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 628 | // BlessPublicKey creates a new blessing. |
| 629 | func (c *Controller) BlessPublicKey(_ ipc.ServerContext, |
| 630 | handle int32, |
| 631 | caveats []security.Caveat, |
| 632 | duration time.Duration, |
| 633 | extension string) (int32, string, error) { |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 634 | var blessee security.Blessings |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 635 | if blessee = c.blessingsStore.Get(handle); blessee == nil { |
| 636 | return 0, "", verror.New(invalidBlessingsHandle, nil) |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 637 | } |
| 638 | |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 639 | expiryCav, err := security.ExpiryCaveat(time.Now().Add(duration)) |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 640 | if err != nil { |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 641 | return 0, "", err |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 642 | } |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 643 | caveats = append(caveats, expiryCav) |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 644 | |
| 645 | // TODO(ataly, ashankar, bjornick): Currently the Bless operation is carried |
| 646 | // out using the Default blessing in this principal's blessings store. We |
| 647 | // should change this so that the JS blessing request can also specify the |
| 648 | // blessing to be used for the Bless operation. |
Matt Rosencrantz | c90eb7b | 2015-01-09 08:32:01 -0800 | [diff] [blame] | 649 | p := veyron2.GetPrincipal(c.ctx) |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 650 | key := blessee.PublicKey() |
| 651 | blessing := p.BlessingStore().Default() |
| 652 | blessings, err := p.Bless(key, blessing, extension, caveats[0], caveats[1:]...) |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 653 | if err != nil { |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 654 | return 0, "", err |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 655 | } |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 656 | handle = c.blessingsStore.Add(blessings) |
| 657 | encodedKey, err := principal.EncodePublicKey(blessings.PublicKey()) |
| 658 | if err != nil { |
| 659 | return 0, "", err |
| 660 | } |
| 661 | return handle, encodedKey, nil |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 662 | } |
| 663 | |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 664 | // CreateBlessings creates a new principal self-blessed with the given extension. |
| 665 | func (c *Controller) CreateBlessings(_ ipc.ServerContext, |
| 666 | extension string) (int32, string, error) { |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 667 | p, err := vsecurity.NewPrincipal() |
| 668 | if err != nil { |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 669 | return 0, "", verror.Convert(verror.ErrInternal, nil, err) |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 670 | } |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 671 | blessings, err := p.BlessSelf(extension) |
| 672 | if err != nil { |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 673 | return 0, "", verror.Convert(verror.ErrInternal, nil, err) |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 674 | } |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 675 | |
| 676 | handle := c.blessingsStore.Add(blessings) |
| 677 | encodedKey, err := principal.EncodePublicKey(blessings.PublicKey()) |
| 678 | if err != nil { |
| 679 | return 0, "", err |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 680 | } |
Matt Rosencrantz | feefb77 | 2015-02-18 14:58:49 -0800 | [diff] [blame] | 681 | return handle, encodedKey, nil |
Ankur | e788924 | 2014-10-20 18:37:29 -0700 | [diff] [blame] | 682 | } |
| 683 | |
Matt Rosencrantz | ec3b950 | 2015-02-19 12:53:13 -0800 | [diff] [blame^] | 684 | func (c *Controller) RemoteBlessings(ctx ipc.ServerContext, name, method string) ([]string, error) { |
| 685 | vlog.VI(2).Infof("requesting remote blessings for %q", name) |
Nicolas LaCasse | e8f8ec0 | 2015-02-04 14:14:52 -0800 | [diff] [blame] | 686 | |
Matt Rosencrantz | ec3b950 | 2015-02-19 12:53:13 -0800 | [diff] [blame^] | 687 | cctx, cancel := context.WithTimeout(ctx.Context(), 5*time.Second) |
Nicolas LaCasse | e8f8ec0 | 2015-02-04 14:14:52 -0800 | [diff] [blame] | 688 | defer cancel() |
| 689 | |
Matt Rosencrantz | ec3b950 | 2015-02-19 12:53:13 -0800 | [diff] [blame^] | 690 | call, err := veyron2.GetClient(cctx).StartCall(cctx, name, method, nil) |
Nicolas LaCasse | e8f8ec0 | 2015-02-04 14:14:52 -0800 | [diff] [blame] | 691 | if err != nil { |
Matt Rosencrantz | ec3b950 | 2015-02-19 12:53:13 -0800 | [diff] [blame^] | 692 | return nil, verror.Convert(verror.ErrInternal, cctx, err) |
Nicolas LaCasse | e8f8ec0 | 2015-02-04 14:14:52 -0800 | [diff] [blame] | 693 | } |
| 694 | |
Nicolas LaCasse | e8f8ec0 | 2015-02-04 14:14:52 -0800 | [diff] [blame] | 695 | blessings, _ := call.RemoteBlessings() |
| 696 | return blessings, nil |
| 697 | } |
| 698 | |
Ali Ghassemi | aa0ea24 | 2014-10-20 12:55:39 -0700 | [diff] [blame] | 699 | // HandleNamespaceRequest uses the namespace client to respond to namespace specific requests such as glob |
Matt Rosencrantz | d2ecc8e | 2014-12-29 11:30:55 -0800 | [diff] [blame] | 700 | func (c *Controller) HandleNamespaceRequest(ctx *context.T, data string, w lib.ClientWriter) { |
Matt Rosencrantz | c90eb7b | 2015-01-09 08:32:01 -0800 | [diff] [blame] | 701 | namespace.HandleRequest(ctx, data, w) |
Ali Ghassemi | aa0ea24 | 2014-10-20 12:55:39 -0700 | [diff] [blame] | 702 | } |