aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/libmdsserver/util.c63
-rw-r--r--src/libmdsserver/util.h10
-rw-r--r--src/mds-colour.c9
3 files changed, 38 insertions, 44 deletions
diff --git a/src/libmdsserver/util.c b/src/libmdsserver/util.c
index af78a62..13f0325 100644
--- a/src/libmdsserver/util.c
+++ b/src/libmdsserver/util.c
@@ -31,6 +31,7 @@
#include <time.h>
#include <sys/wait.h>
#include <stdint.h>
+#include <inttypes.h>
@@ -810,52 +811,32 @@ int verify_utf8(const char* string, int allow_modified_nul)
* @param send_buffer_size Pointer to the allocation size of `*send_buffer`, it should
* contain the current size of `*send_buffer` and will be updated
* with the new size, must not be `NULL`
+ * @param message_id The message ID of this message
* @return The length of the message, zero on error
*/
size_t construct_error_message(const char* restrict recv_client_id, const char* restrict recv_message_id,
int custom, int errnum, const char* restrict message, char** restrict send_buffer,
- size_t* restrict send_buffer_size)
+ size_t* restrict send_buffer_size, uint32_t message_id)
{
ssize_t part_length;
size_t length = 0;
char* temp;
- /* Measure the length of mandatory headers and either values,
- as well as the line to end the headers. The `Error`-header
- however is currently measure without error number. */
- snprintf(NULL, 0,
- "Command: error\n"
- "To: %s\n"
- "In response to: %s\n"
- "Error: %s\n"
- "\n%zn",
- recv_client_id, recv_message_id,
- custom ? "custom" : "",
- &part_length),
- length += (size_t)part_length;
-
- /* If the error number is custom and their is a number,
- a blank space is required between the word ‘custom’
- and the number. */
- if (custom && (errnum >= 0))
- length++;
- /* Measure the length of the error number. */
- if (errnum >= 0)
- snprintf(NULL, 0, "%i%zn", errnum, &part_length),
- length += (size_t)part_length;
-
- /* Measure the length of the error description and
- the length of the header specifying its length. */
+ /* Measure the maximum length of message, including NUL-termination.. */
+ length += sizeof("Command: error\n"
+ "To: 4294967296:4294967296\n"
+ "In response to: 4294967296\n"
+ "Message ID: 4294967296\n"
+ "Error: custom \n"
+ "Length: \n"
+ "\n") / sizeof(char) + 3 * (sizeof(int));
if (message != NULL)
- snprintf(NULL, 0, "Length: %zu\n%zn",
- strlen(message) + 1, &part_length),
- length += (size_t)part_length + strlen(message) + 1;
+ length += (sizeof("Length: \n") / sizeof(char) - 1) + 3 * sizeof(char) + strlen(message) + 1;
/* Ensure that the send buffer is large enough. */
if (length > *send_buffer_size)
{
- if (yrealloc(temp, *send_buffer, length, char))
- return 0;
+ fail_if (yrealloc(temp, *send_buffer, length, char));
*send_buffer_size = length;
}
@@ -869,9 +850,10 @@ size_t construct_error_message(const char* restrict recv_client_id, const char*
"Command: error\n"
"To: %s\n"
"In response to: %s\n"
+ "Message ID: %"PRIu32"\n"
"Error: %s%zn",
recv_client_id, recv_message_id,
- custom ? "custom" : "",
+ message_id, custom ? "custom" : "",
&part_length),
length += (size_t)part_length;
@@ -906,6 +888,8 @@ size_t construct_error_message(const char* restrict recv_client_id, const char*
}
return length;
+ fail:
+ return 0;
}
@@ -930,14 +914,19 @@ size_t construct_error_message(const char* restrict recv_client_id, const char*
* contain the current size of `*send_buffer` and will be updated
* with the new size, must not be `NULL`
* @param socket_fd The file descriptor of the socket
+ * @param message_id The message ID of this message
* @return Zero on success, -1 on error
*/
int send_error(const char* restrict recv_client_id, const char* restrict recv_message_id,
int custom, int errnum, const char* restrict message, char** restrict send_buffer,
- size_t* restrict send_buffer_size, int socket_fd)
+ size_t* restrict send_buffer_size, uint32_t message_id, int socket_fd)
{
- size_t length = construct_error_message(recv_client_id, recv_message_id, custom, errnum,
- message, send_buffer, send_buffer_size);
- return length ? full_send(socket_fd, *send_buffer, length) : -1;
+ size_t length;
+ fail_if ((length = construct_error_message(recv_client_id, recv_message_id, custom, errnum,
+ message, send_buffer, send_buffer_size, message_id)) == 0);
+ fail_if (full_send(socket_fd, *send_buffer, length));
+ return 0;
+ fail:
+ return -1;
}
diff --git a/src/libmdsserver/util.h b/src/libmdsserver/util.h
index cd091d4..9ac0357 100644
--- a/src/libmdsserver/util.h
+++ b/src/libmdsserver/util.h
@@ -400,11 +400,13 @@ int verify_utf8(const char* string, int allow_modified_nul) __attribute__((pure)
* @param send_buffer_size Pointer to the allocation size of `*send_buffer`, it should
* contain the current size of `*send_buffer` and will be updated
* with the new size, must not be `NULL`
+ * @param message_id The message ID of this message
* @return The length of the message, zero on error
*/
size_t construct_error_message(const char* restrict recv_client_id, const char* restrict recv_message_id,
- int custom, int errnum, const char* restrict message, char** restrict send_buffer,
- size_t* restrict send_buffer_size) __attribute__((nonnull(1, 2, 6, 7)));
+ int custom, int errnum, const char* restrict message,
+ char** restrict send_buffer, size_t* restrict send_buffer_size,
+ uint32_t message_id) __attribute__((nonnull(1, 2, 6, 7)));
/**
* Send an error message
@@ -426,12 +428,14 @@ size_t construct_error_message(const char* restrict recv_client_id, const char*
* @param send_buffer_size Pointer to the allocation size of `*send_buffer`, it should
* contain the current size of `*send_buffer` and will be updated
* with the new size, must not be `NULL`
+ * @param message_id The message ID of this message
* @param socket_fd The file descriptor of the socket
* @return Zero on success, -1 on error
*/
int send_error(const char* restrict recv_client_id, const char* restrict recv_message_id,
int custom, int errnum, const char* restrict message, char** restrict send_buffer,
- size_t* restrict send_buffer_size, int socket_fd) __attribute__((nonnull(1, 2, 6, 7)));
+ size_t* restrict send_buffer_size, uint32_t message_id,
+ int socket_fd) __attribute__((nonnull(1, 2, 6, 7)));
#endif
diff --git a/src/mds-colour.c b/src/mds-colour.c
index a7cc7a7..0be6f89 100644
--- a/src/mds-colour.c
+++ b/src/mds-colour.c
@@ -125,7 +125,7 @@ static size_t colour_list_buffer_with_values_length = 0;
((full_send)(socket_fd, message, length))
/**
- * Send an error message
+ * Send an error message, message ID will be incremented
*
* @param recv_client_id:const char* The client ID attached on the message that was received
* @param recv_message_id:const char* The message ID attached on the message that was received
@@ -141,9 +141,10 @@ static size_t colour_list_buffer_with_values_length = 0;
* be omitted
* @return Zero on success, -1 on error
*/
-#define send_error(recv_client_id, recv_message_id, custom, errnum, message) \
- ((send_error)(recv_client_id, recv_message_id, custom, errnum, \
- message, &send_buffer, &send_buffer_size, socket_fd))
+#define send_error(recv_client_id, recv_message_id, custom, errnum, message) \
+ ((send_error)(recv_client_id, recv_message_id, custom, errnum, \
+ message, &send_buffer, &send_buffer_size, message_id, socket_fd) \
+ ? -1 : ((message_id = message_id == INT32_MAX ? 0 : (message_id + 1)), 0))
/**
generated by cgit v1.2.3-70-g09d2 (git 2.47.0) at 2024-12-01 16:06:17 +0000