Nevar pievienot vairāk kā 25 tēmas Tēmai ir jāsākas ar burtu vai ciparu, tā var saturēt domu zīmes ('-') un var būt līdz 35 simboliem gara.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804
  1. // Copyright 2017 The Gitea Authors. All rights reserved.
  2. // SPDX-License-Identifier: MIT
  3. package cmd
  4. import (
  5. "bufio"
  6. "bytes"
  7. "context"
  8. "fmt"
  9. "io"
  10. "os"
  11. "strconv"
  12. "strings"
  13. "time"
  14. "code.gitea.io/gitea/modules/git"
  15. "code.gitea.io/gitea/modules/log"
  16. "code.gitea.io/gitea/modules/private"
  17. repo_module "code.gitea.io/gitea/modules/repository"
  18. "code.gitea.io/gitea/modules/setting"
  19. "github.com/urfave/cli/v2"
  20. )
  21. const (
  22. hookBatchSize = 30
  23. )
  24. var (
  25. // CmdHook represents the available hooks sub-command.
  26. CmdHook = &cli.Command{
  27. Name: "hook",
  28. Usage: "(internal) Should only be called by Git",
  29. Description: "Delegate commands to corresponding Git hooks",
  30. Before: PrepareConsoleLoggerLevel(log.FATAL),
  31. Subcommands: []*cli.Command{
  32. subcmdHookPreReceive,
  33. subcmdHookUpdate,
  34. subcmdHookPostReceive,
  35. subcmdHookProcReceive,
  36. },
  37. }
  38. subcmdHookPreReceive = &cli.Command{
  39. Name: "pre-receive",
  40. Usage: "Delegate pre-receive Git hook",
  41. Description: "This command should only be called by Git",
  42. Action: runHookPreReceive,
  43. Flags: []cli.Flag{
  44. &cli.BoolFlag{
  45. Name: "debug",
  46. },
  47. },
  48. }
  49. subcmdHookUpdate = &cli.Command{
  50. Name: "update",
  51. Usage: "Delegate update Git hook",
  52. Description: "This command should only be called by Git",
  53. Action: runHookUpdate,
  54. Flags: []cli.Flag{
  55. &cli.BoolFlag{
  56. Name: "debug",
  57. },
  58. },
  59. }
  60. subcmdHookPostReceive = &cli.Command{
  61. Name: "post-receive",
  62. Usage: "Delegate post-receive Git hook",
  63. Description: "This command should only be called by Git",
  64. Action: runHookPostReceive,
  65. Flags: []cli.Flag{
  66. &cli.BoolFlag{
  67. Name: "debug",
  68. },
  69. },
  70. }
  71. // Note: new hook since git 2.29
  72. subcmdHookProcReceive = &cli.Command{
  73. Name: "proc-receive",
  74. Usage: "Delegate proc-receive Git hook",
  75. Description: "This command should only be called by Git",
  76. Action: runHookProcReceive,
  77. Flags: []cli.Flag{
  78. &cli.BoolFlag{
  79. Name: "debug",
  80. },
  81. },
  82. }
  83. )
  84. type delayWriter struct {
  85. internal io.Writer
  86. buf *bytes.Buffer
  87. timer *time.Timer
  88. }
  89. func newDelayWriter(internal io.Writer, delay time.Duration) *delayWriter {
  90. timer := time.NewTimer(delay)
  91. return &delayWriter{
  92. internal: internal,
  93. buf: &bytes.Buffer{},
  94. timer: timer,
  95. }
  96. }
  97. func (d *delayWriter) Write(p []byte) (n int, err error) {
  98. if d.buf != nil {
  99. select {
  100. case <-d.timer.C:
  101. _, err := d.internal.Write(d.buf.Bytes())
  102. if err != nil {
  103. return 0, err
  104. }
  105. d.buf = nil
  106. return d.internal.Write(p)
  107. default:
  108. return d.buf.Write(p)
  109. }
  110. }
  111. return d.internal.Write(p)
  112. }
  113. func (d *delayWriter) WriteString(s string) (n int, err error) {
  114. if d.buf != nil {
  115. select {
  116. case <-d.timer.C:
  117. _, err := d.internal.Write(d.buf.Bytes())
  118. if err != nil {
  119. return 0, err
  120. }
  121. d.buf = nil
  122. return d.internal.Write([]byte(s))
  123. default:
  124. return d.buf.WriteString(s)
  125. }
  126. }
  127. return d.internal.Write([]byte(s))
  128. }
  129. func (d *delayWriter) Close() error {
  130. if d == nil {
  131. return nil
  132. }
  133. stopped := d.timer.Stop()
  134. if stopped || d.buf == nil {
  135. return nil
  136. }
  137. _, err := d.internal.Write(d.buf.Bytes())
  138. d.buf = nil
  139. return err
  140. }
  141. type nilWriter struct{}
  142. func (n *nilWriter) Write(p []byte) (int, error) {
  143. return len(p), nil
  144. }
  145. func (n *nilWriter) WriteString(s string) (int, error) {
  146. return len(s), nil
  147. }
  148. func runHookPreReceive(c *cli.Context) error {
  149. if isInternal, _ := strconv.ParseBool(os.Getenv(repo_module.EnvIsInternal)); isInternal {
  150. return nil
  151. }
  152. ctx, cancel := installSignals()
  153. defer cancel()
  154. setup(ctx, c.Bool("debug"))
  155. if len(os.Getenv("SSH_ORIGINAL_COMMAND")) == 0 {
  156. if setting.OnlyAllowPushIfGiteaEnvironmentSet {
  157. return fail(ctx, `Rejecting changes as Gitea environment not set.
  158. If you are pushing over SSH you must push with a key managed by
  159. Gitea or set your environment appropriately.`, "")
  160. }
  161. return nil
  162. }
  163. // the environment is set by serv command
  164. isWiki, _ := strconv.ParseBool(os.Getenv(repo_module.EnvRepoIsWiki))
  165. username := os.Getenv(repo_module.EnvRepoUsername)
  166. reponame := os.Getenv(repo_module.EnvRepoName)
  167. userID, _ := strconv.ParseInt(os.Getenv(repo_module.EnvPusherID), 10, 64)
  168. prID, _ := strconv.ParseInt(os.Getenv(repo_module.EnvPRID), 10, 64)
  169. deployKeyID, _ := strconv.ParseInt(os.Getenv(repo_module.EnvDeployKeyID), 10, 64)
  170. actionPerm, _ := strconv.ParseInt(os.Getenv(repo_module.EnvActionPerm), 10, 64)
  171. hookOptions := private.HookOptions{
  172. UserID: userID,
  173. GitAlternativeObjectDirectories: os.Getenv(private.GitAlternativeObjectDirectories),
  174. GitObjectDirectory: os.Getenv(private.GitObjectDirectory),
  175. GitQuarantinePath: os.Getenv(private.GitQuarantinePath),
  176. GitPushOptions: pushOptions(),
  177. PullRequestID: prID,
  178. DeployKeyID: deployKeyID,
  179. ActionPerm: int(actionPerm),
  180. }
  181. scanner := bufio.NewScanner(os.Stdin)
  182. oldCommitIDs := make([]string, hookBatchSize)
  183. newCommitIDs := make([]string, hookBatchSize)
  184. refFullNames := make([]git.RefName, hookBatchSize)
  185. count := 0
  186. total := 0
  187. lastline := 0
  188. var out io.Writer
  189. out = &nilWriter{}
  190. if setting.Git.VerbosePush {
  191. if setting.Git.VerbosePushDelay > 0 {
  192. dWriter := newDelayWriter(os.Stdout, setting.Git.VerbosePushDelay)
  193. defer dWriter.Close()
  194. out = dWriter
  195. } else {
  196. out = os.Stdout
  197. }
  198. }
  199. supportProcReceive := false
  200. if git.CheckGitVersionAtLeast("2.29") == nil {
  201. supportProcReceive = true
  202. }
  203. for scanner.Scan() {
  204. // TODO: support news feeds for wiki
  205. if isWiki {
  206. continue
  207. }
  208. fields := bytes.Fields(scanner.Bytes())
  209. if len(fields) != 3 {
  210. continue
  211. }
  212. oldCommitID := string(fields[0])
  213. newCommitID := string(fields[1])
  214. refFullName := git.RefName(fields[2])
  215. total++
  216. lastline++
  217. // If the ref is a branch or tag, check if it's protected
  218. // if supportProcReceive all ref should be checked because
  219. // permission check was delayed
  220. if supportProcReceive || refFullName.IsBranch() || refFullName.IsTag() {
  221. oldCommitIDs[count] = oldCommitID
  222. newCommitIDs[count] = newCommitID
  223. refFullNames[count] = refFullName
  224. count++
  225. fmt.Fprintf(out, "*")
  226. if count >= hookBatchSize {
  227. fmt.Fprintf(out, " Checking %d references\n", count)
  228. hookOptions.OldCommitIDs = oldCommitIDs
  229. hookOptions.NewCommitIDs = newCommitIDs
  230. hookOptions.RefFullNames = refFullNames
  231. extra := private.HookPreReceive(ctx, username, reponame, hookOptions)
  232. if extra.HasError() {
  233. return fail(ctx, extra.UserMsg, "HookPreReceive(batch) failed: %v", extra.Error)
  234. }
  235. count = 0
  236. lastline = 0
  237. }
  238. } else {
  239. fmt.Fprintf(out, ".")
  240. }
  241. if lastline >= hookBatchSize {
  242. fmt.Fprintf(out, "\n")
  243. lastline = 0
  244. }
  245. }
  246. if count > 0 {
  247. hookOptions.OldCommitIDs = oldCommitIDs[:count]
  248. hookOptions.NewCommitIDs = newCommitIDs[:count]
  249. hookOptions.RefFullNames = refFullNames[:count]
  250. fmt.Fprintf(out, " Checking %d references\n", count)
  251. extra := private.HookPreReceive(ctx, username, reponame, hookOptions)
  252. if extra.HasError() {
  253. return fail(ctx, extra.UserMsg, "HookPreReceive(last) failed: %v", extra.Error)
  254. }
  255. } else if lastline > 0 {
  256. fmt.Fprintf(out, "\n")
  257. }
  258. fmt.Fprintf(out, "Checked %d references in total\n", total)
  259. return nil
  260. }
  261. func runHookUpdate(c *cli.Context) error {
  262. // Update is empty and is kept only for backwards compatibility
  263. return nil
  264. }
  265. func runHookPostReceive(c *cli.Context) error {
  266. ctx, cancel := installSignals()
  267. defer cancel()
  268. setup(ctx, c.Bool("debug"))
  269. // First of all run update-server-info no matter what
  270. if _, _, err := git.NewCommand(ctx, "update-server-info").RunStdString(nil); err != nil {
  271. return fmt.Errorf("Failed to call 'git update-server-info': %w", err)
  272. }
  273. // Now if we're an internal don't do anything else
  274. if isInternal, _ := strconv.ParseBool(os.Getenv(repo_module.EnvIsInternal)); isInternal {
  275. return nil
  276. }
  277. if len(os.Getenv("SSH_ORIGINAL_COMMAND")) == 0 {
  278. if setting.OnlyAllowPushIfGiteaEnvironmentSet {
  279. return fail(ctx, `Rejecting changes as Gitea environment not set.
  280. If you are pushing over SSH you must push with a key managed by
  281. Gitea or set your environment appropriately.`, "")
  282. }
  283. return nil
  284. }
  285. var out io.Writer
  286. var dWriter *delayWriter
  287. out = &nilWriter{}
  288. if setting.Git.VerbosePush {
  289. if setting.Git.VerbosePushDelay > 0 {
  290. dWriter = newDelayWriter(os.Stdout, setting.Git.VerbosePushDelay)
  291. defer dWriter.Close()
  292. out = dWriter
  293. } else {
  294. out = os.Stdout
  295. }
  296. }
  297. // the environment is set by serv command
  298. repoUser := os.Getenv(repo_module.EnvRepoUsername)
  299. isWiki, _ := strconv.ParseBool(os.Getenv(repo_module.EnvRepoIsWiki))
  300. repoName := os.Getenv(repo_module.EnvRepoName)
  301. pusherID, _ := strconv.ParseInt(os.Getenv(repo_module.EnvPusherID), 10, 64)
  302. pusherName := os.Getenv(repo_module.EnvPusherName)
  303. hookOptions := private.HookOptions{
  304. UserName: pusherName,
  305. UserID: pusherID,
  306. GitAlternativeObjectDirectories: os.Getenv(private.GitAlternativeObjectDirectories),
  307. GitObjectDirectory: os.Getenv(private.GitObjectDirectory),
  308. GitQuarantinePath: os.Getenv(private.GitQuarantinePath),
  309. GitPushOptions: pushOptions(),
  310. }
  311. oldCommitIDs := make([]string, hookBatchSize)
  312. newCommitIDs := make([]string, hookBatchSize)
  313. refFullNames := make([]git.RefName, hookBatchSize)
  314. count := 0
  315. total := 0
  316. wasEmpty := false
  317. masterPushed := false
  318. results := make([]private.HookPostReceiveBranchResult, 0)
  319. scanner := bufio.NewScanner(os.Stdin)
  320. for scanner.Scan() {
  321. // TODO: support news feeds for wiki
  322. if isWiki {
  323. continue
  324. }
  325. fields := bytes.Fields(scanner.Bytes())
  326. if len(fields) != 3 {
  327. continue
  328. }
  329. fmt.Fprintf(out, ".")
  330. oldCommitIDs[count] = string(fields[0])
  331. newCommitIDs[count] = string(fields[1])
  332. refFullNames[count] = git.RefName(fields[2])
  333. commitID, _ := git.NewIDFromString(newCommitIDs[count])
  334. if refFullNames[count] == git.BranchPrefix+"master" && !commitID.IsZero() && count == total {
  335. masterPushed = true
  336. }
  337. count++
  338. total++
  339. if count >= hookBatchSize {
  340. fmt.Fprintf(out, " Processing %d references\n", count)
  341. hookOptions.OldCommitIDs = oldCommitIDs
  342. hookOptions.NewCommitIDs = newCommitIDs
  343. hookOptions.RefFullNames = refFullNames
  344. resp, extra := private.HookPostReceive(ctx, repoUser, repoName, hookOptions)
  345. if extra.HasError() {
  346. _ = dWriter.Close()
  347. hookPrintResults(results)
  348. return fail(ctx, extra.UserMsg, "HookPostReceive failed: %v", extra.Error)
  349. }
  350. wasEmpty = wasEmpty || resp.RepoWasEmpty
  351. results = append(results, resp.Results...)
  352. count = 0
  353. }
  354. }
  355. if count == 0 {
  356. if wasEmpty && masterPushed {
  357. // We need to tell the repo to reset the default branch to master
  358. extra := private.SetDefaultBranch(ctx, repoUser, repoName, "master")
  359. if extra.HasError() {
  360. return fail(ctx, extra.UserMsg, "SetDefaultBranch failed: %v", extra.Error)
  361. }
  362. }
  363. fmt.Fprintf(out, "Processed %d references in total\n", total)
  364. _ = dWriter.Close()
  365. hookPrintResults(results)
  366. return nil
  367. }
  368. hookOptions.OldCommitIDs = oldCommitIDs[:count]
  369. hookOptions.NewCommitIDs = newCommitIDs[:count]
  370. hookOptions.RefFullNames = refFullNames[:count]
  371. fmt.Fprintf(out, " Processing %d references\n", count)
  372. resp, extra := private.HookPostReceive(ctx, repoUser, repoName, hookOptions)
  373. if resp == nil {
  374. _ = dWriter.Close()
  375. hookPrintResults(results)
  376. return fail(ctx, extra.UserMsg, "HookPostReceive failed: %v", extra.Error)
  377. }
  378. wasEmpty = wasEmpty || resp.RepoWasEmpty
  379. results = append(results, resp.Results...)
  380. fmt.Fprintf(out, "Processed %d references in total\n", total)
  381. if wasEmpty && masterPushed {
  382. // We need to tell the repo to reset the default branch to master
  383. extra := private.SetDefaultBranch(ctx, repoUser, repoName, "master")
  384. if extra.HasError() {
  385. return fail(ctx, extra.UserMsg, "SetDefaultBranch failed: %v", extra.Error)
  386. }
  387. }
  388. _ = dWriter.Close()
  389. hookPrintResults(results)
  390. return nil
  391. }
  392. func hookPrintResults(results []private.HookPostReceiveBranchResult) {
  393. for _, res := range results {
  394. hookPrintResult(res.Message, res.Create, res.Branch, res.URL)
  395. }
  396. }
  397. func hookPrintResult(output, isCreate bool, branch, url string) {
  398. if !output {
  399. return
  400. }
  401. fmt.Fprintln(os.Stderr, "")
  402. if isCreate {
  403. fmt.Fprintf(os.Stderr, "Create a new pull request for '%s':\n", branch)
  404. fmt.Fprintf(os.Stderr, " %s\n", url)
  405. } else {
  406. fmt.Fprint(os.Stderr, "Visit the existing pull request:\n")
  407. fmt.Fprintf(os.Stderr, " %s\n", url)
  408. }
  409. fmt.Fprintln(os.Stderr, "")
  410. _ = os.Stderr.Sync()
  411. }
  412. func pushOptions() map[string]string {
  413. opts := make(map[string]string)
  414. if pushCount, err := strconv.Atoi(os.Getenv(private.GitPushOptionCount)); err == nil {
  415. for idx := 0; idx < pushCount; idx++ {
  416. opt := os.Getenv(fmt.Sprintf("GIT_PUSH_OPTION_%d", idx))
  417. kv := strings.SplitN(opt, "=", 2)
  418. if len(kv) == 2 {
  419. opts[kv[0]] = kv[1]
  420. }
  421. }
  422. }
  423. return opts
  424. }
  425. func runHookProcReceive(c *cli.Context) error {
  426. ctx, cancel := installSignals()
  427. defer cancel()
  428. setup(ctx, c.Bool("debug"))
  429. if len(os.Getenv("SSH_ORIGINAL_COMMAND")) == 0 {
  430. if setting.OnlyAllowPushIfGiteaEnvironmentSet {
  431. return fail(ctx, `Rejecting changes as Gitea environment not set.
  432. If you are pushing over SSH you must push with a key managed by
  433. Gitea or set your environment appropriately.`, "")
  434. }
  435. return nil
  436. }
  437. if git.CheckGitVersionAtLeast("2.29") != nil {
  438. return fail(ctx, "No proc-receive support", "current git version doesn't support proc-receive.")
  439. }
  440. reader := bufio.NewReader(os.Stdin)
  441. repoUser := os.Getenv(repo_module.EnvRepoUsername)
  442. repoName := os.Getenv(repo_module.EnvRepoName)
  443. pusherID, _ := strconv.ParseInt(os.Getenv(repo_module.EnvPusherID), 10, 64)
  444. pusherName := os.Getenv(repo_module.EnvPusherName)
  445. // 1. Version and features negotiation.
  446. // S: PKT-LINE(version=1\0push-options atomic...) / PKT-LINE(version=1\n)
  447. // S: flush-pkt
  448. // H: PKT-LINE(version=1\0push-options...)
  449. // H: flush-pkt
  450. rs, err := readPktLine(ctx, reader, pktLineTypeData)
  451. if err != nil {
  452. return err
  453. }
  454. const VersionHead string = "version=1"
  455. var (
  456. hasPushOptions bool
  457. response = []byte(VersionHead)
  458. requestOptions []string
  459. )
  460. index := bytes.IndexByte(rs.Data, byte(0))
  461. if index >= len(rs.Data) {
  462. return fail(ctx, "Protocol: format error", "pkt-line: format error "+fmt.Sprint(rs.Data))
  463. }
  464. if index < 0 {
  465. if len(rs.Data) == 10 && rs.Data[9] == '\n' {
  466. index = 9
  467. } else {
  468. return fail(ctx, "Protocol: format error", "pkt-line: format error "+fmt.Sprint(rs.Data))
  469. }
  470. }
  471. if string(rs.Data[0:index]) != VersionHead {
  472. return fail(ctx, "Protocol: version error", "Received unsupported version: %s", string(rs.Data[0:index]))
  473. }
  474. requestOptions = strings.Split(string(rs.Data[index+1:]), " ")
  475. for _, option := range requestOptions {
  476. if strings.HasPrefix(option, "push-options") {
  477. response = append(response, byte(0))
  478. response = append(response, []byte("push-options")...)
  479. hasPushOptions = true
  480. }
  481. }
  482. response = append(response, '\n')
  483. _, err = readPktLine(ctx, reader, pktLineTypeFlush)
  484. if err != nil {
  485. return err
  486. }
  487. err = writeDataPktLine(ctx, os.Stdout, response)
  488. if err != nil {
  489. return err
  490. }
  491. err = writeFlushPktLine(ctx, os.Stdout)
  492. if err != nil {
  493. return err
  494. }
  495. // 2. receive commands from server.
  496. // S: PKT-LINE(<old-oid> <new-oid> <ref>)
  497. // S: ... ...
  498. // S: flush-pkt
  499. // # [receive push-options]
  500. // S: PKT-LINE(push-option)
  501. // S: ... ...
  502. // S: flush-pkt
  503. hookOptions := private.HookOptions{
  504. UserName: pusherName,
  505. UserID: pusherID,
  506. }
  507. hookOptions.OldCommitIDs = make([]string, 0, hookBatchSize)
  508. hookOptions.NewCommitIDs = make([]string, 0, hookBatchSize)
  509. hookOptions.RefFullNames = make([]git.RefName, 0, hookBatchSize)
  510. for {
  511. // note: pktLineTypeUnknow means pktLineTypeFlush and pktLineTypeData all allowed
  512. rs, err = readPktLine(ctx, reader, pktLineTypeUnknow)
  513. if err != nil {
  514. return err
  515. }
  516. if rs.Type == pktLineTypeFlush {
  517. break
  518. }
  519. t := strings.SplitN(string(rs.Data), " ", 3)
  520. if len(t) != 3 {
  521. continue
  522. }
  523. hookOptions.OldCommitIDs = append(hookOptions.OldCommitIDs, t[0])
  524. hookOptions.NewCommitIDs = append(hookOptions.NewCommitIDs, t[1])
  525. hookOptions.RefFullNames = append(hookOptions.RefFullNames, git.RefName(t[2]))
  526. }
  527. hookOptions.GitPushOptions = make(map[string]string)
  528. if hasPushOptions {
  529. for {
  530. rs, err = readPktLine(ctx, reader, pktLineTypeUnknow)
  531. if err != nil {
  532. return err
  533. }
  534. if rs.Type == pktLineTypeFlush {
  535. break
  536. }
  537. kv := strings.SplitN(string(rs.Data), "=", 2)
  538. if len(kv) == 2 {
  539. hookOptions.GitPushOptions[kv[0]] = kv[1]
  540. }
  541. }
  542. }
  543. // 3. run hook
  544. resp, extra := private.HookProcReceive(ctx, repoUser, repoName, hookOptions)
  545. if extra.HasError() {
  546. return fail(ctx, extra.UserMsg, "HookProcReceive failed: %v", extra.Error)
  547. }
  548. // 4. response result to service
  549. // # a. OK, but has an alternate reference. The alternate reference name
  550. // # and other status can be given in option directives.
  551. // H: PKT-LINE(ok <ref>)
  552. // H: PKT-LINE(option refname <refname>)
  553. // H: PKT-LINE(option old-oid <old-oid>)
  554. // H: PKT-LINE(option new-oid <new-oid>)
  555. // H: PKT-LINE(option forced-update)
  556. // H: ... ...
  557. // H: flush-pkt
  558. // # b. NO, I reject it.
  559. // H: PKT-LINE(ng <ref> <reason>)
  560. // # c. Fall through, let 'receive-pack' to execute it.
  561. // H: PKT-LINE(ok <ref>)
  562. // H: PKT-LINE(option fall-through)
  563. for _, rs := range resp.Results {
  564. if len(rs.Err) > 0 {
  565. err = writeDataPktLine(ctx, os.Stdout, []byte("ng "+rs.OriginalRef.String()+" "+rs.Err))
  566. if err != nil {
  567. return err
  568. }
  569. continue
  570. }
  571. if rs.IsNotMatched {
  572. err = writeDataPktLine(ctx, os.Stdout, []byte("ok "+rs.OriginalRef.String()))
  573. if err != nil {
  574. return err
  575. }
  576. err = writeDataPktLine(ctx, os.Stdout, []byte("option fall-through"))
  577. if err != nil {
  578. return err
  579. }
  580. continue
  581. }
  582. err = writeDataPktLine(ctx, os.Stdout, []byte("ok "+rs.OriginalRef))
  583. if err != nil {
  584. return err
  585. }
  586. err = writeDataPktLine(ctx, os.Stdout, []byte("option refname "+rs.Ref))
  587. if err != nil {
  588. return err
  589. }
  590. commitID, _ := git.NewIDFromString(rs.OldOID)
  591. if !commitID.IsZero() {
  592. err = writeDataPktLine(ctx, os.Stdout, []byte("option old-oid "+rs.OldOID))
  593. if err != nil {
  594. return err
  595. }
  596. }
  597. err = writeDataPktLine(ctx, os.Stdout, []byte("option new-oid "+rs.NewOID))
  598. if err != nil {
  599. return err
  600. }
  601. if rs.IsForcePush {
  602. err = writeDataPktLine(ctx, os.Stdout, []byte("option forced-update"))
  603. if err != nil {
  604. return err
  605. }
  606. }
  607. }
  608. err = writeFlushPktLine(ctx, os.Stdout)
  609. if err == nil {
  610. for _, res := range resp.Results {
  611. hookPrintResult(res.ShouldShowMessage, res.IsCreatePR, res.HeadBranch, res.URL)
  612. }
  613. }
  614. return err
  615. }
  616. // git PKT-Line api
  617. // pktLineType message type of pkt-line
  618. type pktLineType int64
  619. const (
  620. // UnKnow type
  621. pktLineTypeUnknow pktLineType = 0
  622. // flush-pkt "0000"
  623. pktLineTypeFlush pktLineType = iota
  624. // data line
  625. pktLineTypeData
  626. )
  627. // gitPktLine pkt-line api
  628. type gitPktLine struct {
  629. Type pktLineType
  630. Length uint64
  631. Data []byte
  632. }
  633. func readPktLine(ctx context.Context, in *bufio.Reader, requestType pktLineType) (*gitPktLine, error) {
  634. var (
  635. err error
  636. r *gitPktLine
  637. )
  638. // read prefix
  639. lengthBytes := make([]byte, 4)
  640. for i := 0; i < 4; i++ {
  641. lengthBytes[i], err = in.ReadByte()
  642. if err != nil {
  643. return nil, fail(ctx, "Protocol: stdin error", "Pkt-Line: read stdin failed : %v", err)
  644. }
  645. }
  646. r = new(gitPktLine)
  647. r.Length, err = strconv.ParseUint(string(lengthBytes), 16, 32)
  648. if err != nil {
  649. return nil, fail(ctx, "Protocol: format parse error", "Pkt-Line format is wrong :%v", err)
  650. }
  651. if r.Length == 0 {
  652. if requestType == pktLineTypeData {
  653. return nil, fail(ctx, "Protocol: format data error", "Pkt-Line format is wrong")
  654. }
  655. r.Type = pktLineTypeFlush
  656. return r, nil
  657. }
  658. if r.Length <= 4 || r.Length > 65520 || requestType == pktLineTypeFlush {
  659. return nil, fail(ctx, "Protocol: format length error", "Pkt-Line format is wrong")
  660. }
  661. r.Data = make([]byte, r.Length-4)
  662. for i := range r.Data {
  663. r.Data[i], err = in.ReadByte()
  664. if err != nil {
  665. return nil, fail(ctx, "Protocol: data error", "Pkt-Line: read stdin failed : %v", err)
  666. }
  667. }
  668. r.Type = pktLineTypeData
  669. return r, nil
  670. }
  671. func writeFlushPktLine(ctx context.Context, out io.Writer) error {
  672. l, err := out.Write([]byte("0000"))
  673. if err != nil || l != 4 {
  674. return fail(ctx, "Protocol: write error", "Pkt-Line response failed: %v", err)
  675. }
  676. return nil
  677. }
  678. func writeDataPktLine(ctx context.Context, out io.Writer, data []byte) error {
  679. hexchar := []byte("0123456789abcdef")
  680. hex := func(n uint64) byte {
  681. return hexchar[(n)&15]
  682. }
  683. length := uint64(len(data) + 4)
  684. tmp := make([]byte, 4)
  685. tmp[0] = hex(length >> 12)
  686. tmp[1] = hex(length >> 8)
  687. tmp[2] = hex(length >> 4)
  688. tmp[3] = hex(length)
  689. lr, err := out.Write(tmp)
  690. if err != nil || lr != 4 {
  691. return fail(ctx, "Protocol: write error", "Pkt-Line response failed: %v", err)
  692. }
  693. lr, err = out.Write(data)
  694. if err != nil || int(length-4) != lr {
  695. return fail(ctx, "Protocol: write error", "Pkt-Line response failed: %v", err)
  696. }
  697. return nil
  698. }