mirror of
https://gitea.com/Sirherobrine23/tea.git
synced 2024-08-21 19:07:24 +00:00
81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
|
//
|
||
|
// Copyright (c) 2014 David Mzareulyan
|
||
|
//
|
||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy of this software
|
||
|
// and associated documentation files (the "Software"), to deal in the Software without restriction,
|
||
|
// including without limitation the rights to use, copy, modify, merge, publish, distribute,
|
||
|
// sublicense, and/or sell copies of the Software, and to permit persons to whom the Software
|
||
|
// is furnished to do so, subject to the following conditions:
|
||
|
//
|
||
|
// The above copyright notice and this permission notice shall be included in all copies or substantial
|
||
|
// portions of the Software.
|
||
|
//
|
||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
|
||
|
// BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||
|
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
|
||
|
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||
|
//
|
||
|
|
||
|
// +build windows
|
||
|
|
||
|
package sshagent
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"io"
|
||
|
"net"
|
||
|
"sync"
|
||
|
|
||
|
"golang.org/x/crypto/ssh/agent"
|
||
|
)
|
||
|
|
||
|
// New returns a new agent.Agent and the (custom) connection it uses
|
||
|
// to communicate with a running pagent.exe instance (see README.md)
|
||
|
func New() (agent.Agent, net.Conn, error) {
|
||
|
if !Available() {
|
||
|
return nil, nil, errors.New("SSH agent requested but Pageant not running")
|
||
|
}
|
||
|
|
||
|
return agent.NewClient(&conn{}), nil, nil
|
||
|
}
|
||
|
|
||
|
type conn struct {
|
||
|
sync.Mutex
|
||
|
buf []byte
|
||
|
}
|
||
|
|
||
|
func (c *conn) Close() {
|
||
|
c.Lock()
|
||
|
defer c.Unlock()
|
||
|
c.buf = nil
|
||
|
}
|
||
|
|
||
|
func (c *conn) Write(p []byte) (int, error) {
|
||
|
c.Lock()
|
||
|
defer c.Unlock()
|
||
|
|
||
|
resp, err := query(p)
|
||
|
if err != nil {
|
||
|
return 0, err
|
||
|
}
|
||
|
|
||
|
c.buf = append(c.buf, resp...)
|
||
|
|
||
|
return len(p), nil
|
||
|
}
|
||
|
|
||
|
func (c *conn) Read(p []byte) (int, error) {
|
||
|
c.Lock()
|
||
|
defer c.Unlock()
|
||
|
|
||
|
if len(c.buf) == 0 {
|
||
|
return 0, io.EOF
|
||
|
}
|
||
|
|
||
|
n := copy(p, c.buf)
|
||
|
c.buf = c.buf[n:]
|
||
|
|
||
|
return n, nil
|
||
|
}
|