#include "afh.h"
#include "afs.h"
#include "server.h"
+#include "acl.h"
#include "send.h"
#include "close_on_fork.h"
#include "chunk_queue.h"
#include "vss.h"
+/** Clients will be kicked if there are more than that many bytes pending. */
+#define MAX_CQ_BYTES 40000
/**
* Open a passive socket of given layer4 type.
*
* \return The listening fd on success, negative on errors.
*/
-int open_sender(unsigned l4type, int port)
+static int open_sender(unsigned l4type, int port)
{
int fd, ret = para_listen(AF_UNSPEC, l4type, port);
* list, destroy the chunk queue of this client, delete the client from the
* list of connected clients and free the sender_client struct.
*/
-void shutdown_client(struct sender_client *sc)
+void shutdown_client(struct sender_client *sc, struct sender_status *ss)
{
PARA_INFO_LOG("shutting down %s on fd %d\n", sc->name, sc->fd);
free(sc->name);
list_del(&sc->node);
free(sc->private_data);
free(sc);
+ ss->num_clients--;
+}
+
+void shutdown_clients(struct sender_status *ss)
+{
+ struct sender_client *sc, *tmp;
+ list_for_each_entry_safe(sc, tmp, &ss->client_list, node)
+ shutdown_client(sc, ss);
}
/**
}
static int queue_chunk_or_shutdown(struct sender_client *sc,
- long unsigned chunk_num, size_t sent)
+ struct sender_status *ss, long unsigned chunk_num,
+ size_t sent)
{
int ret = cq_enqueue(sc->cq, chunk_num, sent);
if (ret < 0)
- shutdown_client(sc);
+ shutdown_client(sc, ss);
return ret;
}
* On errors, the client is shut down. If only a part of the buffer could be
* written, the remainder is put into the chunk queue for that client.
*/
-void send_chunk(struct sender_client *sc, size_t max_bytes_per_write,
- long unsigned current_chunk, const char *buf, size_t len)
+void send_chunk(struct sender_client *sc, struct sender_status *ss,
+ size_t max_bytes_per_write, long unsigned current_chunk,
+ const char *buf, size_t len)
{
int ret;
size_t header_len;
char *header_buf = vss_get_header(&header_len);
if (header_buf && header_len > 0) {
- ret = queue_chunk_or_shutdown(sc, -1U, 0);
+ ret = queue_chunk_or_shutdown(sc, ss, -1U, 0);
if (ret < 0)
goto out;
}
}
ret = send_queued_chunks(sc, max_bytes_per_write);
if (ret < 0) {
- shutdown_client(sc);
+ shutdown_client(sc, ss);
goto out;
}
if (!len)
goto out;
if (!ret) { /* still data left in the queue */
- ret = queue_chunk_or_shutdown(sc, current_chunk, 0);
+ ret = queue_chunk_or_shutdown(sc, ss, current_chunk, 0);
goto out;
}
ret = write_nonblock(sc->fd, buf, len, max_bytes_per_write);
if (ret < 0) {
- shutdown_client(sc);
+ shutdown_client(sc, ss);
goto out;
}
if (ret != len)
- ret = queue_chunk_or_shutdown(sc, current_chunk, ret);
+ ret = queue_chunk_or_shutdown(sc, ss, current_chunk, ret);
out:
if (ret < 0)
PARA_NOTICE_LOG("%s\n", para_strerror(-ret));
}
+
+void init_sender_status(struct sender_status *ss, char **access_arg,
+ int num_access_args, int port, int max_clients, int default_deny)
+{
+ ss->listen_fd = -1;
+ INIT_LIST_HEAD(&ss->client_list);
+ ss->port = port;
+ acl_init(&ss->acl, access_arg, num_access_args);
+ ss->num_clients = 0;
+ ss->max_clients = max_clients;
+ ss->default_deny = default_deny;
+}
+
+char *get_sender_info(struct sender_status *ss, char *name)
+{
+ char *clnts = NULL, *ret;
+ struct sender_client *sc, *tmp_sc;
+
+ char *acl_contents = acl_get_contents(&ss->acl);
+ list_for_each_entry_safe(sc, tmp_sc, &ss->client_list, node) {
+ char *tmp = make_message("%s%s ", clnts? clnts : "", sc->name);
+ free(clnts);
+ clnts = tmp;
+ }
+ ret = make_message(
+ "%s sender:\n"
+ "\tstatus: %s\n"
+ "\tport: %d\n"
+ "\tnumber of connected clients: %d\n"
+ "\tmaximal number of clients: %d%s\n"
+ "\tconnected clients: %s\n"
+ "\taccess %s list: %s\n",
+ name,
+ (ss->listen_fd >= 0)? "on" : "off",
+ ss->port,
+ ss->num_clients,
+ ss->max_clients,
+ ss->max_clients > 0? "" : " (unlimited)",
+ clnts? clnts : "(none)",
+ ss->default_deny? "allow" : "deny",
+ acl_contents? acl_contents : "(empty)"
+ );
+ free(acl_contents);
+ free(clnts);
+ return ret;
+}
+
+void generic_com_allow(struct sender_command_data *scd,
+ struct sender_status *ss)
+{
+ acl_allow(scd->addr, scd->netmask, &ss->acl, ss->default_deny);
+}
+
+void generic_com_deny(struct sender_command_data *scd,
+ struct sender_status *ss)
+{
+ acl_deny(scd->addr, scd->netmask, &ss->acl, ss->default_deny);
+}
+
+int generic_com_on(struct sender_status *ss, unsigned protocol)
+{
+ int ret;
+
+ if (ss->listen_fd >= 0)
+ return 1;
+ ret = open_sender(protocol, ss->port);
+ if (ret < 0)
+ return ret;
+ ss->listen_fd = ret;
+ return 1;
+}
+
+void generic_com_off(struct sender_status *ss)
+{
+ if (ss->listen_fd < 0)
+ return;
+ PARA_NOTICE_LOG("closing port %d\n", ss->port);
+ close(ss->listen_fd);
+ del_close_on_fork_list(ss->listen_fd);
+ shutdown_clients(ss);
+ ss->listen_fd = -1;
+}
+
+struct sender_client *accept_sender_client(struct sender_status *ss)
+{
+ struct sender_client *sc;
+ int fd, ret = para_accept(ss->listen_fd, NULL, 0);
+ if (ret < 0) {
+ PARA_ERROR_LOG("%s\n", para_strerror(-ret));
+ return NULL;
+ }
+ fd = ret;
+ ret = -E_MAX_CLIENTS;
+ if (ss->max_clients > 0 && ss->num_clients >= ss->max_clients)
+ goto err_out;
+ ret = mark_fd_nonblocking(fd);
+ if (ret < 0)
+ goto err_out;
+ ret = acl_check_access(fd, &ss->acl, ss->default_deny);
+ if (ret < 0)
+ goto err_out;
+ ss->num_clients++;
+ sc = para_calloc(sizeof(*sc));
+ sc->fd = fd;
+ sc->name = make_message("%s", remote_name(fd));
+ sc->cq = cq_new(MAX_CQ_BYTES);
+ para_list_add(&sc->node, &ss->client_list);
+ add_close_on_fork_list(fd);
+ PARA_INFO_LOG("accepted client #%d: %s (fd %d)\n", ss->num_clients,
+ sc->name, fd);
+ return sc;
+err_out:
+ PARA_WARNING_LOG("%s\n", para_strerror(-ret));
+ close(fd);
+ return NULL;
+}
+
+char *generic_sender_help(void)
+{
+ return make_message(
+ "usage: {on|off}\n"
+ "usage: {allow|deny} IP mask\n"
+ "example: allow 127.0.0.1 32\n"
+ );
+}
projects / paraslash.git / blobdiff