mirror of
https://github.com/gitpod-io/gitpod.git
synced 2025-12-08 17:36:30 +00:00
368 lines
10 KiB
Go
368 lines
10 KiB
Go
// Copyright (c) 2020 Gitpod 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 terminal
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/creack/pty"
|
|
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"github.com/gitpod-io/gitpod/common-go/log"
|
|
"github.com/gitpod-io/gitpod/supervisor/api"
|
|
)
|
|
|
|
// NewMuxTerminalService creates a new terminal service.
|
|
func NewMuxTerminalService(m *Mux) *MuxTerminalService {
|
|
shell := os.Getenv("SHELL")
|
|
if shell == "" {
|
|
shell = "/bin/bash"
|
|
}
|
|
return &MuxTerminalService{
|
|
Mux: m,
|
|
DefaultWorkdir: "/workspace",
|
|
DefaultShell: shell,
|
|
Env: os.Environ(),
|
|
}
|
|
}
|
|
|
|
// MuxTerminalService implements the terminal service API using a terminal Mux.
|
|
type MuxTerminalService struct {
|
|
Mux *Mux
|
|
|
|
DefaultWorkdir string
|
|
// DefaultWorkdirProvider allows dynamically to compute workdir
|
|
// if returns empty string then DefaultWorkdir is used
|
|
DefaultWorkdirProvider func() string
|
|
|
|
DefaultShell string
|
|
Env []string
|
|
DefaultCreds *syscall.Credential
|
|
DefaultAmbientCaps []uintptr
|
|
|
|
api.UnimplementedTerminalServiceServer
|
|
}
|
|
|
|
// RegisterGRPC registers a gRPC service.
|
|
func (srv *MuxTerminalService) RegisterGRPC(s *grpc.Server) {
|
|
api.RegisterTerminalServiceServer(s, srv)
|
|
}
|
|
|
|
// RegisterREST registers a REST service.
|
|
func (srv *MuxTerminalService) RegisterREST(mux *runtime.ServeMux, grpcEndpoint string) error {
|
|
return api.RegisterTerminalServiceHandlerFromEndpoint(context.Background(), mux, grpcEndpoint, []grpc.DialOption{grpc.WithTransportCredentials(insecure.NewCredentials())})
|
|
}
|
|
|
|
// Open opens a new terminal running the shell.
|
|
func (srv *MuxTerminalService) Open(ctx context.Context, req *api.OpenTerminalRequest) (*api.OpenTerminalResponse, error) {
|
|
return srv.OpenWithOptions(ctx, req, TermOptions{
|
|
ReadTimeout: 5 * time.Second,
|
|
Annotations: req.Annotations,
|
|
})
|
|
}
|
|
|
|
// OpenWithOptions opens a new terminal running the shell with given options.
|
|
// req.Annotations override options.Annotations.
|
|
func (srv *MuxTerminalService) OpenWithOptions(ctx context.Context, req *api.OpenTerminalRequest, options TermOptions) (*api.OpenTerminalResponse, error) {
|
|
shell := req.Shell
|
|
if shell == "" {
|
|
shell = srv.DefaultShell
|
|
}
|
|
cmd := exec.Command(shell, req.ShellArgs...)
|
|
if srv.DefaultCreds != nil {
|
|
cmd.SysProcAttr = &syscall.SysProcAttr{
|
|
Credential: srv.DefaultCreds,
|
|
}
|
|
}
|
|
if req.Workdir != "" {
|
|
cmd.Dir = req.Workdir
|
|
} else if srv.DefaultWorkdirProvider != nil {
|
|
cmd.Dir = srv.DefaultWorkdirProvider()
|
|
}
|
|
if cmd.Dir == "" {
|
|
cmd.Dir = srv.DefaultWorkdir
|
|
}
|
|
cmd.Env = append(srv.Env, "TERM=xterm-256color")
|
|
for key, value := range req.Env {
|
|
cmd.Env = append(cmd.Env, fmt.Sprintf("%v=%v", key, value))
|
|
}
|
|
for k, v := range req.Annotations {
|
|
options.Annotations[k] = v
|
|
}
|
|
if req.Size != nil {
|
|
options.Size = &pty.Winsize{
|
|
Cols: uint16(req.Size.Cols),
|
|
Rows: uint16(req.Size.Rows),
|
|
X: uint16(req.Size.WidthPx),
|
|
Y: uint16(req.Size.HeightPx),
|
|
}
|
|
}
|
|
|
|
if srv.DefaultAmbientCaps != nil {
|
|
if cmd.SysProcAttr == nil {
|
|
cmd.SysProcAttr = &syscall.SysProcAttr{}
|
|
}
|
|
cmd.SysProcAttr.AmbientCaps = srv.DefaultAmbientCaps
|
|
}
|
|
|
|
alias, err := srv.Mux.Start(cmd, options)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
|
|
// starterToken is just relevant for the service, hence it's not exposed at the Start() call
|
|
var starterToken string
|
|
term := srv.Mux.terms[alias]
|
|
if term != nil {
|
|
starterToken = term.StarterToken
|
|
}
|
|
|
|
terminal, found := srv.get(alias)
|
|
if !found {
|
|
return nil, status.Error(codes.NotFound, "terminal not found")
|
|
}
|
|
return &api.OpenTerminalResponse{
|
|
Terminal: terminal,
|
|
StarterToken: starterToken,
|
|
}, nil
|
|
}
|
|
|
|
// Close closes a terminal for the given alias.
|
|
func (srv *MuxTerminalService) Shutdown(ctx context.Context, req *api.ShutdownTerminalRequest) (*api.ShutdownTerminalResponse, error) {
|
|
err := srv.Mux.CloseTerminal(ctx, req.Alias, req.ForceSuccess)
|
|
if err == ErrNotFound {
|
|
return nil, status.Error(codes.NotFound, err.Error())
|
|
}
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
return &api.ShutdownTerminalResponse{}, nil
|
|
}
|
|
|
|
// List lists all open terminals.
|
|
func (srv *MuxTerminalService) List(ctx context.Context, req *api.ListTerminalsRequest) (*api.ListTerminalsResponse, error) {
|
|
srv.Mux.mu.RLock()
|
|
defer srv.Mux.mu.RUnlock()
|
|
|
|
res := make([]*api.Terminal, 0, len(srv.Mux.terms))
|
|
for _, alias := range srv.Mux.aliases {
|
|
term, ok := srv.get(alias)
|
|
if !ok {
|
|
continue
|
|
}
|
|
res = append(res, term)
|
|
}
|
|
|
|
return &api.ListTerminalsResponse{
|
|
Terminals: res,
|
|
}, nil
|
|
}
|
|
|
|
// Get returns an open terminal info.
|
|
func (srv *MuxTerminalService) Get(ctx context.Context, req *api.GetTerminalRequest) (*api.Terminal, error) {
|
|
srv.Mux.mu.RLock()
|
|
defer srv.Mux.mu.RUnlock()
|
|
term, ok := srv.get(req.Alias)
|
|
if !ok {
|
|
return nil, status.Error(codes.NotFound, "terminal not found")
|
|
}
|
|
return term, nil
|
|
}
|
|
|
|
func (srv *MuxTerminalService) get(alias string) (*api.Terminal, bool) {
|
|
term, ok := srv.Mux.terms[alias]
|
|
if !ok {
|
|
return nil, false
|
|
}
|
|
|
|
var (
|
|
pid int64
|
|
cwd string
|
|
err error
|
|
)
|
|
if proc := term.Command.Process; proc != nil {
|
|
pid = int64(proc.Pid)
|
|
cwd, err = filepath.EvalSymlinks(fmt.Sprintf("/proc/%d/cwd", pid))
|
|
if err != nil {
|
|
log.WithError(err).WithField("pid", pid).Warn("unable to resolve terminal's current working dir")
|
|
cwd = term.Command.Dir
|
|
}
|
|
}
|
|
|
|
title, titleSource, err := term.GetTitle()
|
|
if err != nil {
|
|
log.WithError(err).WithField("pid", pid).Warn("unable to resolve terminal's title")
|
|
}
|
|
|
|
return &api.Terminal{
|
|
Alias: alias,
|
|
Command: term.Command.Args,
|
|
Pid: pid,
|
|
InitialWorkdir: term.Command.Dir,
|
|
CurrentWorkdir: cwd,
|
|
Annotations: term.GetAnnotations(),
|
|
Title: title,
|
|
TitleSource: titleSource,
|
|
}, true
|
|
}
|
|
|
|
// Listen listens to a terminal.
|
|
func (srv *MuxTerminalService) Listen(req *api.ListenTerminalRequest, resp api.TerminalService_ListenServer) error {
|
|
srv.Mux.mu.RLock()
|
|
term, ok := srv.Mux.terms[req.Alias]
|
|
srv.Mux.mu.RUnlock()
|
|
if !ok {
|
|
return status.Error(codes.NotFound, "terminal not found")
|
|
}
|
|
stdout := term.Stdout.Listen()
|
|
defer stdout.Close()
|
|
|
|
log.WithField("alias", req.Alias).Info("new terminal client")
|
|
defer log.WithField("alias", req.Alias).Info("terminal client left")
|
|
|
|
errchan := make(chan error, 1)
|
|
messages := make(chan *api.ListenTerminalResponse, 1)
|
|
go func() {
|
|
for {
|
|
buf := make([]byte, 4096)
|
|
n, err := stdout.Read(buf)
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
errchan <- err
|
|
return
|
|
}
|
|
messages <- &api.ListenTerminalResponse{Output: &api.ListenTerminalResponse_Data{Data: buf[:n]}}
|
|
}
|
|
|
|
state, err := term.Wait()
|
|
if err != nil {
|
|
errchan <- err
|
|
return
|
|
}
|
|
|
|
messages <- &api.ListenTerminalResponse{Output: &api.ListenTerminalResponse_ExitCode{ExitCode: int32(state.ExitCode())}}
|
|
errchan <- io.EOF
|
|
}()
|
|
go func() {
|
|
title, titleSource, _ := term.GetTitle()
|
|
messages <- &api.ListenTerminalResponse{Output: &api.ListenTerminalResponse_Title{Title: title}, TitleSource: titleSource}
|
|
|
|
t := time.NewTicker(200 * time.Millisecond)
|
|
defer t.Stop()
|
|
for {
|
|
select {
|
|
case <-resp.Context().Done():
|
|
return
|
|
case <-t.C:
|
|
newTitle, newTitleSource, _ := term.GetTitle()
|
|
if title == newTitle && titleSource == newTitleSource {
|
|
continue
|
|
}
|
|
title = newTitle
|
|
titleSource = newTitleSource
|
|
messages <- &api.ListenTerminalResponse{Output: &api.ListenTerminalResponse_Title{Title: title}, TitleSource: titleSource}
|
|
}
|
|
}
|
|
}()
|
|
for {
|
|
var err error
|
|
select {
|
|
case message := <-messages:
|
|
err = resp.Send(message)
|
|
case err = <-errchan:
|
|
case <-resp.Context().Done():
|
|
return status.Error(codes.DeadlineExceeded, resp.Context().Err().Error())
|
|
}
|
|
if err == io.EOF {
|
|
// EOF isn't really an error here
|
|
return nil
|
|
}
|
|
if err != nil {
|
|
return status.Error(codes.Internal, err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
// Write writes to a terminal.
|
|
func (srv *MuxTerminalService) Write(ctx context.Context, req *api.WriteTerminalRequest) (*api.WriteTerminalResponse, error) {
|
|
srv.Mux.mu.RLock()
|
|
term, ok := srv.Mux.terms[req.Alias]
|
|
srv.Mux.mu.RUnlock()
|
|
if !ok {
|
|
return nil, status.Error(codes.NotFound, "terminal not found")
|
|
}
|
|
|
|
n, err := term.PTY.Write(req.Stdin)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
return &api.WriteTerminalResponse{BytesWritten: uint32(n)}, nil
|
|
}
|
|
|
|
// SetSize sets the terminal's size.
|
|
func (srv *MuxTerminalService) SetSize(ctx context.Context, req *api.SetTerminalSizeRequest) (*api.SetTerminalSizeResponse, error) {
|
|
srv.Mux.mu.RLock()
|
|
term, ok := srv.Mux.terms[req.Alias]
|
|
srv.Mux.mu.RUnlock()
|
|
if !ok {
|
|
return nil, status.Error(codes.NotFound, "terminal not found")
|
|
}
|
|
|
|
// Setting the size only works with the starter token or when forcing it.
|
|
// This protects us from multiple listener mangling the terminal.
|
|
if !(req.GetForce() || req.GetToken() == term.StarterToken) {
|
|
return nil, status.Error(codes.FailedPrecondition, "wrong token or force not set")
|
|
}
|
|
|
|
err := pty.Setsize(term.PTY, &pty.Winsize{
|
|
Cols: uint16(req.Size.Cols),
|
|
Rows: uint16(req.Size.Rows),
|
|
X: uint16(req.Size.WidthPx),
|
|
Y: uint16(req.Size.HeightPx),
|
|
})
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, err.Error())
|
|
}
|
|
|
|
return &api.SetTerminalSizeResponse{}, nil
|
|
}
|
|
|
|
// SetTitle sets the terminal's title.
|
|
func (srv *MuxTerminalService) SetTitle(ctx context.Context, req *api.SetTerminalTitleRequest) (*api.SetTerminalTitleResponse, error) {
|
|
srv.Mux.mu.RLock()
|
|
term, ok := srv.Mux.terms[req.Alias]
|
|
srv.Mux.mu.RUnlock()
|
|
if !ok {
|
|
return nil, status.Error(codes.NotFound, "terminal not found")
|
|
}
|
|
term.SetTitle(req.Title)
|
|
return &api.SetTerminalTitleResponse{}, nil
|
|
}
|
|
|
|
// UpdateAnnotations sets the terminal's title.
|
|
func (srv *MuxTerminalService) UpdateAnnotations(ctx context.Context, req *api.UpdateTerminalAnnotationsRequest) (*api.UpdateTerminalAnnotationsResponse, error) {
|
|
srv.Mux.mu.RLock()
|
|
term, ok := srv.Mux.terms[req.Alias]
|
|
srv.Mux.mu.RUnlock()
|
|
if !ok {
|
|
return nil, status.Error(codes.NotFound, "terminal not found")
|
|
}
|
|
term.UpdateAnnotations(req.Changed, req.Deleted)
|
|
return &api.UpdateTerminalAnnotationsResponse{}, nil
|
|
}
|