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