Introduce nvim namespace: Move files.

Move files from src/ to src/nvim/.
- src/nvim/ becomes the new root dir for nvim executable sources.
- src/libnvim/ is planned to become root dir of the neovim library.
This commit is contained in:
Eliseo Martínez
2014-05-12 02:25:17 +02:00
parent ffe61e5ba1
commit da51dc9cf2
463 changed files with 0 additions and 0 deletions

243
src/nvim/os/server.c Normal file
View File

@@ -0,0 +1,243 @@
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include <stdbool.h>
#include <uv.h>
#include "os/channel_defs.h"
#include "os/channel.h"
#include "os/server.h"
#include "os/os.h"
#include "vim.h"
#include "memory.h"
#include "message.h"
#include "fileio.h"
#include "map.h"
#define MAX_CONNECTIONS 32
#define ADDRESS_MAX_SIZE 256
#define NEOVIM_DEFAULT_TCP_PORT 7450
typedef enum {
kServerTypeTcp,
kServerTypePipe
} ServerType;
typedef struct {
// Protocol for channels established through this server
ChannelProtocol protocol;
// Type of the union below
ServerType type;
// This is either a tcp server or unix socket(named pipe on windows)
union {
struct {
uv_tcp_t handle;
struct sockaddr_in addr;
} tcp;
struct {
uv_pipe_t handle;
char addr[ADDRESS_MAX_SIZE];
} pipe;
} socket;
} Server;
static Map *servers = NULL;
static void close_server(Map *map, const char *endpoint, void *server);
static void connection_cb(uv_stream_t *server, int status);
static void free_client(uv_handle_t *handle);
static void free_server(uv_handle_t *handle);
void server_init()
{
servers = map_new();
if (!os_getenv("NEOVIM_LISTEN_ADDRESS")) {
char *listen_address = (char *)vim_tempname('s');
os_setenv("NEOVIM_LISTEN_ADDRESS", listen_address, 1);
free(listen_address);
}
server_start((char *)os_getenv("NEOVIM_LISTEN_ADDRESS"),
kChannelProtocolMsgpack);
}
void server_teardown()
{
if (!servers) {
return;
}
map_foreach(servers, close_server);
}
void server_start(char *endpoint, ChannelProtocol prot)
{
char addr[ADDRESS_MAX_SIZE];
// Trim to `ADDRESS_MAX_SIZE`
strncpy(addr, endpoint, sizeof(addr));
// Check if the server already exists
if (map_has(servers, addr)) {
EMSG2("Already listening on %s", addr);
return;
}
ServerType server_type = kServerTypeTcp;
Server *server = xmalloc(sizeof(Server));
char ip[16], *ip_end = strrchr(addr, ':');
server->protocol = prot;
if (!ip_end) {
ip_end = strchr(addr, NUL);
}
uint32_t addr_len = ip_end - addr;
if (addr_len > sizeof(ip) - 1) {
// Maximum length of a ip address buffer is 15(eg: 255.255.255.255)
addr_len = sizeof(ip);
}
// Extract the address part
strncpy(ip, addr, addr_len);
int port = NEOVIM_DEFAULT_TCP_PORT;
if (*ip_end == ':') {
char *port_end;
// Extract the port
port = strtol(ip_end + 1, &port_end, 10);
errno = 0;
if (errno != 0 || port == 0 || port > 0xffff) {
// Invalid port, treat as named pipe or unix socket
server_type = kServerTypePipe;
}
}
if (server_type == kServerTypeTcp) {
// Try to parse ip address
if (uv_ip4_addr(ip, port, &server->socket.tcp.addr)) {
// Invalid address, treat as named pipe or unix socket
server_type = kServerTypePipe;
}
}
int result;
if (server_type == kServerTypeTcp) {
// Listen on tcp address/port
uv_tcp_init(uv_default_loop(), &server->socket.tcp.handle);
server->socket.tcp.handle.data = server;
uv_tcp_bind(&server->socket.tcp.handle,
(const struct sockaddr *)&server->socket.tcp.addr,
0);
result = uv_listen((uv_stream_t *)&server->socket.tcp.handle,
MAX_CONNECTIONS,
connection_cb);
if (result) {
uv_close((uv_handle_t *)&server->socket.tcp.handle, free_server);
}
} else {
// Listen on named pipe or unix socket
strcpy(server->socket.pipe.addr, addr);
uv_pipe_init(uv_default_loop(), &server->socket.pipe.handle, 0);
server->socket.pipe.handle.data = server;
uv_pipe_bind(&server->socket.pipe.handle, server->socket.pipe.addr);
result = uv_listen((uv_stream_t *)&server->socket.pipe.handle,
MAX_CONNECTIONS,
connection_cb);
if (result) {
uv_close((uv_handle_t *)&server->socket.pipe.handle, free_server);
}
}
if (result) {
EMSG2("Failed to start server: %s", uv_strerror(result));
return;
}
server->type = server_type;
// Add the server to the hash table
map_put(servers, addr, server);
}
void server_stop(char *endpoint)
{
Server *server;
char addr[ADDRESS_MAX_SIZE];
// Trim to `ADDRESS_MAX_SIZE`
strncpy(addr, endpoint, sizeof(addr));
if ((server = map_get(servers, addr)) == NULL) {
EMSG2("Not listening on %s", addr);
return;
}
if (server->type == kServerTypeTcp) {
uv_close((uv_handle_t *)&server->socket.tcp.handle, free_server);
} else {
uv_close((uv_handle_t *)&server->socket.pipe.handle, free_server);
}
map_del(servers, addr);
}
static void connection_cb(uv_stream_t *server, int status)
{
int result;
uv_stream_t *client;
Server *srv = server->data;
if (status < 0) {
abort();
}
if (srv->type == kServerTypeTcp) {
client = xmalloc(sizeof(uv_tcp_t));
uv_tcp_init(uv_default_loop(), (uv_tcp_t *)client);
} else {
client = xmalloc(sizeof(uv_pipe_t));
uv_pipe_init(uv_default_loop(), (uv_pipe_t *)client, 0);
}
result = uv_accept(server, client);
if (result) {
EMSG2("Failed to accept connection: %s", uv_strerror(result));
uv_close((uv_handle_t *)client, free_client);
return;
}
channel_from_stream(client, srv->protocol);
}
static void close_server(Map *map, const char *endpoint, void *srv)
{
Server *server = srv;
if (server->type == kServerTypeTcp) {
uv_close((uv_handle_t *)&server->socket.tcp.handle, free_server);
} else {
uv_close((uv_handle_t *)&server->socket.pipe.handle, free_server);
}
}
static void free_client(uv_handle_t *handle)
{
free(handle);
}
static void free_server(uv_handle_t *handle)
{
free(handle->data);
}