(ctx context.Context, sessionID uuid.UUID, clientConn *websocket.Conn)
| 78 | } |
| 79 | |
| 80 | func (a *Agent) CreateShellSession(ctx context.Context, sessionID uuid.UUID, clientConn *websocket.Conn) (*ClientSession, error) { |
| 81 | a.mux.Lock() |
| 82 | defer a.mux.Unlock() |
| 83 | |
| 84 | _, ok := a.shellSessions[sessionID.String()] |
| 85 | if ok { |
| 86 | return nil, runnerErrors.NewConflictError("session ID %q already in use", sessionID) |
| 87 | } |
| 88 | sess, err := NewClientSession(ctx, clientConn, a.writeMessage, sessionID) |
| 89 | if err != nil { |
| 90 | return nil, fmt.Errorf("failed to create new client session: %w", err) |
| 91 | } |
| 92 | |
| 93 | if err := sess.Start(); err != nil { |
| 94 | return nil, fmt.Errorf("failed to start client session: %w", err) |
| 95 | } |
| 96 | |
| 97 | if !a.instance.Capabilities.Shell { |
| 98 | shellDisabled := messaging.ShellReadyMessage{ |
| 99 | SessionID: sessionID, |
| 100 | IsError: 1, |
| 101 | Message: []byte("agent shell is disabled"), |
| 102 | } |
| 103 | sess.safeWrite(websocket.BinaryMessage, shellDisabled.Marshal()) |
| 104 | sess.Stop() |
| 105 | return nil, fmt.Errorf("agent shell is disabled") |
| 106 | } |
| 107 | a.shellSessions[sessionID.String()] = sess |
| 108 | return sess, nil |
| 109 | } |
| 110 | |
| 111 | func (a *Agent) RemoveClientSession(sessionID uuid.UUID, safe bool) error { |
| 112 | if !safe { |
no test coverage detected