You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.1 KiB
88 lines
2.1 KiB
#pragma once
|
|
|
|
#include <assert.h>
|
|
#include <arpa/inet.h>
|
|
#include <errno.h>
|
|
#include <stdbool.h>
|
|
#include <string.h>
|
|
#include <sys/socket.h>
|
|
#include <unistd.h>
|
|
|
|
#include "debuglog.h"
|
|
#include "http.h"
|
|
#include "http_total.h"
|
|
#include "panic.h"
|
|
#include "likely.h"
|
|
|
|
|
|
static inline void
|
|
client_socket_close(int client_socket, struct sockaddr *client_address)
|
|
{
|
|
/* Should never close 0, 1, or 2 */
|
|
assert(client_socket != STDIN_FILENO);
|
|
assert(client_socket != STDOUT_FILENO);
|
|
assert(client_socket != STDERR_FILENO);
|
|
|
|
if (unlikely(close(client_socket) < 0)) {
|
|
char client_address_text[INET6_ADDRSTRLEN] = { '\0' };
|
|
if (unlikely(inet_ntop(AF_INET, &client_address, client_address_text, INET6_ADDRSTRLEN) == NULL)) {
|
|
debuglog("Failed to log client_address: %s", strerror(errno));
|
|
}
|
|
debuglog("Error closing client socket %d associated with %s - %s", client_socket, client_address_text,
|
|
strerror(errno));
|
|
}
|
|
}
|
|
|
|
typedef void (*void_cb)(void);
|
|
|
|
/**
|
|
* Writes buffer to the client socket
|
|
* @param client_socket - the client we are rejecting
|
|
* @param buffer - buffer to write to socket
|
|
* @param on_eagain - cb to execute when client socket returns EAGAIN. If NULL, error out
|
|
* @returns 0 on success, -1 on error.
|
|
*/
|
|
static inline int
|
|
client_socket_send(int client_socket, const char *buffer, size_t buffer_len, void_cb on_eagain)
|
|
{
|
|
int rc;
|
|
|
|
size_t cursor = 0;
|
|
|
|
while (cursor < buffer_len) {
|
|
ssize_t sent = write(client_socket, &buffer[cursor], buffer_len - cursor);
|
|
if (sent < 0) {
|
|
if (errno == EAGAIN) {
|
|
if (on_eagain == NULL) {
|
|
rc = -1;
|
|
goto done;
|
|
}
|
|
on_eagain();
|
|
} else {
|
|
debuglog("Error sending to client: %s", strerror(errno));
|
|
rc = -1;
|
|
goto done;
|
|
}
|
|
}
|
|
|
|
assert(sent > 0);
|
|
cursor += (size_t)sent;
|
|
};
|
|
|
|
rc = 0;
|
|
done:
|
|
return rc;
|
|
}
|
|
|
|
/**
|
|
* Rejects request due to admission control or error
|
|
* @param client_socket - the client we are rejecting
|
|
* @param buffer - buffer to write to socket
|
|
* @returns 0
|
|
*/
|
|
static inline int
|
|
client_socket_send_oneshot(int client_socket, const char *buffer, size_t buffer_len)
|
|
{
|
|
return client_socket_send(client_socket, buffer, buffer_len, NULL);
|
|
}
|