You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
133 lines
3.1 KiB
133 lines
3.1 KiB
#include "cache.h" |
|
#include "pkt-line.h" |
|
#include "sideband.h" |
|
|
|
/* |
|
* Receive multiplexed output stream over git native protocol. |
|
* in_stream is the input stream from the remote, which carries data |
|
* in pkt_line format with band designator. Demultiplex it into out |
|
* and err and return error appropriately. Band #1 carries the |
|
* primary payload. Things coming over band #2 is not necessarily |
|
* error; they are usually informative message on the standard error |
|
* stream, aka "verbose"). A message over band #3 is a signal that |
|
* the remote died unexpectedly. A flush() concludes the stream. |
|
*/ |
|
|
|
#define PREFIX "remote: " |
|
|
|
#define ANSI_SUFFIX "\033[K" |
|
#define DUMB_SUFFIX " " |
|
|
|
int recv_sideband(const char *me, int in_stream, int out) |
|
{ |
|
const char *suffix; |
|
char buf[LARGE_PACKET_MAX + 1]; |
|
struct strbuf outbuf = STRBUF_INIT; |
|
int retval = 0; |
|
|
|
if (isatty(2) && !is_terminal_dumb()) |
|
suffix = ANSI_SUFFIX; |
|
else |
|
suffix = DUMB_SUFFIX; |
|
|
|
while (!retval) { |
|
const char *b, *brk; |
|
int band, len; |
|
len = packet_read(in_stream, NULL, NULL, buf, LARGE_PACKET_MAX, 0); |
|
if (len == 0) |
|
break; |
|
if (len < 1) { |
|
strbuf_addf(&outbuf, |
|
"%s%s: protocol error: no band designator", |
|
outbuf.len ? "\n" : "", me); |
|
retval = SIDEBAND_PROTOCOL_ERROR; |
|
break; |
|
} |
|
band = buf[0] & 0xff; |
|
buf[len] = '\0'; |
|
len--; |
|
switch (band) { |
|
case 3: |
|
strbuf_addf(&outbuf, "%s%s%s", outbuf.len ? "\n" : "", |
|
PREFIX, buf + 1); |
|
retval = SIDEBAND_REMOTE_ERROR; |
|
break; |
|
case 2: |
|
b = buf + 1; |
|
|
|
/* |
|
* Append a suffix to each nonempty line to clear the |
|
* end of the screen line. |
|
* |
|
* The output is accumulated in a buffer and |
|
* each line is printed to stderr using |
|
* write(2) to ensure inter-process atomicity. |
|
*/ |
|
while ((brk = strpbrk(b, "\n\r"))) { |
|
int linelen = brk - b; |
|
|
|
if (!outbuf.len) |
|
strbuf_addstr(&outbuf, PREFIX); |
|
if (linelen > 0) { |
|
strbuf_addf(&outbuf, "%.*s%s%c", |
|
linelen, b, suffix, *brk); |
|
} else { |
|
strbuf_addch(&outbuf, *brk); |
|
} |
|
xwrite(2, outbuf.buf, outbuf.len); |
|
strbuf_reset(&outbuf); |
|
|
|
b = brk + 1; |
|
} |
|
|
|
if (*b) |
|
strbuf_addf(&outbuf, "%s%s", |
|
outbuf.len ? "" : PREFIX, b); |
|
break; |
|
case 1: |
|
write_or_die(out, buf + 1, len); |
|
break; |
|
default: |
|
strbuf_addf(&outbuf, "%s%s: protocol error: bad band #%d", |
|
outbuf.len ? "\n" : "", me, band); |
|
retval = SIDEBAND_PROTOCOL_ERROR; |
|
break; |
|
} |
|
} |
|
|
|
if (outbuf.len) { |
|
strbuf_addch(&outbuf, '\n'); |
|
xwrite(2, outbuf.buf, outbuf.len); |
|
} |
|
strbuf_release(&outbuf); |
|
return retval; |
|
} |
|
|
|
/* |
|
* fd is connected to the remote side; send the sideband data |
|
* over multiplexed packet stream. |
|
*/ |
|
void send_sideband(int fd, int band, const char *data, ssize_t sz, int packet_max) |
|
{ |
|
const char *p = data; |
|
|
|
while (sz) { |
|
unsigned n; |
|
char hdr[5]; |
|
|
|
n = sz; |
|
if (packet_max - 5 < n) |
|
n = packet_max - 5; |
|
if (0 <= band) { |
|
xsnprintf(hdr, sizeof(hdr), "%04x", n + 5); |
|
hdr[4] = band; |
|
write_or_die(fd, hdr, 5); |
|
} else { |
|
xsnprintf(hdr, sizeof(hdr), "%04x", n + 4); |
|
write_or_die(fd, hdr, 4); |
|
} |
|
write_or_die(fd, p, n); |
|
p += n; |
|
sz -= n; |
|
} |
|
}
|
|
|