stdio.c File Reference

Go to the source code of this file.

Functions
asciichat_error_t	mcp_stdio_reserve (void)
void	mcp_stdio_release (void)
void	mcp_stdio_send (mcp_server_t *s, yyjson_mut_doc *response)
asciichat_error_t	mcp_stdio_run (mcp_server_t *s)

Function Documentation

mcp_stdio_release()

void mcp_stdio_release

(

void

)

Definition at line 13 of file stdio.c.

                             {
  if (protocol_fd >= 0) {
    platform_close(protocol_fd);
    protocol_fd = -1;
  }
}

Referenced by mcp_server_main().

mcp_stdio_reserve()

asciichat_error_t mcp_stdio_reserve

(

void

)

Definition at line 7 of file stdio.c.

                                          {
  protocol_fd = platform_dup(STDOUT_FILENO);
  if (protocol_fd < 0)
    return SET_ERRNO(ERROR_FILE_OPERATION, "Cannot reserve MCP stdout");
  return platform_dup2(STDERR_FILENO, STDOUT_FILENO);
}

References ERROR_FILE_OPERATION, platform_dup(), platform_dup2(), and SET_ERRNO.

Referenced by main().

mcp_stdio_run()

asciichat_error_t mcp_stdio_run

(

mcp_server_t *

s

)

Definition at line 31 of file stdio.c.

                                                 {
  char *line = SAFE_MALLOC(MCP_MAX_INPUT + 1, char *);
  size_t n = 0;
  bool overflow = false;
  while (!atomic_load_bool(&s->stopping)) {
    int c = fgetc(stdin);
    if (c == EOF)
      break;
    if (c != '\n') {
      if (n < MCP_MAX_INPUT)
        line[n++] = (char)c;
      else
        overflow = true;
      continue;
    }
    if (n && line[n - 1] == '\r')
      --n;
    if (!n && !overflow)
      continue;
    yyjson_doc *doc = overflow ? NULL : yyjson_read(line, n, YYJSON_READ_NOFLAG);
    bool pending = false;
    yyjson_mut_doc *response =
        doc ? mcp_dispatch(s, doc, NULL, NULL, &pending)
            : mcp_error(NULL, overflow ? -32600 : -32700, overflow ? "Request exceeds 24 MiB" : "Parse error");
    if (response) {
      mcp_stdio_send(s, response);
      yyjson_mut_doc_free(response);
    }
    if (doc)
      yyjson_doc_free(doc);
    n = 0;
    overflow = false;
    mcp_jobs_reap(s, false);
  }
  SAFE_FREE(line);
  for (unsigned i = 0; i < s->policy.max_jobs; ++i)
    if (s->jobs[i].renderer_job.started)
      http_renderer_job_cancel(&s->jobs[i].renderer_job);
  mcp_jobs_reap(s, true);
  return ASCIICHAT_OK;
}

References ASCIICHAT_OK, http_renderer_job_cancel(), mcp_server::jobs, mcp_policy_t::max_jobs, mcp_dispatch(), mcp_error(), mcp_jobs_reap(), MCP_MAX_INPUT, mcp_stdio_send(), mcp_server::policy, mcp_job::renderer_job, SAFE_FREE, SAFE_MALLOC, http_renderer_job::started, and mcp_server::stopping.

Referenced by mcp_server_main().

mcp_stdio_send()

void mcp_stdio_send	(	mcp_server_t *	s,
		yyjson_mut_doc *	response )

Definition at line 19 of file stdio.c.

                                                               {
  size_t size;
  char *json = yyjson_mut_write(response, 0, &size);
  if (!json)
    return;
  mutex_lock(&s->output_mutex);
  bool ok = platform_write_all(protocol_fd, json, size) == size && platform_write_all(protocol_fd, "\n", 1) == 1;
  mutex_unlock(&s->output_mutex);
  free(json); // yyjson owns allocations made by its default allocator.
  if (!ok)
    atomic_store_bool(&s->stopping, true);
}

References mcp_server::output_mutex, platform_write_all(), and mcp_server::stopping.

Referenced by mcp_stdio_run().