mirror of
https://github.com/wassname/ray.git
synced 2026-08-11 11:24:51 +08:00
Make it possible to read and write data that is not null-terminated (#9)
* Make it possible to read and write data that is not null-terminated * formating
This commit is contained in:
committed by
Robert Nishihara
parent
ff11ee21ef
commit
b18f214d55
@@ -2,32 +2,39 @@
|
||||
|
||||
#include <assert.h>
|
||||
#include <unistd.h>
|
||||
#include <inttypes.h>
|
||||
|
||||
#include "sockets.h"
|
||||
#include "io.h"
|
||||
|
||||
SUITE(socket_tests);
|
||||
SUITE(io_tests);
|
||||
|
||||
TEST ipc_socket_test(void) {
|
||||
const char* socket_pathname = "test-socket";
|
||||
const char *socket_pathname = "test-socket";
|
||||
int socket_fd = bind_ipc_sock(socket_pathname);
|
||||
ASSERT(socket_fd >= 0);
|
||||
|
||||
char* test_string = "hello world";
|
||||
char *test_string = "hello world";
|
||||
char *test_bytes = "another string";
|
||||
pid_t pid = fork();
|
||||
if (pid == 0) {
|
||||
close(socket_fd);
|
||||
socket_fd = connect_ipc_sock(socket_pathname);
|
||||
ASSERT(socket_fd >= 0);
|
||||
send_ipc_sock(socket_fd, test_string);
|
||||
write_string(socket_fd, test_string);
|
||||
write_bytes(socket_fd, (uint8_t *) test_bytes, strlen(test_bytes));
|
||||
close(socket_fd);
|
||||
exit(0);
|
||||
} else {
|
||||
int client_fd = accept_client(socket_fd);
|
||||
ASSERT(client_fd >= 0);
|
||||
char* message = recv_ipc_sock(client_fd);
|
||||
char *message = read_string(client_fd);
|
||||
ASSERT(message != NULL);
|
||||
ASSERT_STR_EQ(test_string, message);
|
||||
free(message);
|
||||
int64_t len;
|
||||
uint8_t *bytes;
|
||||
read_bytes(client_fd, &bytes, &len);
|
||||
ASSERT(memcmp(test_bytes, bytes, len) == 0);
|
||||
close(client_fd);
|
||||
close(socket_fd);
|
||||
unlink(socket_pathname);
|
||||
@@ -36,14 +43,14 @@ TEST ipc_socket_test(void) {
|
||||
PASS();
|
||||
}
|
||||
|
||||
SUITE(socket_tests) {
|
||||
SUITE(io_tests) {
|
||||
RUN_TEST(ipc_socket_test);
|
||||
}
|
||||
|
||||
GREATEST_MAIN_DEFS();
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
int main(int argc, char **argv) {
|
||||
GREATEST_MAIN_BEGIN();
|
||||
RUN_SUITE(socket_tests);
|
||||
RUN_SUITE(io_tests);
|
||||
GREATEST_MAIN_END();
|
||||
}
|
||||
Reference in New Issue
Block a user