🌐 AI搜索 & 代理 主页
Google Git
Sign in
chromium / external / github.com / git / git / refs/tags/v2.4.8 / . / sideband.c
blob: 7f9dc229fbc82d005b5c2b417c2af2fc9f6b497a [file] [log] [blame]
Jeff Kingcdf4fb82013-02-20 20:01:56[diff] [blame]1#include "cache.h"
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]2#include "pkt-line.h"
3#include "sideband.h"
4
5/*
6 * Receive multiplexed output stream over git native protocol.
7 * in_stream is the input stream from the remote, which carries data
8 * in pkt_line format with band designator. Demultiplex it into out
9 * and err and return error appropriately. Band #1 carries the
10 * primary payload. Things coming over band #2 is not necessarily
11 * error; they are usually informative message on the standard error
12 * stream, aka "verbose"). A message over band #3 is a signal that
13 * the remote died unexpectedly. A flush() concludes the stream.
14 */
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]15
16#define PREFIX "remote:"
Johannes Sixt13e47602008-01-08 16:24:53[diff] [blame]17
18#define ANSI_SUFFIX "\033[K"
19#define DUMB_SUFFIX " "
20
21#define FIX_SIZE 10 /* large enough for any of the above */
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]22
Johannes Sixt34df8ab2009-03-10 21:54:17[diff] [blame]23int recv_sideband(const char *me, int in_stream, int out)
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]24{
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]25 unsigned pf = strlen(PREFIX);
Johannes Sixt13e47602008-01-08 16:24:53[diff] [blame]26 unsigned sf;
27 char buf[LARGE_PACKET_MAX + 2*FIX_SIZE];
28 char *suffix, *term;
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]29 int skip_pf = 0;
Johannes Sixt13e47602008-01-08 16:24:53[diff] [blame]30
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]31 memcpy(buf, PREFIX, pf);
Johannes Sixt13e47602008-01-08 16:24:53[diff] [blame]32 term = getenv("TERM");
Michael Naumov38de1562014-05-28 03:12:15[diff] [blame]33 if (isatty(2) && term && strcmp(term, "dumb"))
Johannes Sixt13e47602008-01-08 16:24:53[diff] [blame]34 suffix = ANSI_SUFFIX;
35 else
36 suffix = DUMB_SUFFIX;
37 sf = strlen(suffix);
38
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]39 while (1) {
Nicolas Pitre9ac13ec2006-10-11 15:49:15[diff] [blame]40 int band, len;
Jeff King4981fe72013-02-23 22:31:34[diff] [blame]41 len = packet_read(in_stream, NULL, NULL, buf + pf, LARGE_PACKET_MAX, 0);
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]42 if (len == 0)
43 break;
44 if (len < 1) {
Johannes Sixt34df8ab2009-03-10 21:54:17[diff] [blame]45 fprintf(stderr, "%s: protocol error: no band designator\n", me);
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]46 return SIDEBAND_PROTOCOL_ERROR;
47 }
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]48 band = buf[pf] & 0xff;
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]49 len--;
Nicolas Pitre9ac13ec2006-10-11 15:49:15[diff] [blame]50 switch (band) {
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]51 case 3:
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]52 buf[pf] = ' ';
Johannes Sixt34df8ab2009-03-10 21:54:17[diff] [blame]53 buf[pf+1+len] = '\0';
54 fprintf(stderr, "%s\n", buf);
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]55 return SIDEBAND_REMOTE_ERROR;
56 case 2:
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]57 buf[pf] = ' ';
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]58 do {
59 char *b = buf;
60 int brk = 0;
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]61
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]62 /*
63 * If the last buffer didn't end with a line
64 * break then we should not print a prefix
65 * this time around.
66 */
67 if (skip_pf) {
68 b += pf+1;
69 } else {
70 len += pf+1;
71 brk += pf+1;
72 }
73
74 /* Look for a line break. */
75 for (;;) {
Nicolas Pitreed1902e2007-10-17 01:55:46[diff] [blame]76 brk++;
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]77 if (brk > len) {
78 brk = 0;
79 break;
80 }
81 if (b[brk-1] == '\n' ||
82 b[brk-1] == '\r')
Nicolas Pitreed1902e2007-10-17 01:55:46[diff] [blame]83 break;
84 }
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]85
86 /*
87 * Let's insert a suffix to clear the end
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]88 * of the screen line if a line break was
89 * found. Also, if we don't skip the
90 * prefix, then a non-empty string must be
91 * present too.
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]92 */
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]93 if (brk > (skip_pf ? 0 : (pf+1 + 1))) {
Johannes Sixt13e47602008-01-08 16:24:53[diff] [blame]94 char save[FIX_SIZE];
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]95 memcpy(save, b + brk, sf);
96 b[brk + sf - 1] = b[brk - 1];
97 memcpy(b + brk - 1, suffix, sf);
Johannes Sixt34df8ab2009-03-10 21:54:17[diff] [blame]98 fprintf(stderr, "%.*s", brk + sf, b);
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]99 memcpy(b + brk, save, sf);
100 len -= brk;
101 } else {
102 int l = brk ? brk : len;
Johannes Sixt34df8ab2009-03-10 21:54:17[diff] [blame]103 fprintf(stderr, "%.*s", l, b);
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]104 len -= l;
105 }
Nicolas Pitreebe8fa72007-11-04 04:22:42[diff] [blame]106
Nicolas Pitre6b9c42b2008-09-03 19:13:42[diff] [blame]107 skip_pf = !brk;
108 memmove(buf + pf+1, b + brk, len);
109 } while (len);
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]110 continue;
111 case 1:
Jeff Kingcdf4fb82013-02-20 20:01:56[diff] [blame]112 write_or_die(out, buf + pf+1, len);
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]113 continue;
114 default:
Johannes Sixt34df8ab2009-03-10 21:54:17[diff] [blame]115 fprintf(stderr, "%s: protocol error: bad band #%d\n",
116 me, band);
Junio C Hamano49a52b12006-09-10 08:06:33[diff] [blame]117 return SIDEBAND_PROTOCOL_ERROR;
118 }
119 }
120 return 0;
121}
Junio C Hamano958c24b2006-09-10 10:20:24[diff] [blame]122
123/*
124 * fd is connected to the remote side; send the sideband data
125 * over multiplexed packet stream.
126 */
127ssize_t send_sideband(int fd, int band, const char *data, ssize_t sz, int packet_max)
128{
129 ssize_t ssz = sz;
130 const char *p = data;
131
132 while (sz) {
133 unsigned n;
134 char hdr[5];
135
136 n = sz;
137 if (packet_max - 5 < n)
138 n = packet_max - 5;
Shawn O. Pearcede1a2fd2009-10-31 00:47:41[diff] [blame]139 if (0 <= band) {
140 sprintf(hdr, "%04x", n + 5);
141 hdr[4] = band;
Jeff Kingcdf4fb82013-02-20 20:01:56[diff] [blame]142 write_or_die(fd, hdr, 5);
Shawn O. Pearcede1a2fd2009-10-31 00:47:41[diff] [blame]143 } else {
144 sprintf(hdr, "%04x", n + 4);
Jeff Kingcdf4fb82013-02-20 20:01:56[diff] [blame]145 write_or_die(fd, hdr, 4);
Shawn O. Pearcede1a2fd2009-10-31 00:47:41[diff] [blame]146 }
Jeff Kingcdf4fb82013-02-20 20:01:56[diff] [blame]147 write_or_die(fd, p, n);
Junio C Hamano958c24b2006-09-10 10:20:24[diff] [blame]148 p += n;
149 sz -= n;
150 }
151 return ssz;
152}