mirror of
https://github.com/containers/podman.git
synced 2025-07-18 10:08:07 +08:00
280f5d8cb0
implement new ssh interface into podman this completely redesigns the entire functionality of podman image scp, podman system connection add, and podman --remote. All references to golang.org/x/crypto/ssh have been moved to common as have native ssh/scp execs and the new usage of the sftp package. this PR adds a global flag, --ssh to podman which has two valid inputs `golang` and `native` where golang is the default. Users should not notice any difference in their everyday workflows if they continue using the golang option. UNLESS they have been using an improperly verified ssh key, this will now fail. This is because podman was incorrectly using the ssh callback method to IGNORE the ssh known hosts file which is very insecure and golang tells you not yo use this in production. The native paths allows for immense flexibility, with a new containers.conf field `SSH_CONFIG` that specifies a specific ssh config file to be used in all operations. Else the users ~/.ssh/config file will be used. podman --remote currently only uses the golang path, given its deep interconnection with dialing multiple clients and urls. My goal after this PR is to go back and abstract the idea of podman --remote from golang's dialed clients, as it should not be so intrinsically connected. Overall, this is a v1 of a long process of offering native ssh, and one that covers some good ground with podman system connection add and podman image scp. Signed-off-by: Charlie Doern <cdoern@redhat.com>
97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
package sftp
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
type allocator struct {
|
|
sync.Mutex
|
|
available [][]byte
|
|
// map key is the request order
|
|
used map[uint32][][]byte
|
|
}
|
|
|
|
func newAllocator() *allocator {
|
|
return &allocator{
|
|
// micro optimization: initialize available pages with an initial capacity
|
|
available: make([][]byte, 0, SftpServerWorkerCount*2),
|
|
used: make(map[uint32][][]byte),
|
|
}
|
|
}
|
|
|
|
// GetPage returns a previously allocated and unused []byte or create a new one.
|
|
// The slice have a fixed size = maxMsgLength, this value is suitable for both
|
|
// receiving new packets and reading the files to serve
|
|
func (a *allocator) GetPage(requestOrderID uint32) []byte {
|
|
a.Lock()
|
|
defer a.Unlock()
|
|
|
|
var result []byte
|
|
|
|
// get an available page and remove it from the available ones.
|
|
if len(a.available) > 0 {
|
|
truncLength := len(a.available) - 1
|
|
result = a.available[truncLength]
|
|
|
|
a.available[truncLength] = nil // clear out the internal pointer
|
|
a.available = a.available[:truncLength] // truncate the slice
|
|
}
|
|
|
|
// no preallocated slice found, just allocate a new one
|
|
if result == nil {
|
|
result = make([]byte, maxMsgLength)
|
|
}
|
|
|
|
// put result in used pages
|
|
a.used[requestOrderID] = append(a.used[requestOrderID], result)
|
|
|
|
return result
|
|
}
|
|
|
|
// ReleasePages marks unused all pages in use for the given requestID
|
|
func (a *allocator) ReleasePages(requestOrderID uint32) {
|
|
a.Lock()
|
|
defer a.Unlock()
|
|
|
|
if used := a.used[requestOrderID]; len(used) > 0 {
|
|
a.available = append(a.available, used...)
|
|
}
|
|
delete(a.used, requestOrderID)
|
|
}
|
|
|
|
// Free removes all the used and available pages.
|
|
// Call this method when the allocator is not needed anymore
|
|
func (a *allocator) Free() {
|
|
a.Lock()
|
|
defer a.Unlock()
|
|
|
|
a.available = nil
|
|
a.used = make(map[uint32][][]byte)
|
|
}
|
|
|
|
func (a *allocator) countUsedPages() int {
|
|
a.Lock()
|
|
defer a.Unlock()
|
|
|
|
num := 0
|
|
for _, p := range a.used {
|
|
num += len(p)
|
|
}
|
|
return num
|
|
}
|
|
|
|
func (a *allocator) countAvailablePages() int {
|
|
a.Lock()
|
|
defer a.Unlock()
|
|
|
|
return len(a.available)
|
|
}
|
|
|
|
func (a *allocator) isRequestOrderIDUsed(requestOrderID uint32) bool {
|
|
a.Lock()
|
|
defer a.Unlock()
|
|
|
|
_, ok := a.used[requestOrderID]
|
|
return ok
|
|
}
|