bcachefs: bch2_stdio_redirect_readline_timeout()

Signed-off-by: Kent Overstreet <kent.overstreet@linux.dev>
This commit is contained in:
Kent Overstreet 2024-05-29 20:37:39 -04:00
parent 0c97c437e3
commit d37dd9b604
2 changed files with 23 additions and 3 deletions

View File

@ -364,13 +364,21 @@ int bch2_stdio_redirect_read(struct stdio_redirect *stdio, char *ubuf, size_t le
return ret;
}
int bch2_stdio_redirect_readline(struct stdio_redirect *stdio, darray_char *line)
int bch2_stdio_redirect_readline_timeout(struct stdio_redirect *stdio,
darray_char *line,
unsigned long timeout)
{
unsigned long until = jiffies + timeout, t;
struct stdio_buf *buf = &stdio->input;
size_t seen = 0;
again:
wait_event_timeout(buf->wait, stdio_redirect_has_more_input(stdio, seen),
sysctl_hung_task_timeout_secs * HZ / 2);
t = timeout != MAX_SCHEDULE_TIMEOUT
? max_t(long, until - jiffies, 0)
: timeout;
t = min(t, sysctl_hung_task_timeout_secs * HZ / 2);
wait_event_timeout(buf->wait, stdio_redirect_has_more_input(stdio, seen), t);
if (stdio->done)
return -1;
@ -378,6 +386,12 @@ int bch2_stdio_redirect_readline(struct stdio_redirect *stdio, darray_char *line
spin_lock(&buf->lock);
seen = buf->buf.nr;
char *n = memchr(buf->buf.data, '\n', seen);
if (!n && timeout != MAX_SCHEDULE_TIMEOUT && jiffies >= until) {
spin_unlock(&buf->lock);
return -ETIME;
}
if (!n) {
buf->waiting_for_line = true;
spin_unlock(&buf->lock);
@ -408,6 +422,11 @@ int bch2_stdio_redirect_readline(struct stdio_redirect *stdio, darray_char *line
return 0;
}
int bch2_stdio_redirect_readline(struct stdio_redirect *stdio, darray_char *line)
{
return bch2_stdio_redirect_readline_timeout(stdio, line, MAX_SCHEDULE_TIMEOUT);
}
__printf(3, 0)
static ssize_t bch2_darray_vprintf(darray_char *out, gfp_t gfp, const char *fmt, va_list args)
{

View File

@ -72,6 +72,7 @@ int bch2_run_thread_with_stdout(struct thread_with_stdio *,
const struct thread_with_stdio_ops *);
int bch2_stdio_redirect_read(struct stdio_redirect *, char *, size_t);
int bch2_stdio_redirect_readline_timeout(struct stdio_redirect *, darray_char *, unsigned long);
int bch2_stdio_redirect_readline(struct stdio_redirect *, darray_char *);
__printf(3, 0) ssize_t bch2_stdio_redirect_vprintf(struct stdio_redirect *, bool, const char *, va_list);