| OLD | NEW |
| 1 // Copyright 2015 The LUCI Authors. All rights reserved. | 1 // Copyright 2015 The LUCI Authors. All rights reserved. |
| 2 // Use of this source code is governed under the Apache License, Version 2.0 | 2 // Use of this source code is governed under the Apache License, Version 2.0 |
| 3 // that can be found in the LICENSE file. | 3 // that can be found in the LICENSE file. |
| 4 | 4 |
| 5 package streamserver | 5 package streamserver |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "net" | 8 "net" |
| 9 | 9 |
| 10 "golang.org/x/net/context" | 10 "golang.org/x/net/context" |
| 11 | 11 |
| 12 "github.com/luci/luci-go/common/errors" | 12 "github.com/luci/luci-go/common/errors" |
| 13 log "github.com/luci/luci-go/common/logging" | 13 log "github.com/luci/luci-go/common/logging" |
| 14 "github.com/luci/luci-go/logdog/client/butlerlib/streamclient" | 14 "github.com/luci/luci-go/logdog/client/butlerlib/streamclient" |
| 15 | 15 |
| 16 » "github.com/Microsoft/go-winio" | 16 » "gopkg.in/natefinch/npipe.v2" |
| 17 ) | 17 ) |
| 18 | 18 |
| 19 // maxWindowsNamedPipeLength is the maximum length of a Windows named pipe. | 19 // maxWindowsNamedPipeLength is the maximum length of a Windows named pipe. |
| 20 const maxWindowsNamedPipeLength = 256 | 20 const maxWindowsNamedPipeLength = 256 |
| 21 | 21 |
| 22 // NewNamedPipeServer instantiates a new Windows named pipe server instance. | 22 // NewNamedPipeServer instantiates a new Windows named pipe server instance. |
| 23 func NewNamedPipeServer(ctx context.Context, name string) (StreamServer, error)
{ | 23 func NewNamedPipeServer(ctx context.Context, name string) (StreamServer, error)
{ |
| 24 switch l := len(name); { | 24 switch l := len(name); { |
| 25 case l == 0: | 25 case l == 0: |
| 26 return nil, errors.New("cannot have empty name") | 26 return nil, errors.New("cannot have empty name") |
| 27 case l > maxWindowsNamedPipeLength: | 27 case l > maxWindowsNamedPipeLength: |
| 28 return nil, errors.Reason("name exceeds maximum length %(max)d")
. | 28 return nil, errors.Reason("name exceeds maximum length %(max)d")
. |
| 29 D("name", name). | 29 D("name", name). |
| 30 D("max", maxWindowsNamedPipeLength). | 30 D("max", maxWindowsNamedPipeLength). |
| 31 Err() | 31 Err() |
| 32 } | 32 } |
| 33 | 33 |
| 34 ctx = log.SetField(ctx, "name", name) | 34 ctx = log.SetField(ctx, "name", name) |
| 35 return &listenerStreamServer{ | 35 return &listenerStreamServer{ |
| 36 Context: ctx, | 36 Context: ctx, |
| 37 gen: func() (net.Listener, string, error) { | 37 gen: func() (net.Listener, string, error) { |
| 38 address := "net.pipe:" + name | 38 address := "net.pipe:" + name |
| 39 pipePath := streamclient.LocalNamedPipePath(name) | 39 pipePath := streamclient.LocalNamedPipePath(name) |
| 40 log.Fields{ | 40 log.Fields{ |
| 41 "addr": address, | 41 "addr": address, |
| 42 "pipePath": pipePath, | 42 "pipePath": pipePath, |
| 43 }.Debugf(ctx, "Creating Windows server socket Listener."
) | 43 }.Debugf(ctx, "Creating Windows server socket Listener."
) |
| 44 | 44 |
| 45 » » » l, err := winio.ListenPipe(pipePath, nil) | 45 » » » l, err := npipe.Listen(pipePath) |
| 46 if err != nil { | 46 if err != nil { |
| 47 return nil, "", errors.Annotate(err).Reason("fai
led to listen on named pipe").Err() | 47 return nil, "", errors.Annotate(err).Reason("fai
led to listen on named pipe").Err() |
| 48 } | 48 } |
| 49 return l, address, nil | 49 return l, address, nil |
| 50 }, | 50 }, |
| 51 }, nil | 51 }, nil |
| 52 } | 52 } |
| OLD | NEW |