mirror of
https://github.com/gitpod-io/gitpod.git
synced 2025-12-08 17:36:30 +00:00
85 lines
2.2 KiB
Go
85 lines
2.2 KiB
Go
// Copyright (c) 2020 TypeFox GmbH. All rights reserved.
|
|
// Licensed under the GNU Affero General Public License (AGPL).
|
|
// See License-AGPL.txt in the project root for license information.
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gitpod-io/gitpod/supervisor/api"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var newuidmapCmd = &cobra.Command{
|
|
Use: "newuidmap <pid> <inContainerID> <hostID> <size> ... [<inContainerID> <hostID> <size>]",
|
|
Short: "establishes a new UID mapping for a user-namespace",
|
|
Hidden: true,
|
|
Args: cobra.MinimumNArgs(4),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return setIDMapping(args, false)
|
|
},
|
|
}
|
|
|
|
var newgidmapCmd = &cobra.Command{
|
|
Use: "newgidmap <pid> <inContainerID> <hostID> <size> ... [<inContainerID> <hostID> <size>]",
|
|
Short: "establishes a new GID mapping for a user-namespace",
|
|
Hidden: true,
|
|
Args: cobra.MinimumNArgs(4),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return setIDMapping(args, true)
|
|
},
|
|
}
|
|
|
|
func setIDMapping(args []string, gid bool) error {
|
|
conn := dialSupervisor()
|
|
client := api.NewControlServiceClient(conn)
|
|
|
|
pid, err := strconv.ParseInt(args[0], 10, 64)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot parse PID: %w", err)
|
|
}
|
|
|
|
mapping := make([]*api.NewuidmapRequest_Mapping, 0, (len(args)-1)/3)
|
|
for i := 1; i < len(args); i++ {
|
|
icid, err := strconv.ParseUint(args[i], 10, 32)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot parse inContainerID (arg %d): %w", i, err)
|
|
}
|
|
i++
|
|
|
|
hid, err := strconv.ParseUint(args[i], 10, 32)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot parse inContainerID (arg %d): %w", i, err)
|
|
}
|
|
i++
|
|
|
|
sze, err := strconv.ParseUint(args[i], 10, 32)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot parse inContainerID (arg %d): %w", i, err)
|
|
}
|
|
|
|
mapping = append(mapping, &api.NewuidmapRequest_Mapping{
|
|
ContainerId: uint32(icid),
|
|
HostId: uint32(hid),
|
|
Size: uint32(sze),
|
|
})
|
|
}
|
|
|
|
if (len(args)-1)%3 != 0 {
|
|
return fmt.Errorf("arguments must be tripples")
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
_, err = client.Newuidmap(ctx, &api.NewuidmapRequest{
|
|
Pid: pid,
|
|
Gid: gid,
|
|
Mapping: mapping,
|
|
})
|
|
return err
|
|
}
|