ssh_key.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package db
  5. import (
  6. "context"
  7. "encoding/base64"
  8. "encoding/binary"
  9. "errors"
  10. "fmt"
  11. "io/ioutil"
  12. "math/big"
  13. "os"
  14. "path"
  15. "path/filepath"
  16. "strings"
  17. "sync"
  18. "time"
  19. "github.com/unknwon/com"
  20. "golang.org/x/crypto/ssh"
  21. log "unknwon.dev/clog/v2"
  22. "xorm.io/xorm"
  23. "gogs.io/gogs/internal/conf"
  24. "gogs.io/gogs/internal/errutil"
  25. "gogs.io/gogs/internal/process"
  26. )
  27. const (
  28. _TPL_PUBLICK_KEY = `command="%s serv key-%d --config='%s'",no-port-forwarding,no-X11-forwarding,no-agent-forwarding,no-pty %s` + "\n"
  29. )
  30. var sshOpLocker sync.Mutex
  31. type KeyType int
  32. const (
  33. KEY_TYPE_USER = iota + 1
  34. KEY_TYPE_DEPLOY
  35. )
  36. // PublicKey represents a user or deploy SSH public key.
  37. type PublicKey struct {
  38. ID int64
  39. OwnerID int64 `xorm:"INDEX NOT NULL"`
  40. Name string `xorm:"NOT NULL"`
  41. Fingerprint string `xorm:"NOT NULL"`
  42. Content string `xorm:"TEXT NOT NULL"`
  43. Mode AccessMode `xorm:"NOT NULL DEFAULT 2"`
  44. Type KeyType `xorm:"NOT NULL DEFAULT 1"`
  45. Created time.Time `xorm:"-" json:"-"`
  46. CreatedUnix int64
  47. Updated time.Time `xorm:"-" json:"-"` // Note: Updated must below Created for AfterSet.
  48. UpdatedUnix int64
  49. HasRecentActivity bool `xorm:"-" json:"-"`
  50. HasUsed bool `xorm:"-" json:"-"`
  51. }
  52. func (k *PublicKey) BeforeInsert() {
  53. k.CreatedUnix = time.Now().Unix()
  54. }
  55. func (k *PublicKey) BeforeUpdate() {
  56. k.UpdatedUnix = time.Now().Unix()
  57. }
  58. func (k *PublicKey) AfterSet(colName string, _ xorm.Cell) {
  59. switch colName {
  60. case "created_unix":
  61. k.Created = time.Unix(k.CreatedUnix, 0).Local()
  62. case "updated_unix":
  63. k.Updated = time.Unix(k.UpdatedUnix, 0).Local()
  64. k.HasUsed = k.Updated.After(k.Created)
  65. k.HasRecentActivity = k.Updated.Add(7 * 24 * time.Hour).After(time.Now())
  66. }
  67. }
  68. // OmitEmail returns content of public key without email address.
  69. func (k *PublicKey) OmitEmail() string {
  70. return strings.Join(strings.Split(k.Content, " ")[:2], " ")
  71. }
  72. // AuthorizedString returns formatted public key string for authorized_keys file.
  73. func (k *PublicKey) AuthorizedString() string {
  74. return fmt.Sprintf(_TPL_PUBLICK_KEY, conf.AppPath(), k.ID, conf.CustomConf, k.Content)
  75. }
  76. // IsDeployKey returns true if the public key is used as deploy key.
  77. func (k *PublicKey) IsDeployKey() bool {
  78. return k.Type == KEY_TYPE_DEPLOY
  79. }
  80. func extractTypeFromBase64Key(key string) (string, error) {
  81. b, err := base64.StdEncoding.DecodeString(key)
  82. if err != nil || len(b) < 4 {
  83. return "", fmt.Errorf("invalid key format: %v", err)
  84. }
  85. keyLength := int(binary.BigEndian.Uint32(b))
  86. if len(b) < 4+keyLength {
  87. return "", fmt.Errorf("invalid key format: not enough length %d", keyLength)
  88. }
  89. return string(b[4 : 4+keyLength]), nil
  90. }
  91. // parseKeyString parses any key string in OpenSSH or SSH2 format to clean OpenSSH string (RFC4253).
  92. func parseKeyString(content string) (string, error) {
  93. // Transform all legal line endings to a single "\n"
  94. // Replace all windows full new lines ("\r\n")
  95. content = strings.ReplaceAll(content, "\r\n", "\n")
  96. // Replace all windows half new lines ("\r"), if it happen not to match replace above
  97. content = strings.ReplaceAll(content, "\r", "\n")
  98. // Replace ending new line as its may cause unwanted behaviour (extra line means not a single line key | OpenSSH key)
  99. content = strings.TrimRight(content, "\n")
  100. // split lines
  101. lines := strings.Split(content, "\n")
  102. var keyType, keyContent, keyComment string
  103. if len(lines) == 1 {
  104. // Parse OpenSSH format.
  105. parts := strings.SplitN(lines[0], " ", 3)
  106. switch len(parts) {
  107. case 0:
  108. return "", errors.New("empty key")
  109. case 1:
  110. keyContent = parts[0]
  111. case 2:
  112. keyType = parts[0]
  113. keyContent = parts[1]
  114. default:
  115. keyType = parts[0]
  116. keyContent = parts[1]
  117. keyComment = parts[2]
  118. }
  119. // If keyType is not given, extract it from content. If given, validate it.
  120. t, err := extractTypeFromBase64Key(keyContent)
  121. if err != nil {
  122. return "", fmt.Errorf("extractTypeFromBase64Key: %v", err)
  123. }
  124. if keyType == "" {
  125. keyType = t
  126. } else if keyType != t {
  127. return "", fmt.Errorf("key type and content does not match: %s - %s", keyType, t)
  128. }
  129. } else {
  130. // Parse SSH2 file format.
  131. continuationLine := false
  132. for _, line := range lines {
  133. // Skip lines that:
  134. // 1) are a continuation of the previous line,
  135. // 2) contain ":" as that are comment lines
  136. // 3) contain "-" as that are begin and end tags
  137. if continuationLine || strings.ContainsAny(line, ":-") {
  138. continuationLine = strings.HasSuffix(line, "\\")
  139. } else {
  140. keyContent = keyContent + line
  141. }
  142. }
  143. t, err := extractTypeFromBase64Key(keyContent)
  144. if err != nil {
  145. return "", fmt.Errorf("extractTypeFromBase64Key: %v", err)
  146. }
  147. keyType = t
  148. }
  149. return keyType + " " + keyContent + " " + keyComment, nil
  150. }
  151. // writeTmpKeyFile writes key content to a temporary file
  152. // and returns the name of that file, along with any possible errors.
  153. func writeTmpKeyFile(content string) (string, error) {
  154. tmpFile, err := ioutil.TempFile(conf.SSH.KeyTestPath, "gogs_keytest")
  155. if err != nil {
  156. return "", fmt.Errorf("TempFile: %v", err)
  157. }
  158. defer tmpFile.Close()
  159. if _, err = tmpFile.WriteString(content); err != nil {
  160. return "", fmt.Errorf("WriteString: %v", err)
  161. }
  162. return tmpFile.Name(), nil
  163. }
  164. // SSHKeyGenParsePublicKey extracts key type and length using ssh-keygen.
  165. func SSHKeyGenParsePublicKey(key string) (string, int, error) {
  166. tmpName, err := writeTmpKeyFile(key)
  167. if err != nil {
  168. return "", 0, fmt.Errorf("writeTmpKeyFile: %v", err)
  169. }
  170. defer os.Remove(tmpName)
  171. stdout, stderr, err := process.Exec("SSHKeyGenParsePublicKey", conf.SSH.KeygenPath, "-lf", tmpName)
  172. if err != nil {
  173. return "", 0, fmt.Errorf("fail to parse public key: %s - %s", err, stderr)
  174. }
  175. if strings.Contains(stdout, "is not a public key file") {
  176. return "", 0, ErrKeyUnableVerify{stdout}
  177. }
  178. fields := strings.Split(stdout, " ")
  179. if len(fields) < 4 {
  180. return "", 0, fmt.Errorf("invalid public key line: %s", stdout)
  181. }
  182. keyType := strings.Trim(fields[len(fields)-1], "()\r\n")
  183. return strings.ToLower(keyType), com.StrTo(fields[0]).MustInt(), nil
  184. }
  185. // SSHNativeParsePublicKey extracts the key type and length using the golang SSH library.
  186. func SSHNativeParsePublicKey(keyLine string) (string, int, error) {
  187. fields := strings.Fields(keyLine)
  188. if len(fields) < 2 {
  189. return "", 0, fmt.Errorf("not enough fields in public key line: %s", keyLine)
  190. }
  191. raw, err := base64.StdEncoding.DecodeString(fields[1])
  192. if err != nil {
  193. return "", 0, err
  194. }
  195. pkey, err := ssh.ParsePublicKey(raw)
  196. if err != nil {
  197. if strings.Contains(err.Error(), "ssh: unknown key algorithm") {
  198. return "", 0, ErrKeyUnableVerify{err.Error()}
  199. }
  200. return "", 0, fmt.Errorf("ParsePublicKey: %v", err)
  201. }
  202. // The ssh library can parse the key, so next we find out what key exactly we have.
  203. switch pkey.Type() {
  204. case ssh.KeyAlgoDSA:
  205. rawPub := struct {
  206. Name string
  207. P, Q, G, Y *big.Int
  208. }{}
  209. if err := ssh.Unmarshal(pkey.Marshal(), &rawPub); err != nil {
  210. return "", 0, err
  211. }
  212. // as per https://bugzilla.mindrot.org/show_bug.cgi?id=1647 we should never
  213. // see dsa keys != 1024 bit, but as it seems to work, we will not check here
  214. return "dsa", rawPub.P.BitLen(), nil // use P as per crypto/dsa/dsa.go (is L)
  215. case ssh.KeyAlgoRSA:
  216. rawPub := struct {
  217. Name string
  218. E *big.Int
  219. N *big.Int
  220. }{}
  221. if err := ssh.Unmarshal(pkey.Marshal(), &rawPub); err != nil {
  222. return "", 0, err
  223. }
  224. return "rsa", rawPub.N.BitLen(), nil // use N as per crypto/rsa/rsa.go (is bits)
  225. case ssh.KeyAlgoECDSA256:
  226. return "ecdsa", 256, nil
  227. case ssh.KeyAlgoECDSA384:
  228. return "ecdsa", 384, nil
  229. case ssh.KeyAlgoECDSA521:
  230. return "ecdsa", 521, nil
  231. case ssh.KeyAlgoED25519:
  232. return "ed25519", 256, nil
  233. }
  234. return "", 0, fmt.Errorf("unsupported key length detection for type: %s", pkey.Type())
  235. }
  236. // CheckPublicKeyString checks if the given public key string is recognized by SSH.
  237. // It returns the actual public key line on success.
  238. func CheckPublicKeyString(content string) (_ string, err error) {
  239. if conf.SSH.Disabled {
  240. return "", errors.New("SSH is disabled")
  241. }
  242. content, err = parseKeyString(content)
  243. if err != nil {
  244. return "", err
  245. }
  246. content = strings.TrimRight(content, "\n\r")
  247. if strings.ContainsAny(content, "\n\r") {
  248. return "", errors.New("only a single line with a single key please")
  249. }
  250. // Remove any unnecessary whitespace
  251. content = strings.TrimSpace(content)
  252. if !conf.SSH.MinimumKeySizeCheck {
  253. return content, nil
  254. }
  255. var (
  256. fnName string
  257. keyType string
  258. length int
  259. )
  260. if conf.SSH.StartBuiltinServer {
  261. fnName = "SSHNativeParsePublicKey"
  262. keyType, length, err = SSHNativeParsePublicKey(content)
  263. } else {
  264. fnName = "SSHKeyGenParsePublicKey"
  265. keyType, length, err = SSHKeyGenParsePublicKey(content)
  266. }
  267. if err != nil {
  268. return "", fmt.Errorf("%s: %v", fnName, err)
  269. }
  270. log.Trace("Key info [native: %v]: %s-%d", conf.SSH.StartBuiltinServer, keyType, length)
  271. if minLen, found := conf.SSH.MinimumKeySizes[keyType]; found && length >= minLen {
  272. return content, nil
  273. } else if found && length < minLen {
  274. return "", fmt.Errorf("key length is not enough: got %d, needs %d", length, minLen)
  275. }
  276. return "", fmt.Errorf("key type is not allowed: %s", keyType)
  277. }
  278. // appendAuthorizedKeysToFile appends new SSH keys' content to authorized_keys file.
  279. func appendAuthorizedKeysToFile(keys ...*PublicKey) error {
  280. sshOpLocker.Lock()
  281. defer sshOpLocker.Unlock()
  282. fpath := filepath.Join(conf.SSH.RootPath, "authorized_keys")
  283. f, err := os.OpenFile(fpath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0600)
  284. if err != nil {
  285. return err
  286. }
  287. defer f.Close()
  288. // Note: chmod command does not support in Windows.
  289. if !conf.IsWindowsRuntime() {
  290. fi, err := f.Stat()
  291. if err != nil {
  292. return err
  293. }
  294. // .ssh directory should have mode 700, and authorized_keys file should have mode 600.
  295. if fi.Mode().Perm() > 0600 {
  296. log.Error("authorized_keys file has unusual permission flags: %s - setting to -rw-------", fi.Mode().Perm().String())
  297. if err = f.Chmod(0600); err != nil {
  298. return err
  299. }
  300. }
  301. }
  302. for _, key := range keys {
  303. if _, err = f.WriteString(key.AuthorizedString()); err != nil {
  304. return err
  305. }
  306. }
  307. return nil
  308. }
  309. // checkKeyContent onlys checks if key content has been used as public key,
  310. // it is OK to use same key as deploy key for multiple repositories/users.
  311. func checkKeyContent(content string) error {
  312. has, err := x.Get(&PublicKey{
  313. Content: content,
  314. Type: KEY_TYPE_USER,
  315. })
  316. if err != nil {
  317. return err
  318. } else if has {
  319. return ErrKeyAlreadyExist{0, content}
  320. }
  321. return nil
  322. }
  323. func addKey(e Engine, key *PublicKey) (err error) {
  324. // Calculate fingerprint.
  325. tmpPath := strings.ReplaceAll(path.Join(os.TempDir(), fmt.Sprintf("%d", time.Now().Nanosecond()), "id_rsa.pub"), "\\", "/")
  326. _ = os.MkdirAll(path.Dir(tmpPath), os.ModePerm)
  327. if err = ioutil.WriteFile(tmpPath, []byte(key.Content), 0644); err != nil {
  328. return err
  329. }
  330. stdout, stderr, err := process.Exec("AddPublicKey", conf.SSH.KeygenPath, "-lf", tmpPath)
  331. if err != nil {
  332. return fmt.Errorf("fail to parse public key: %s - %s", err, stderr)
  333. } else if len(stdout) < 2 {
  334. return errors.New("not enough output for calculating fingerprint: " + stdout)
  335. }
  336. key.Fingerprint = strings.Split(stdout, " ")[1]
  337. // Save SSH key.
  338. if _, err = e.Insert(key); err != nil {
  339. return err
  340. }
  341. // Don't need to rewrite this file if builtin SSH server is enabled.
  342. if conf.SSH.StartBuiltinServer {
  343. return nil
  344. }
  345. return appendAuthorizedKeysToFile(key)
  346. }
  347. // AddPublicKey adds new public key to database and authorized_keys file.
  348. func AddPublicKey(ownerID int64, name, content string) (*PublicKey, error) {
  349. log.Trace(content)
  350. if err := checkKeyContent(content); err != nil {
  351. return nil, err
  352. }
  353. // Key name of same user cannot be duplicated.
  354. has, err := x.Where("owner_id = ? AND name = ?", ownerID, name).Get(new(PublicKey))
  355. if err != nil {
  356. return nil, err
  357. } else if has {
  358. return nil, ErrKeyNameAlreadyUsed{ownerID, name}
  359. }
  360. sess := x.NewSession()
  361. defer sess.Close()
  362. if err = sess.Begin(); err != nil {
  363. return nil, err
  364. }
  365. key := &PublicKey{
  366. OwnerID: ownerID,
  367. Name: name,
  368. Content: content,
  369. Mode: AccessModeWrite,
  370. Type: KEY_TYPE_USER,
  371. }
  372. if err = addKey(sess, key); err != nil {
  373. return nil, fmt.Errorf("addKey: %v", err)
  374. }
  375. return key, sess.Commit()
  376. }
  377. // GetPublicKeyByID returns public key by given ID.
  378. func GetPublicKeyByID(keyID int64) (*PublicKey, error) {
  379. key := new(PublicKey)
  380. has, err := x.Id(keyID).Get(key)
  381. if err != nil {
  382. return nil, err
  383. } else if !has {
  384. return nil, ErrKeyNotExist{keyID}
  385. }
  386. return key, nil
  387. }
  388. // SearchPublicKeyByContent searches content as prefix (leak e-mail part)
  389. // and returns public key found.
  390. func SearchPublicKeyByContent(content string) (*PublicKey, error) {
  391. key := new(PublicKey)
  392. has, err := x.Where("content like ?", content+"%").Get(key)
  393. if err != nil {
  394. return nil, err
  395. } else if !has {
  396. return nil, ErrKeyNotExist{}
  397. }
  398. return key, nil
  399. }
  400. // ListPublicKeys returns a list of public keys belongs to given user.
  401. func ListPublicKeys(uid int64) ([]*PublicKey, error) {
  402. keys := make([]*PublicKey, 0, 5)
  403. return keys, x.Where("owner_id = ?", uid).Find(&keys)
  404. }
  405. // UpdatePublicKey updates given public key.
  406. func UpdatePublicKey(key *PublicKey) error {
  407. _, err := x.Id(key.ID).AllCols().Update(key)
  408. return err
  409. }
  410. // deletePublicKeys does the actual key deletion but does not update authorized_keys file.
  411. func deletePublicKeys(e *xorm.Session, keyIDs ...int64) error {
  412. if len(keyIDs) == 0 {
  413. return nil
  414. }
  415. _, err := e.In("id", keyIDs).Delete(new(PublicKey))
  416. return err
  417. }
  418. // DeletePublicKey deletes SSH key information both in database and authorized_keys file.
  419. func DeletePublicKey(doer *User, id int64) (err error) {
  420. key, err := GetPublicKeyByID(id)
  421. if err != nil {
  422. if IsErrKeyNotExist(err) {
  423. return nil
  424. }
  425. return fmt.Errorf("GetPublicKeyByID: %v", err)
  426. }
  427. // Check if user has access to delete this key.
  428. if !doer.IsAdmin && doer.ID != key.OwnerID {
  429. return ErrKeyAccessDenied{doer.ID, key.ID, "public"}
  430. }
  431. sess := x.NewSession()
  432. defer sess.Close()
  433. if err = sess.Begin(); err != nil {
  434. return err
  435. }
  436. if err = deletePublicKeys(sess, id); err != nil {
  437. return err
  438. }
  439. if err = sess.Commit(); err != nil {
  440. return err
  441. }
  442. return RewriteAuthorizedKeys()
  443. }
  444. // RewriteAuthorizedKeys removes any authorized key and rewrite all keys from database again.
  445. // Note: x.Iterate does not get latest data after insert/delete, so we have to call this function
  446. // outside any session scope independently.
  447. func RewriteAuthorizedKeys() error {
  448. sshOpLocker.Lock()
  449. defer sshOpLocker.Unlock()
  450. log.Trace("Doing: RewriteAuthorizedKeys")
  451. _ = os.MkdirAll(conf.SSH.RootPath, os.ModePerm)
  452. fpath := filepath.Join(conf.SSH.RootPath, "authorized_keys")
  453. tmpPath := fpath + ".tmp"
  454. f, err := os.OpenFile(tmpPath, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0600)
  455. if err != nil {
  456. return err
  457. }
  458. defer os.Remove(tmpPath)
  459. err = x.Iterate(new(PublicKey), func(idx int, bean interface{}) (err error) {
  460. _, err = f.WriteString((bean.(*PublicKey)).AuthorizedString())
  461. return err
  462. })
  463. _ = f.Close()
  464. if err != nil {
  465. return err
  466. }
  467. if com.IsExist(fpath) {
  468. if err = os.Remove(fpath); err != nil {
  469. return err
  470. }
  471. }
  472. if err = os.Rename(tmpPath, fpath); err != nil {
  473. return err
  474. }
  475. return nil
  476. }
  477. // ________ .__ ____ __.
  478. // \______ \ ____ ______ | | ____ ___.__.| |/ _|____ ___.__.
  479. // | | \_/ __ \\____ \| | / _ < | || <_/ __ < | |
  480. // | ` \ ___/| |_> > |_( <_> )___ || | \ ___/\___ |
  481. // /_______ /\___ > __/|____/\____// ____||____|__ \___ > ____|
  482. // \/ \/|__| \/ \/ \/\/
  483. // DeployKey represents deploy key information and its relation with repository.
  484. type DeployKey struct {
  485. ID int64
  486. KeyID int64 `xorm:"UNIQUE(s) INDEX"`
  487. RepoID int64 `xorm:"UNIQUE(s) INDEX"`
  488. Name string
  489. Fingerprint string
  490. Content string `xorm:"-" json:"-"`
  491. Created time.Time `xorm:"-" json:"-"`
  492. CreatedUnix int64
  493. Updated time.Time `xorm:"-" json:"-"` // Note: Updated must below Created for AfterSet.
  494. UpdatedUnix int64
  495. HasRecentActivity bool `xorm:"-" json:"-"`
  496. HasUsed bool `xorm:"-" json:"-"`
  497. }
  498. func (k *DeployKey) BeforeInsert() {
  499. k.CreatedUnix = time.Now().Unix()
  500. }
  501. func (k *DeployKey) BeforeUpdate() {
  502. k.UpdatedUnix = time.Now().Unix()
  503. }
  504. func (k *DeployKey) AfterSet(colName string, _ xorm.Cell) {
  505. switch colName {
  506. case "created_unix":
  507. k.Created = time.Unix(k.CreatedUnix, 0).Local()
  508. case "updated_unix":
  509. k.Updated = time.Unix(k.UpdatedUnix, 0).Local()
  510. k.HasUsed = k.Updated.After(k.Created)
  511. k.HasRecentActivity = k.Updated.Add(7 * 24 * time.Hour).After(time.Now())
  512. }
  513. }
  514. // GetContent gets associated public key content.
  515. func (k *DeployKey) GetContent() error {
  516. pkey, err := GetPublicKeyByID(k.KeyID)
  517. if err != nil {
  518. return err
  519. }
  520. k.Content = pkey.Content
  521. return nil
  522. }
  523. func checkDeployKey(e Engine, keyID, repoID int64, name string) error {
  524. // Note: We want error detail, not just true or false here.
  525. has, err := e.Where("key_id = ? AND repo_id = ?", keyID, repoID).Get(new(DeployKey))
  526. if err != nil {
  527. return err
  528. } else if has {
  529. return ErrDeployKeyAlreadyExist{keyID, repoID}
  530. }
  531. has, err = e.Where("repo_id = ? AND name = ?", repoID, name).Get(new(DeployKey))
  532. if err != nil {
  533. return err
  534. } else if has {
  535. return ErrDeployKeyNameAlreadyUsed{repoID, name}
  536. }
  537. return nil
  538. }
  539. // addDeployKey adds new key-repo relation.
  540. func addDeployKey(e *xorm.Session, keyID, repoID int64, name, fingerprint string) (*DeployKey, error) {
  541. if err := checkDeployKey(e, keyID, repoID, name); err != nil {
  542. return nil, err
  543. }
  544. key := &DeployKey{
  545. KeyID: keyID,
  546. RepoID: repoID,
  547. Name: name,
  548. Fingerprint: fingerprint,
  549. }
  550. _, err := e.Insert(key)
  551. return key, err
  552. }
  553. // HasDeployKey returns true if public key is a deploy key of given repository.
  554. func HasDeployKey(keyID, repoID int64) bool {
  555. has, _ := x.Where("key_id = ? AND repo_id = ?", keyID, repoID).Get(new(DeployKey))
  556. return has
  557. }
  558. // AddDeployKey add new deploy key to database and authorized_keys file.
  559. func AddDeployKey(repoID int64, name, content string) (*DeployKey, error) {
  560. if err := checkKeyContent(content); err != nil {
  561. return nil, err
  562. }
  563. pkey := &PublicKey{
  564. Content: content,
  565. Mode: AccessModeRead,
  566. Type: KEY_TYPE_DEPLOY,
  567. }
  568. has, err := x.Get(pkey)
  569. if err != nil {
  570. return nil, err
  571. }
  572. sess := x.NewSession()
  573. defer sess.Close()
  574. if err = sess.Begin(); err != nil {
  575. return nil, err
  576. }
  577. // First time use this deploy key.
  578. if !has {
  579. if err = addKey(sess, pkey); err != nil {
  580. return nil, fmt.Errorf("addKey: %v", err)
  581. }
  582. }
  583. key, err := addDeployKey(sess, pkey.ID, repoID, name, pkey.Fingerprint)
  584. if err != nil {
  585. return nil, fmt.Errorf("addDeployKey: %v", err)
  586. }
  587. return key, sess.Commit()
  588. }
  589. var _ errutil.NotFound = (*ErrDeployKeyNotExist)(nil)
  590. type ErrDeployKeyNotExist struct {
  591. args map[string]interface{}
  592. }
  593. func IsErrDeployKeyNotExist(err error) bool {
  594. _, ok := err.(ErrDeployKeyNotExist)
  595. return ok
  596. }
  597. func (err ErrDeployKeyNotExist) Error() string {
  598. return fmt.Sprintf("deploy key does not exist: %v", err.args)
  599. }
  600. func (ErrDeployKeyNotExist) NotFound() bool {
  601. return true
  602. }
  603. // GetDeployKeyByID returns deploy key by given ID.
  604. func GetDeployKeyByID(id int64) (*DeployKey, error) {
  605. key := new(DeployKey)
  606. has, err := x.Id(id).Get(key)
  607. if err != nil {
  608. return nil, err
  609. } else if !has {
  610. return nil, ErrDeployKeyNotExist{args: map[string]interface{}{"deployKeyID": id}}
  611. }
  612. return key, nil
  613. }
  614. // GetDeployKeyByRepo returns deploy key by given public key ID and repository ID.
  615. func GetDeployKeyByRepo(keyID, repoID int64) (*DeployKey, error) {
  616. key := &DeployKey{
  617. KeyID: keyID,
  618. RepoID: repoID,
  619. }
  620. has, err := x.Get(key)
  621. if err != nil {
  622. return nil, err
  623. } else if !has {
  624. return nil, ErrDeployKeyNotExist{args: map[string]interface{}{"keyID": keyID, "repoID": repoID}}
  625. }
  626. return key, nil
  627. }
  628. // UpdateDeployKey updates deploy key information.
  629. func UpdateDeployKey(key *DeployKey) error {
  630. _, err := x.Id(key.ID).AllCols().Update(key)
  631. return err
  632. }
  633. // DeleteDeployKey deletes deploy key from its repository authorized_keys file if needed.
  634. func DeleteDeployKey(doer *User, id int64) error {
  635. key, err := GetDeployKeyByID(id)
  636. if err != nil {
  637. if IsErrDeployKeyNotExist(err) {
  638. return nil
  639. }
  640. return fmt.Errorf("GetDeployKeyByID: %v", err)
  641. }
  642. // Check if user has access to delete this key.
  643. if !doer.IsAdmin {
  644. repo, err := GetRepositoryByID(key.RepoID)
  645. if err != nil {
  646. return fmt.Errorf("GetRepositoryByID: %v", err)
  647. }
  648. if !Perms.Authorize(context.TODO(), doer.ID, repo.ID, AccessModeAdmin,
  649. AccessModeOptions{
  650. OwnerID: repo.OwnerID,
  651. Private: repo.IsPrivate,
  652. },
  653. ) {
  654. return ErrKeyAccessDenied{doer.ID, key.ID, "deploy"}
  655. }
  656. }
  657. sess := x.NewSession()
  658. defer sess.Close()
  659. if err = sess.Begin(); err != nil {
  660. return err
  661. }
  662. if _, err = sess.ID(key.ID).Delete(new(DeployKey)); err != nil {
  663. return fmt.Errorf("delete deploy key [%d]: %v", key.ID, err)
  664. }
  665. // Check if this is the last reference to same key content.
  666. has, err := sess.Where("key_id = ?", key.KeyID).Get(new(DeployKey))
  667. if err != nil {
  668. return err
  669. } else if !has {
  670. if err = deletePublicKeys(sess, key.KeyID); err != nil {
  671. return err
  672. }
  673. }
  674. return sess.Commit()
  675. }
  676. // ListDeployKeys returns all deploy keys by given repository ID.
  677. func ListDeployKeys(repoID int64) ([]*DeployKey, error) {
  678. keys := make([]*DeployKey, 0, 5)
  679. return keys, x.Where("repo_id = ?", repoID).Find(&keys)
  680. }