finger.c

来自「一个很有名的浏览器」· C语言 代码 · 共 113 行

/* Internal "finger" protocol implementation */
/* $Id: finger.c,v 1.4 2004/12/19 01:47:17 miciah Exp $ */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "elinks.h"

#include "cache/cache.h"
#include "intl/gettext/libintl.h"
#include "lowlevel/connect.h"
#include "modules/module.h"
#include "protocol/finger/finger.h"
#include "protocol/protocol.h"
#include "protocol/uri.h"
#include "sched/connection.h"
#include "util/memory.h"
#include "util/string.h"

struct module finger_protocol_module = struct_module(
	/* name: */		N_("Finger"),
	/* options: */		NULL,
	/* hooks: */		NULL,
	/* submodules: */	NULL,
	/* data: */		NULL,
	/* init: */		NULL,
	/* done: */		NULL
);

static void
finger_end_request(struct connection *conn, enum connection_state state)
{
	set_connection_state(conn, state);

	if (conn->state == S_OK) {
		if (conn->cached) {
			truncate_entry(conn->cached, conn->from, 1);
			conn->cached->incomplete = 0;
		}
	}
	abort_connection(conn);
}

static void
finger_get_response(struct connection *conn, struct read_buffer *rb)
{
	struct cache_entry *cached = get_cache_entry(conn->uri);
	int l;

	set_connection_timeout(conn);

	if (!cached) {
		abort_conn_with_state(conn, S_OUT_OF_MEM);
		return;
	}
	conn->cached = cached;

	if (rb->close == READ_BUFFER_END) {
		finger_end_request(conn, S_OK);
		return;
	}

	l = rb->len;
	conn->received += l;

	if (add_fragment(conn->cached, conn->from, rb->data, l) == 1)
		conn->tries = 0;

	conn->from += l;
	kill_buffer_data(rb, l);
	read_from_socket(conn, &conn->socket, rb, finger_get_response);
	set_connection_state(conn, S_TRANS);
}

static void
finger_sent_request(struct connection *conn)
{
	struct read_buffer *rb;

	set_connection_timeout(conn);
	rb = alloc_read_buffer(conn);
	if (!rb) return;
	rb->close = READ_BUFFER_END_ONCLOSE;
	read_from_socket(conn, &conn->socket, rb, finger_get_response);
}

static void
finger_send_request(struct connection *conn)
{
	struct string req;

	if (!init_string(&req)) return;
	/* add_to_string(&req, &rl, "/W"); */

	if (conn->uri->user) {
		add_char_to_string(&req, ' ');
		add_bytes_to_string(&req, conn->uri->user, conn->uri->userlen);
	}
	add_crlf_to_string(&req);
	write_to_socket(conn, &conn->socket, req.source, req.length, finger_sent_request);
	done_string(&req);
	set_connection_state(conn, S_SENT);
}

void
finger_protocol_handler(struct connection *conn)
{
	set_connection_timeout(conn);
	conn->from = 0;
	make_connection(conn, &conn->socket, finger_send_request);
}