waveterm/pkg/sstore/sstore.go

1008 lines
29 KiB
Go
Raw Normal View History

2022-06-13 20:11:56 +02:00
package sstore
import (
2022-07-01 21:17:19 +02:00
"context"
"crypto/ecdsa"
"crypto/elliptic"
"crypto/rand"
"crypto/x509"
"database/sql/driver"
"fmt"
2022-07-01 23:07:13 +02:00
"log"
2022-08-17 00:08:28 +02:00
"os"
"os/user"
2022-07-01 02:02:19 +02:00
"path"
"regexp"
"strings"
2022-06-21 06:57:23 +02:00
"sync"
2022-06-13 20:11:56 +02:00
"time"
"github.com/google/uuid"
"github.com/jmoiron/sqlx"
2022-07-01 21:17:19 +02:00
"github.com/scripthaus-dev/mshell/pkg/base"
"github.com/scripthaus-dev/mshell/pkg/packet"
2022-07-01 02:02:19 +02:00
"github.com/scripthaus-dev/sh2-server/pkg/scbase"
_ "github.com/mattn/go-sqlite3"
2022-06-13 20:11:56 +02:00
)
const LineTypeCmd = "cmd"
const LineTypeText = "text"
const LineNoHeight = -1
const DBFileName = "prompt.db"
2022-07-01 02:02:19 +02:00
2022-07-01 21:17:19 +02:00
const DefaultSessionName = "default"
const DefaultWindowName = "default"
const LocalRemoteAlias = "local"
const DefaultScreenWindowName = "w1"
2022-07-01 21:17:19 +02:00
2022-07-01 23:07:13 +02:00
const DefaultCwd = "~"
const (
CmdStatusRunning = "running"
CmdStatusDetached = "detached"
CmdStatusError = "error"
CmdStatusDone = "done"
CmdStatusHangup = "hangup"
CmdStatusWaiting = "waiting"
)
const (
ShareModeLocal = "local"
ShareModePrivate = "private"
ShareModeView = "view"
ShareModeShared = "shared"
)
const (
ConnectModeStartup = "startup"
ConnectModeAuto = "auto"
ConnectModeManual = "manual"
)
const (
RemoteTypeSsh = "ssh"
)
2022-10-11 10:11:04 +02:00
const (
SWFocusInput = "input"
SWFocusCmd = "cmd"
SWFocusCmdFg = "cmd-fg"
)
2023-01-17 08:36:52 +01:00
const MaxTzNameLen = 50
2022-07-01 21:17:19 +02:00
var globalDBLock = &sync.Mutex{}
var globalDB *sqlx.DB
var globalDBErr error
func GetSessionDBName() string {
scHome := scbase.GetPromptHomeDir()
2022-07-01 02:02:19 +02:00
return path.Join(scHome, DBFileName)
}
2022-06-13 20:11:56 +02:00
func IsValidConnectMode(mode string) bool {
2022-09-14 02:11:36 +02:00
return mode == ConnectModeStartup || mode == ConnectModeAuto || mode == ConnectModeManual
}
func GetDB(ctx context.Context) (*sqlx.DB, error) {
if IsTxWrapContext(ctx) {
return nil, fmt.Errorf("cannot call GetDB from within a running transaction")
}
2022-07-01 21:17:19 +02:00
globalDBLock.Lock()
defer globalDBLock.Unlock()
if globalDB == nil && globalDBErr == nil {
dbName := GetSessionDBName()
globalDB, globalDBErr = sqlx.Open("sqlite3", fmt.Sprintf("file:%s?cache=shared&mode=rwc&_journal_mode=WAL&_busy_timeout=5000", dbName))
if globalDBErr != nil {
globalDBErr = fmt.Errorf("opening db[%s]: %w", dbName, globalDBErr)
log.Printf("[db] error: %v\n", globalDBErr)
} else {
log.Printf("[db] successfully opened db %s\n", dbName)
}
}
2022-07-01 21:17:19 +02:00
return globalDB, globalDBErr
}
2022-09-25 09:26:33 +02:00
type ClientWinSizeType struct {
Width int `json:"width"`
Height int `json:"height"`
Top int `json:"top"`
Left int `json:"left"`
FullScreen bool `json:"fullscreen,omitempty"`
}
2023-01-17 08:36:52 +01:00
type ActivityUpdate struct {
FgMinutes int
ActiveMinutes int
OpenMinutes int
NumLines int
}
type ActivityType struct {
Day string `json:"day"`
Uploaded bool `json:"-"`
NumLines int `json:"numlines"`
ActiveMinutes int `json:"activeminutes"`
FgMinutes int `json:"fgminutes"`
OpenMinutes int `json:"openminutes"`
TzName string `json:"tzname"`
TzOffset int `json:"tzoffset"`
ClientVersion string `json:"clientversion"`
ClientArch string `json:"clientarch"`
}
type ClientData struct {
2022-09-25 09:26:33 +02:00
ClientId string `json:"clientid"`
UserId string `json:"userid"`
UserPrivateKeyBytes []byte `json:"-"`
UserPublicKeyBytes []byte `json:"-"`
UserPrivateKey *ecdsa.PrivateKey `json:"-"`
UserPublicKey *ecdsa.PublicKey `json:"-"`
ActiveSessionId string `json:"activesessionid"`
WinSize ClientWinSizeType `json:"winsize"`
}
2022-09-22 07:02:38 +02:00
func (c *ClientData) ToMap() map[string]interface{} {
rtn := make(map[string]interface{})
rtn["clientid"] = c.ClientId
rtn["userid"] = c.UserId
rtn["userprivatekeybytes"] = c.UserPrivateKeyBytes
rtn["userpublickeybytes"] = c.UserPublicKeyBytes
rtn["activesessionid"] = c.ActiveSessionId
2022-09-25 09:26:33 +02:00
rtn["winsize"] = quickJson(c.WinSize)
2022-09-22 07:02:38 +02:00
return rtn
}
func ClientDataFromMap(m map[string]interface{}) *ClientData {
if len(m) == 0 {
return nil
}
var c ClientData
quickSetStr(&c.ClientId, m, "clientid")
quickSetStr(&c.UserId, m, "userid")
quickSetBytes(&c.UserPrivateKeyBytes, m, "userprivatekeybytes")
quickSetBytes(&c.UserPublicKeyBytes, m, "userpublickeybytes")
quickSetStr(&c.ActiveSessionId, m, "activesessionid")
2022-09-25 09:26:33 +02:00
quickSetJson(&c.WinSize, m, "winsize")
2022-09-22 07:02:38 +02:00
return &c
}
2022-06-21 06:57:23 +02:00
type SessionType struct {
SessionId string `json:"sessionid"`
Name string `json:"name"`
SessionIdx int64 `json:"sessionidx"`
ActiveScreenId string `json:"activescreenid"`
2022-08-24 22:21:54 +02:00
OwnerId string `json:"ownerid"`
ShareMode string `json:"sharemode"`
AccessKey string `json:"-"`
NotifyNum int64 `json:"notifynum"`
2022-12-25 22:03:11 +01:00
Archived bool `json:"archived,omitempty"`
2022-12-25 22:21:48 +01:00
ArchivedTs int64 `json:"archivedts,omitempty"`
Screens []*ScreenType `json:"screens"`
Remotes []*RemoteInstance `json:"remotes"`
2022-07-15 03:39:40 +02:00
// only for updates
Remove bool `json:"remove,omitempty"`
Full bool `json:"full,omitempty"`
2022-07-12 23:27:16 +02:00
}
2022-09-20 23:15:20 +02:00
type SessionStatsType struct {
2022-12-25 22:03:11 +01:00
SessionId string `json:"sessionid"`
NumScreens int `json:"numscreens"`
NumArchivedScreens int `json:"numarchivedscreens"`
NumWindows int `json:"numwindows"`
NumLines int `json:"numlines"`
NumCmds int `json:"numcmds"`
DiskStats SessionDiskSizeType `json:"diskstats"`
2022-09-20 23:15:20 +02:00
}
2022-07-12 23:27:16 +02:00
type WindowOptsType struct {
2022-11-28 09:13:00 +01:00
PTerm string `json:"pterm,omitempty"`
2022-07-12 23:27:16 +02:00
}
func (opts *WindowOptsType) Scan(val interface{}) error {
return quickScanJson(opts, val)
}
func (opts WindowOptsType) Value() (driver.Value, error) {
2022-07-12 23:27:16 +02:00
return quickValueJson(opts)
2022-07-05 07:18:01 +02:00
}
type WindowShareOptsType struct {
}
func (opts *WindowShareOptsType) Scan(val interface{}) error {
return quickScanJson(opts, val)
}
func (opts WindowShareOptsType) Value() (driver.Value, error) {
return quickValueJson(opts)
}
var RemoteNameRe = regexp.MustCompile("^\\*?[a-zA-Z0-9_-]+$")
type RemotePtrType struct {
2022-08-24 22:21:54 +02:00
OwnerId string `json:"ownerid"`
RemoteId string `json:"remoteid"`
Name string `json:"name"`
}
func (r RemotePtrType) IsSessionScope() bool {
return strings.HasPrefix(r.Name, "*")
}
2022-12-31 02:01:17 +01:00
func (rptr *RemotePtrType) GetDisplayName(baseDisplayName string) string {
name := baseDisplayName
if rptr == nil {
return name
}
if rptr.Name != "" {
name = name + ":" + rptr.Name
}
if rptr.OwnerId != "" {
name = "@" + rptr.OwnerId + ":" + name
}
return name
}
func (r RemotePtrType) Validate() error {
if r.OwnerId != "" {
if _, err := uuid.Parse(r.OwnerId); err != nil {
return fmt.Errorf("invalid ownerid format: %v", err)
}
}
if r.RemoteId != "" {
if _, err := uuid.Parse(r.RemoteId); err != nil {
return fmt.Errorf("invalid remoteid format: %v", err)
}
}
if r.Name != "" {
ok := RemoteNameRe.MatchString(r.Name)
if !ok {
return fmt.Errorf("invalid remote name")
}
}
return nil
}
func (r RemotePtrType) MakeFullRemoteRef() string {
if r.RemoteId == "" {
return ""
}
2022-08-24 22:21:54 +02:00
if r.OwnerId == "" && r.Name == "" {
return r.RemoteId
}
2022-08-24 22:21:54 +02:00
if r.OwnerId != "" && r.Name == "" {
return fmt.Sprintf("@%s:%s", r.OwnerId, r.RemoteId)
}
2022-08-24 22:21:54 +02:00
if r.OwnerId == "" && r.Name != "" {
return fmt.Sprintf("%s:%s", r.RemoteId, r.Name)
}
2022-08-24 22:21:54 +02:00
return fmt.Sprintf("@%s:%s:%s", r.OwnerId, r.RemoteId, r.Name)
}
2022-07-05 07:18:01 +02:00
type WindowType struct {
SessionId string `json:"sessionid"`
WindowId string `json:"windowid"`
CurRemote RemotePtrType `json:"curremote"`
WinOpts WindowOptsType `json:"winopts"`
OwnerId string `json:"ownerid"`
NextLineNum int64 `json:"nextlinenum"`
ShareMode string `json:"sharemode"`
ShareOpts WindowShareOptsType `json:"shareopts"`
Lines []*LineType `json:"lines"`
Cmds []*CmdType `json:"cmds"`
2022-07-15 03:39:40 +02:00
// only for updates
Remove bool `json:"remove,omitempty"`
}
func (w *WindowType) ToMap() map[string]interface{} {
rtn := make(map[string]interface{})
rtn["sessionid"] = w.SessionId
rtn["windowid"] = w.WindowId
2022-08-24 22:21:54 +02:00
rtn["curremoteownerid"] = w.CurRemote.OwnerId
rtn["curremoteid"] = w.CurRemote.RemoteId
rtn["curremotename"] = w.CurRemote.Name
rtn["nextlinenum"] = w.NextLineNum
rtn["winopts"] = quickJson(w.WinOpts)
2022-08-24 22:21:54 +02:00
rtn["ownerid"] = w.OwnerId
rtn["sharemode"] = w.ShareMode
rtn["shareopts"] = quickJson(w.ShareOpts)
return rtn
}
func WindowFromMap(m map[string]interface{}) *WindowType {
if len(m) == 0 {
return nil
}
var w WindowType
quickSetStr(&w.SessionId, m, "sessionid")
quickSetStr(&w.WindowId, m, "windowid")
2022-08-24 22:21:54 +02:00
quickSetStr(&w.CurRemote.OwnerId, m, "curremoteownerid")
quickSetStr(&w.CurRemote.RemoteId, m, "curremoteid")
quickSetStr(&w.CurRemote.Name, m, "curremotename")
quickSetInt64(&w.NextLineNum, m, "nextlinenum")
quickSetJson(&w.WinOpts, m, "winopts")
2022-08-24 22:21:54 +02:00
quickSetStr(&w.OwnerId, m, "ownerid")
quickSetStr(&w.ShareMode, m, "sharemode")
quickSetJson(&w.ShareOpts, m, "shareopts")
return &w
}
func (h *HistoryItemType) ToMap() map[string]interface{} {
rtn := make(map[string]interface{})
rtn["historyid"] = h.HistoryId
rtn["ts"] = h.Ts
rtn["userid"] = h.UserId
rtn["sessionid"] = h.SessionId
rtn["screenid"] = h.ScreenId
rtn["windowid"] = h.WindowId
rtn["lineid"] = h.LineId
rtn["haderror"] = h.HadError
rtn["cmdid"] = h.CmdId
rtn["cmdstr"] = h.CmdStr
rtn["remoteownerid"] = h.Remote.OwnerId
rtn["remoteid"] = h.Remote.RemoteId
rtn["remotename"] = h.Remote.Name
rtn["ismetacmd"] = h.IsMetaCmd
rtn["incognito"] = h.Incognito
return rtn
}
func HistoryItemFromMap(m map[string]interface{}) *HistoryItemType {
if len(m) == 0 {
return nil
}
var h HistoryItemType
quickSetStr(&h.HistoryId, m, "historyid")
quickSetInt64(&h.Ts, m, "ts")
quickSetStr(&h.UserId, m, "userid")
quickSetStr(&h.SessionId, m, "sessionid")
quickSetStr(&h.ScreenId, m, "screenid")
quickSetStr(&h.WindowId, m, "windowid")
quickSetStr(&h.LineId, m, "lineid")
quickSetBool(&h.HadError, m, "haderror")
quickSetStr(&h.CmdId, m, "cmdid")
quickSetStr(&h.CmdStr, m, "cmdstr")
quickSetStr(&h.Remote.OwnerId, m, "remoteownerid")
quickSetStr(&h.Remote.RemoteId, m, "remoteid")
quickSetStr(&h.Remote.Name, m, "remotename")
quickSetBool(&h.IsMetaCmd, m, "ismetacmd")
2022-08-30 04:18:02 +02:00
quickSetStr(&h.HistoryNum, m, "historynum")
quickSetBool(&h.Incognito, m, "incognito")
return &h
}
type ScreenOptsType struct {
2022-08-27 06:44:18 +02:00
TabColor string `json:"tabcolor,omitempty"`
}
func (opts *ScreenOptsType) Scan(val interface{}) error {
return quickScanJson(opts, val)
}
func (opts ScreenOptsType) Value() (driver.Value, error) {
return quickValueJson(opts)
2022-07-12 23:27:16 +02:00
}
type ScreenType struct {
SessionId string `json:"sessionid"`
ScreenId string `json:"screenid"`
ScreenIdx int64 `json:"screenidx"`
Name string `json:"name"`
ActiveWindowId string `json:"activewindowid"`
2022-08-27 02:51:28 +02:00
ScreenOpts *ScreenOptsType `json:"screenopts"`
2022-08-24 22:21:54 +02:00
OwnerId string `json:"ownerid"`
ShareMode string `json:"sharemode"`
Incognito bool `json:"incognito,omitempty"`
2022-12-25 22:03:11 +01:00
Archived bool `json:"archived,omitempty"`
2022-12-25 22:21:48 +01:00
ArchivedTs int64 `json:"archivedts,omitempty"`
Windows []*ScreenWindowType `json:"windows"`
2022-07-15 03:39:40 +02:00
// only for updates
Remove bool `json:"remove,omitempty"`
Full bool `json:"full,omitempty"`
2022-07-12 23:27:16 +02:00
}
const (
LayoutFull = "full"
)
2022-07-12 23:27:16 +02:00
type LayoutType struct {
Type string `json:"type"`
Parent string `json:"parent,omitempty"`
ZIndex int64 `json:"zindex,omitempty"`
Float bool `json:"float,omitempty"`
Top string `json:"top,omitempty"`
Bottom string `json:"bottom,omitempty"`
Left string `json:"left,omitempty"`
Right string `json:"right,omitempty"`
Width string `json:"width,omitempty"`
Height string `json:"height,omitempty"`
2022-07-12 23:27:16 +02:00
}
func (l *LayoutType) Scan(val interface{}) error {
return quickScanJson(l, val)
}
func (l LayoutType) Value() (driver.Value, error) {
2022-07-12 23:27:16 +02:00
return quickValueJson(l)
}
2022-10-11 10:11:04 +02:00
type SWAnchorType struct {
AnchorLine int `json:"anchorline,omitempty"`
AnchorOffset int `json:"anchoroffset,omitempty"`
}
func (a *SWAnchorType) Scan(val interface{}) error {
return quickScanJson(a, val)
}
func (a SWAnchorType) Value() (driver.Value, error) {
return quickValueJson(a)
}
type SWKey struct {
SessionId string
ScreenId string
WindowId string
}
2022-07-12 23:27:16 +02:00
type ScreenWindowType struct {
2022-10-11 10:11:04 +02:00
SessionId string `json:"sessionid"`
ScreenId string `json:"screenid"`
WindowId string `json:"windowid"`
Name string `json:"name"`
Layout LayoutType `json:"layout"`
SelectedLine int `json:"selectedline"`
Anchor SWAnchorType `json:"anchor"`
FocusType string `json:"focustype"`
2022-07-15 03:39:40 +02:00
// only for updates
Remove bool `json:"remove,omitempty"`
2022-07-12 22:50:44 +02:00
}
type HistoryItemType struct {
HistoryId string `json:"historyid"`
Ts int64 `json:"ts"`
UserId string `json:"userid"`
SessionId string `json:"sessionid"`
ScreenId string `json:"screenid"`
WindowId string `json:"windowid"`
LineId string `json:"lineid"`
HadError bool `json:"haderror"`
CmdId string `json:"cmdid"`
CmdStr string `json:"cmdstr"`
Remote RemotePtrType `json:"remote"`
IsMetaCmd bool `json:"ismetacmd"`
Incognito bool `json:"incognito,omitempty"`
2022-08-11 21:07:41 +02:00
// only for updates
Remove bool `json:"remove"`
// transient (string because of different history orderings)
HistoryNum string `json:"historynum"`
}
2022-08-30 04:18:02 +02:00
type HistoryQueryOpts struct {
MaxItems int
FromTs int64
}
2022-07-07 09:10:37 +02:00
type TermOpts struct {
Rows int64 `json:"rows"`
Cols int64 `json:"cols"`
FlexRows bool `json:"flexrows,omitempty"`
MaxPtySize int64 `json:"maxptysize,omitempty"`
2022-07-07 09:10:37 +02:00
}
func (opts *TermOpts) Scan(val interface{}) error {
2022-07-12 23:27:16 +02:00
return quickScanJson(opts, val)
2022-07-07 09:10:37 +02:00
}
func (opts TermOpts) Value() (driver.Value, error) {
2022-07-12 23:27:16 +02:00
return quickValueJson(opts)
2022-07-07 09:10:37 +02:00
}
type ShellStatePtr struct {
BaseHash string
DiffHashArr []string
}
func (ssptr *ShellStatePtr) IsEmpty() bool {
if ssptr == nil || ssptr.BaseHash == "" {
return true
}
return false
}
type RemoteInstance struct {
2022-11-28 09:13:00 +01:00
RIId string `json:"riid"`
Name string `json:"name"`
SessionId string `json:"sessionid"`
WindowId string `json:"windowid"`
RemoteOwnerId string `json:"remoteownerid"`
RemoteId string `json:"remoteid"`
FeState FeStateType `json:"festate"`
StateBaseHash string `json:"-"`
StateDiffHashArr []string `json:"-"`
2022-08-11 03:33:32 +02:00
// only for updates
Remove bool `json:"remove,omitempty"`
2022-06-21 06:57:23 +02:00
}
2022-11-28 09:13:00 +01:00
type StateBase struct {
BaseHash string
Version string
Ts int64
Data []byte
}
type StateDiff struct {
DiffHash string
Ts int64
BaseHash string
DiffHashArr []string
Data []byte
}
func StateDiffFromMap(m map[string]interface{}) *StateDiff {
if len(m) == 0 {
return nil
}
var sd StateDiff
quickSetStr(&sd.DiffHash, m, "diffhash")
quickSetInt64(&sd.Ts, m, "ts")
quickSetStr(&sd.BaseHash, m, "basehash")
quickSetJsonArr(&sd.DiffHashArr, m, "diffhasharr")
quickSetBytes(&sd.Data, m, "data")
return &sd
}
func (sd *StateDiff) ToMap() map[string]interface{} {
rtn := make(map[string]interface{})
2022-11-28 09:13:00 +01:00
rtn["diffhash"] = sd.DiffHash
rtn["ts"] = sd.Ts
rtn["basehash"] = sd.BaseHash
rtn["diffhasharr"] = quickJsonArr(sd.DiffHashArr)
rtn["data"] = sd.Data
return rtn
}
2022-11-28 09:13:00 +01:00
type FeStateType struct {
Cwd string `json:"cwd"`
// maybe later we can add some vars
}
func FeStateFromShellState(state *packet.ShellState) *FeStateType {
if state == nil {
return nil
}
return &FeStateType{Cwd: state.Cwd}
}
func RIFromMap(m map[string]interface{}) *RemoteInstance {
if len(m) == 0 {
return nil
}
var ri RemoteInstance
quickSetStr(&ri.RIId, m, "riid")
quickSetStr(&ri.Name, m, "name")
quickSetStr(&ri.SessionId, m, "sessionid")
quickSetStr(&ri.WindowId, m, "windowid")
quickSetStr(&ri.RemoteOwnerId, m, "remoteownerid")
quickSetStr(&ri.RemoteId, m, "remoteid")
2022-11-28 09:13:00 +01:00
quickSetJson(&ri.FeState, m, "festate")
quickSetStr(&ri.StateBaseHash, m, "statebasehash")
quickSetJsonArr(&ri.StateDiffHashArr, m, "statediffhasharr")
return &ri
}
2022-11-28 09:13:00 +01:00
func (ri *RemoteInstance) ToMap() map[string]interface{} {
rtn := make(map[string]interface{})
rtn["riid"] = ri.RIId
rtn["name"] = ri.Name
rtn["sessionid"] = ri.SessionId
rtn["windowid"] = ri.WindowId
rtn["remoteownerid"] = ri.RemoteOwnerId
rtn["remoteid"] = ri.RemoteId
rtn["festate"] = quickJson(ri.FeState)
rtn["statebasehash"] = ri.StateBaseHash
rtn["statediffhasharr"] = quickJsonArr(ri.StateDiffHashArr)
return rtn
}
2022-06-13 20:11:56 +02:00
type LineType struct {
SessionId string `json:"sessionid"`
WindowId string `json:"windowid"`
UserId string `json:"userid"`
LineId string `json:"lineid"`
Ts int64 `json:"ts"`
LineNum int64 `json:"linenum"`
LineNumTemp bool `json:"linenumtemp,omitempty"`
LineLocal bool `json:"linelocal"`
LineType string `json:"linetype"`
Text string `json:"text,omitempty"`
CmdId string `json:"cmdid,omitempty"`
Ephemeral bool `json:"ephemeral,omitempty"`
ContentHeight int64 `json:"contentheight,omitempty"`
Star bool `json:"star,omitempty"`
Archived bool `json:"archived,omitempty"`
Remove bool `json:"remove,omitempty"`
}
type ResolveItem struct {
2022-12-24 00:56:29 +01:00
Name string
Num int
Id string
Hidden bool
}
type SSHOpts struct {
2022-10-01 01:23:40 +02:00
Local bool `json:"local,omitempty"`
SSHHost string `json:"sshhost"`
SSHUser string `json:"sshuser"`
2022-10-01 01:23:40 +02:00
SSHOptsStr string `json:"sshopts,omitempty"`
SSHIdentity string `json:"sshidentity,omitempty"`
SSHPort int `json:"sshport,omitempty"`
2022-10-01 02:22:28 +02:00
SSHPassword string `json:"sshpassword,omitempty"`
}
type RemoteOptsType struct {
Color string `json:"color"`
}
func (opts *RemoteOptsType) Scan(val interface{}) error {
return quickScanJson(opts, val)
}
func (opts RemoteOptsType) Value() (driver.Value, error) {
return quickValueJson(opts)
}
type RemoteType struct {
2022-11-28 09:13:00 +01:00
RemoteId string `json:"remoteid"`
PhysicalId string `json:"physicalid"`
RemoteType string `json:"remotetype"`
RemoteAlias string `json:"remotealias"`
RemoteCanonicalName string `json:"remotecanonicalname"`
RemoteSudo bool `json:"remotesudo"`
RemoteUser string `json:"remoteuser"`
RemoteHost string `json:"remotehost"`
ConnectMode string `json:"connectmode"`
AutoInstall bool `json:"autoinstall"`
SSHOpts *SSHOpts `json:"sshopts"`
RemoteOpts *RemoteOptsType `json:"remoteopts"`
LastConnectTs int64 `json:"lastconnectts"`
Archived bool `json:"archived"`
RemoteIdx int64 `json:"remoteidx"`
Local bool `json:"local"`
2022-08-17 00:08:28 +02:00
}
func (r *RemoteType) GetName() string {
if r.RemoteAlias != "" {
return r.RemoteAlias
}
2022-09-14 02:11:36 +02:00
return r.RemoteCanonicalName
}
type CmdDoneInfo struct {
Ts int64 `json:"ts"`
ExitCode int64 `json:"exitcode"`
DurationMs int64 `json:"durationms"`
}
type CmdType struct {
2022-09-22 07:02:38 +02:00
SessionId string `json:"sessionid"`
CmdId string `json:"cmdid"`
Remote RemotePtrType `json:"remote"`
CmdStr string `json:"cmdstr"`
FeState FeStateType `json:"festate"`
StatePtr ShellStatePtr `json:"state"`
2022-09-22 07:02:38 +02:00
TermOpts TermOpts `json:"termopts"`
OrigTermOpts TermOpts `json:"origtermopts"`
Status string `json:"status"`
StartPk *packet.CmdStartPacketType `json:"startpk,omitempty"`
DoneInfo *CmdDoneInfo `json:"doneinfo,omitempty"`
RunOut []packet.PacketType `json:"runout,omitempty"`
RtnState bool `json:"rtnstate,omitempty"`
RtnStatePtr ShellStatePtr `json:"rtnstateptr,omitempty"`
Remove bool `json:"remove,omitempty"`
2022-07-07 09:10:37 +02:00
}
func (r *RemoteType) ToMap() map[string]interface{} {
rtn := make(map[string]interface{})
rtn["remoteid"] = r.RemoteId
2022-08-17 00:08:28 +02:00
rtn["physicalid"] = r.PhysicalId
rtn["remotetype"] = r.RemoteType
2022-08-17 00:08:28 +02:00
rtn["remotealias"] = r.RemoteAlias
rtn["remotecanonicalname"] = r.RemoteCanonicalName
rtn["remotesudo"] = r.RemoteSudo
rtn["remoteuser"] = r.RemoteUser
rtn["remotehost"] = r.RemoteHost
rtn["connectmode"] = r.ConnectMode
rtn["autoinstall"] = r.AutoInstall
rtn["sshopts"] = quickJson(r.SSHOpts)
rtn["remoteopts"] = quickJson(r.RemoteOpts)
rtn["lastconnectts"] = r.LastConnectTs
2022-09-14 02:11:36 +02:00
rtn["archived"] = r.Archived
2022-09-14 21:06:55 +02:00
rtn["remoteidx"] = r.RemoteIdx
rtn["local"] = r.Local
return rtn
}
func RemoteFromMap(m map[string]interface{}) *RemoteType {
if len(m) == 0 {
return nil
2022-07-07 09:10:37 +02:00
}
var r RemoteType
quickSetStr(&r.RemoteId, m, "remoteid")
2022-08-17 00:08:28 +02:00
quickSetStr(&r.PhysicalId, m, "physicalid")
quickSetStr(&r.RemoteType, m, "remotetype")
2022-08-17 00:08:28 +02:00
quickSetStr(&r.RemoteAlias, m, "remotealias")
quickSetStr(&r.RemoteCanonicalName, m, "remotecanonicalname")
quickSetBool(&r.RemoteSudo, m, "remotesudo")
quickSetStr(&r.RemoteUser, m, "remoteuser")
quickSetStr(&r.RemoteHost, m, "remotehost")
quickSetStr(&r.ConnectMode, m, "connectmode")
quickSetBool(&r.AutoInstall, m, "autoinstall")
quickSetJson(&r.SSHOpts, m, "sshopts")
quickSetJson(&r.RemoteOpts, m, "remoteopts")
quickSetInt64(&r.LastConnectTs, m, "lastconnectts")
2022-09-14 02:11:36 +02:00
quickSetBool(&r.Archived, m, "archived")
2022-09-14 21:06:55 +02:00
quickSetInt64(&r.RemoteIdx, m, "remoteidx")
quickSetBool(&r.Local, m, "local")
return &r
2022-07-07 09:10:37 +02:00
}
func (cmd *CmdType) ToMap() map[string]interface{} {
rtn := make(map[string]interface{})
rtn["sessionid"] = cmd.SessionId
rtn["cmdid"] = cmd.CmdId
2022-08-24 22:21:54 +02:00
rtn["remoteownerid"] = cmd.Remote.OwnerId
rtn["remoteid"] = cmd.Remote.RemoteId
rtn["remotename"] = cmd.Remote.Name
2022-07-07 09:10:37 +02:00
rtn["cmdstr"] = cmd.CmdStr
rtn["festate"] = quickJson(cmd.FeState)
rtn["statebasehash"] = cmd.StatePtr.BaseHash
rtn["statediffhasharr"] = quickJsonArr(cmd.StatePtr.DiffHashArr)
2022-07-07 09:10:37 +02:00
rtn["termopts"] = quickJson(cmd.TermOpts)
2022-09-22 07:02:38 +02:00
rtn["origtermopts"] = quickJson(cmd.OrigTermOpts)
2022-07-07 09:10:37 +02:00
rtn["status"] = cmd.Status
rtn["startpk"] = quickJson(cmd.StartPk)
rtn["doneinfo"] = quickJson(cmd.DoneInfo)
2022-07-07 09:10:37 +02:00
rtn["runout"] = quickJson(cmd.RunOut)
rtn["rtnstate"] = cmd.RtnState
rtn["rtnbasehash"] = cmd.RtnStatePtr.BaseHash
rtn["rtndiffhasharr"] = quickJsonArr(cmd.RtnStatePtr.DiffHashArr)
2022-07-07 09:10:37 +02:00
return rtn
}
func CmdFromMap(m map[string]interface{}) *CmdType {
if len(m) == 0 {
return nil
}
2022-07-07 09:10:37 +02:00
var cmd CmdType
quickSetStr(&cmd.SessionId, m, "sessionid")
quickSetStr(&cmd.CmdId, m, "cmdid")
2022-08-24 22:21:54 +02:00
quickSetStr(&cmd.Remote.OwnerId, m, "remoteownerid")
quickSetStr(&cmd.Remote.RemoteId, m, "remoteid")
quickSetStr(&cmd.Remote.Name, m, "remotename")
2022-07-07 09:10:37 +02:00
quickSetStr(&cmd.CmdStr, m, "cmdstr")
quickSetJson(&cmd.FeState, m, "festate")
quickSetStr(&cmd.StatePtr.BaseHash, m, "statebasehash")
quickSetJsonArr(&cmd.StatePtr.DiffHashArr, m, "statediffhasharr")
2022-07-07 09:10:37 +02:00
quickSetJson(&cmd.TermOpts, m, "termopts")
2022-09-22 07:02:38 +02:00
quickSetJson(&cmd.OrigTermOpts, m, "origtermopts")
2022-07-07 09:10:37 +02:00
quickSetStr(&cmd.Status, m, "status")
quickSetJson(&cmd.StartPk, m, "startpk")
quickSetJson(&cmd.DoneInfo, m, "doneinfo")
2022-07-07 09:10:37 +02:00
quickSetJson(&cmd.RunOut, m, "runout")
quickSetBool(&cmd.RtnState, m, "rtnstate")
quickSetStr(&cmd.RtnStatePtr.BaseHash, m, "rtnbasehash")
quickSetJsonArr(&cmd.RtnStatePtr.DiffHashArr, m, "rtndiffhasharr")
2022-07-07 09:10:37 +02:00
return &cmd
2022-06-13 20:11:56 +02:00
}
func makeNewLineCmd(sessionId string, windowId string, userId string, cmdId string) *LineType {
2022-06-13 20:11:56 +02:00
rtn := &LineType{}
2022-06-17 00:51:41 +02:00
rtn.SessionId = sessionId
rtn.WindowId = windowId
rtn.UserId = userId
rtn.LineId = scbase.GenPromptUUID()
2022-06-13 20:11:56 +02:00
rtn.Ts = time.Now().UnixMilli()
rtn.LineLocal = true
2022-06-13 20:11:56 +02:00
rtn.LineType = LineTypeCmd
rtn.CmdId = cmdId
rtn.ContentHeight = LineNoHeight
2022-06-13 20:11:56 +02:00
return rtn
}
2022-07-05 19:51:47 +02:00
func makeNewLineText(sessionId string, windowId string, userId string, text string) *LineType {
2022-06-13 20:11:56 +02:00
rtn := &LineType{}
2022-06-17 00:51:41 +02:00
rtn.SessionId = sessionId
rtn.WindowId = windowId
rtn.UserId = userId
rtn.LineId = scbase.GenPromptUUID()
2022-06-13 20:11:56 +02:00
rtn.Ts = time.Now().UnixMilli()
rtn.LineLocal = true
2022-06-13 20:11:56 +02:00
rtn.LineType = LineTypeText
rtn.Text = text
rtn.ContentHeight = LineNoHeight
2022-06-13 20:11:56 +02:00
return rtn
}
2022-06-21 06:57:23 +02:00
2022-07-05 19:51:47 +02:00
func AddCommentLine(ctx context.Context, sessionId string, windowId string, userId string, commentText string) (*LineType, error) {
rtnLine := makeNewLineText(sessionId, windowId, userId, commentText)
err := InsertLine(ctx, rtnLine, nil)
2022-07-05 19:51:47 +02:00
if err != nil {
return nil, err
}
return rtnLine, nil
}
func AddCmdLine(ctx context.Context, sessionId string, windowId string, userId string, cmd *CmdType) (*LineType, error) {
rtnLine := makeNewLineCmd(sessionId, windowId, userId, cmd.CmdId)
err := InsertLine(ctx, rtnLine, cmd)
2022-07-05 19:51:47 +02:00
if err != nil {
return nil, err
}
return rtnLine, nil
}
2022-07-01 21:17:19 +02:00
func EnsureLocalRemote(ctx context.Context) error {
physicalId, err := base.GetRemoteId()
2022-07-01 21:17:19 +02:00
if err != nil {
return fmt.Errorf("getting local physical remoteid: %w", err)
2022-07-01 21:17:19 +02:00
}
remote, err := GetLocalRemote(ctx)
2022-07-01 21:17:19 +02:00
if err != nil {
return fmt.Errorf("getting local remote from db: %w", err)
2022-07-01 21:17:19 +02:00
}
if remote != nil {
return nil
}
2022-08-17 00:08:28 +02:00
hostName, err := os.Hostname()
if err != nil {
return fmt.Errorf("getting hostname: %w", err)
}
user, err := user.Current()
if err != nil {
return fmt.Errorf("getting user: %w", err)
}
2022-07-01 21:17:19 +02:00
// create the local remote
localRemote := &RemoteType{
RemoteId: scbase.GenPromptUUID(),
PhysicalId: physicalId,
RemoteType: RemoteTypeSsh,
RemoteAlias: LocalRemoteAlias,
2022-08-17 00:08:28 +02:00
RemoteCanonicalName: fmt.Sprintf("%s@%s", user.Username, hostName),
RemoteSudo: false,
RemoteUser: user.Username,
RemoteHost: hostName,
ConnectMode: ConnectModeStartup,
2022-09-25 07:42:52 +02:00
AutoInstall: true,
2022-08-24 06:05:49 +02:00
SSHOpts: &SSHOpts{Local: true},
Local: true,
2022-07-01 21:17:19 +02:00
}
2022-09-14 02:11:36 +02:00
err = UpsertRemote(ctx, localRemote)
2022-07-01 21:17:19 +02:00
if err != nil {
return err
}
log.Printf("[db] added local remote '%s', id=%s\n", localRemote.RemoteCanonicalName, localRemote.RemoteId)
2022-12-29 09:07:16 +01:00
sudoRemote := &RemoteType{
RemoteId: scbase.GenPromptUUID(),
PhysicalId: "",
RemoteType: RemoteTypeSsh,
RemoteAlias: "sudo",
RemoteCanonicalName: fmt.Sprintf("sudo@%s@%s", user.Username, hostName),
RemoteSudo: true,
RemoteUser: "root",
RemoteHost: hostName,
ConnectMode: ConnectModeManual,
AutoInstall: true,
SSHOpts: &SSHOpts{Local: true},
RemoteOpts: &RemoteOptsType{Color: "red"},
Local: true,
}
err = UpsertRemote(ctx, sudoRemote)
if err != nil {
return err
}
log.Printf("[db] added sudo remote '%s', id=%s\n", sudoRemote.RemoteCanonicalName, sudoRemote.RemoteId)
return nil
}
2022-07-01 23:45:33 +02:00
func EnsureDefaultSession(ctx context.Context) (*SessionType, error) {
2022-07-01 23:07:13 +02:00
session, err := GetSessionByName(ctx, DefaultSessionName)
if err != nil {
2022-07-01 23:45:33 +02:00
return nil, err
2022-07-01 23:07:13 +02:00
}
if session != nil {
2022-07-01 23:45:33 +02:00
return session, nil
2022-07-01 23:07:13 +02:00
}
_, err = InsertSessionWithName(ctx, DefaultSessionName, true)
2022-07-01 23:07:13 +02:00
if err != nil {
2022-07-01 23:45:33 +02:00
return nil, err
2022-07-01 23:07:13 +02:00
}
2022-07-01 23:45:33 +02:00
return GetSessionByName(ctx, DefaultSessionName)
2022-07-01 21:17:19 +02:00
}
func createClientData(tx *TxWrap) error {
curve := elliptic.P384()
pkey, err := ecdsa.GenerateKey(curve, rand.Reader)
if err != nil {
return fmt.Errorf("generating P-834 key: %w", err)
}
pkBytes, err := x509.MarshalECPrivateKey(pkey)
if err != nil {
return fmt.Errorf("marshaling (pkcs8) private key bytes: %w", err)
}
pubBytes, err := x509.MarshalPKIXPublicKey(&pkey.PublicKey)
if err != nil {
return fmt.Errorf("marshaling (pkix) public key bytes: %w", err)
}
2022-09-22 07:02:38 +02:00
c := ClientData{
ClientId: uuid.New().String(),
UserId: uuid.New().String(),
UserPrivateKeyBytes: pkBytes,
UserPublicKeyBytes: pubBytes,
ActiveSessionId: "",
2022-09-25 09:26:33 +02:00
WinSize: ClientWinSizeType{},
2022-09-22 07:02:38 +02:00
}
2022-09-25 09:26:33 +02:00
query := `INSERT INTO client ( clientid, userid, activesessionid, userpublickeybytes, userprivatekeybytes, winsize)
VALUES (:clientid,:userid,:activesessionid,:userpublickeybytes,:userprivatekeybytes,:winsize)`
2022-09-22 07:02:38 +02:00
tx.NamedExecWrap(query, c.ToMap())
2022-10-31 20:40:45 +01:00
log.Printf("create new clientid[%s] userid[%s] with public/private keypair\n", c.ClientId, c.UserId)
return nil
}
func EnsureClientData(ctx context.Context) (*ClientData, error) {
var rtn ClientData
err := WithTx(ctx, func(tx *TxWrap) error {
query := `SELECT count(*) FROM client`
count := tx.GetInt(query)
if count > 1 {
return fmt.Errorf("invalid client database, multiple (%d) rows in client table", count)
}
if count == 0 {
createErr := createClientData(tx)
if createErr != nil {
return createErr
}
}
2022-09-25 09:26:33 +02:00
m := tx.GetMap(`SELECT * FROM client`)
2022-09-22 07:02:38 +02:00
cdata := ClientDataFromMap(m)
if cdata == nil {
2022-09-25 09:26:33 +02:00
return fmt.Errorf("no client data found")
}
2022-09-22 07:02:38 +02:00
rtn = *cdata
return nil
})
if err != nil {
return nil, err
}
if rtn.UserId == "" {
return nil, fmt.Errorf("invalid client data (no userid)")
}
if len(rtn.UserPrivateKeyBytes) == 0 || len(rtn.UserPublicKeyBytes) == 0 {
return nil, fmt.Errorf("invalid client data (no public/private keypair)")
}
rtn.UserPrivateKey, err = x509.ParseECPrivateKey(rtn.UserPrivateKeyBytes)
if err != nil {
return nil, fmt.Errorf("invalid client data, cannot parse private key: %w", err)
}
pubKey, err := x509.ParsePKIXPublicKey(rtn.UserPublicKeyBytes)
if err != nil {
return nil, fmt.Errorf("invalid client data, cannot parse public key: %w", err)
}
var ok bool
rtn.UserPublicKey, ok = pubKey.(*ecdsa.PublicKey)
if !ok {
return nil, fmt.Errorf("invalid client data, wrong public key type: %T", pubKey)
}
return &rtn, nil
}