pptx704
a3898d68fb
- Copy envd source from e2b-dev/infra, internalize shared dependencies
into envd/internal/shared/ (keys, filesystem, id, smap, utils)
- Switch from gRPC to Connect RPC for all envd services
- Update module paths to git.omukk.dev/wrenn/{sandbox,sandbox/envd}
- Add proto specs (process, filesystem) with buf-based code generation
- Implement full envd: process exec, filesystem ops, port forwarding,
cgroup management, MMDS integration, and HTTP API
- Update main module dependencies (firecracker SDK, pgx, goose, etc.)
- Remove placeholder .gitkeep files replaced by real implementations
85 lines
2.2 KiB
Go
85 lines
2.2 KiB
Go
package process
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"connectrpc.com/connect"
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/rs/zerolog"
|
|
|
|
"git.omukk.dev/wrenn/sandbox/envd/internal/execcontext"
|
|
"git.omukk.dev/wrenn/sandbox/envd/internal/logs"
|
|
"git.omukk.dev/wrenn/sandbox/envd/internal/services/cgroups"
|
|
"git.omukk.dev/wrenn/sandbox/envd/internal/services/process/handler"
|
|
rpc "git.omukk.dev/wrenn/sandbox/envd/internal/services/spec/process"
|
|
spec "git.omukk.dev/wrenn/sandbox/envd/internal/services/spec/process/processconnect"
|
|
"git.omukk.dev/wrenn/sandbox/envd/internal/utils"
|
|
)
|
|
|
|
type Service struct {
|
|
processes *utils.Map[uint32, *handler.Handler]
|
|
logger *zerolog.Logger
|
|
defaults *execcontext.Defaults
|
|
cgroupManager cgroups.Manager
|
|
}
|
|
|
|
func newService(l *zerolog.Logger, defaults *execcontext.Defaults, cgroupManager cgroups.Manager) *Service {
|
|
return &Service{
|
|
logger: l,
|
|
processes: utils.NewMap[uint32, *handler.Handler](),
|
|
defaults: defaults,
|
|
cgroupManager: cgroupManager,
|
|
}
|
|
}
|
|
|
|
func Handle(server *chi.Mux, l *zerolog.Logger, defaults *execcontext.Defaults, cgroupManager cgroups.Manager) *Service {
|
|
service := newService(l, defaults, cgroupManager)
|
|
|
|
interceptors := connect.WithInterceptors(logs.NewUnaryLogInterceptor(l))
|
|
|
|
path, h := spec.NewProcessHandler(service, interceptors)
|
|
|
|
server.Mount(path, h)
|
|
|
|
return service
|
|
}
|
|
|
|
func (s *Service) getProcess(selector *rpc.ProcessSelector) (*handler.Handler, error) {
|
|
var proc *handler.Handler
|
|
|
|
switch selector.GetSelector().(type) {
|
|
case *rpc.ProcessSelector_Pid:
|
|
p, ok := s.processes.Load(selector.GetPid())
|
|
if !ok {
|
|
return nil, connect.NewError(connect.CodeNotFound, fmt.Errorf("process with pid %d not found", selector.GetPid()))
|
|
}
|
|
|
|
proc = p
|
|
case *rpc.ProcessSelector_Tag:
|
|
tag := selector.GetTag()
|
|
|
|
s.processes.Range(func(_ uint32, value *handler.Handler) bool {
|
|
if value.Tag == nil {
|
|
return true
|
|
}
|
|
|
|
if *value.Tag == tag {
|
|
proc = value
|
|
|
|
return true
|
|
}
|
|
|
|
return false
|
|
})
|
|
|
|
if proc == nil {
|
|
return nil, connect.NewError(connect.CodeNotFound, fmt.Errorf("process with tag %s not found", tag))
|
|
}
|
|
|
|
default:
|
|
return nil, connect.NewError(connect.CodeUnimplemented, fmt.Errorf("invalid input type %T", selector))
|
|
}
|
|
|
|
return proc, nil
|
|
}
|