2021-02-10 21:33:03 +03:00
|
|
|
#ifndef _XOPEN_SOURCE
|
|
|
|
#define _XOPEN_SOURCE 700
|
|
|
|
#endif
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
#include "sc_sock.h"
|
|
|
|
|
|
|
|
#include <assert.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
|
|
|
|
#if defined(_WIN32) || defined(_WIN64)
|
2021-02-01 07:11:12 +03:00
|
|
|
#include <synchapi.h>
|
|
|
|
#define sleep(n) (Sleep(n * 1000))
|
2020-11-25 09:25:38 +03:00
|
|
|
#endif
|
|
|
|
|
|
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
#include <windows.h>
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
struct sc_thread
|
|
|
|
{
|
|
|
|
HANDLE id;
|
2021-02-01 07:11:12 +03:00
|
|
|
void *(*fn)(void *);
|
|
|
|
void *arg;
|
|
|
|
void *ret;
|
2020-11-25 09:25:38 +03:00
|
|
|
};
|
|
|
|
|
|
|
|
#else
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
#include <errno.h>
|
2021-02-08 23:01:44 +03:00
|
|
|
#include <fcntl.h>
|
2021-02-01 07:11:12 +03:00
|
|
|
#include <pthread.h>
|
2021-02-08 23:01:44 +03:00
|
|
|
#include <stdarg.h>
|
2021-02-01 07:11:12 +03:00
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <unistd.h>
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
struct sc_thread
|
|
|
|
{
|
|
|
|
pthread_t id;
|
|
|
|
};
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void sc_thread_init(struct sc_thread *thread);
|
|
|
|
int sc_thread_term(struct sc_thread *thread);
|
|
|
|
int sc_thread_start(struct sc_thread *thread, void *(*fn)(void *), void *arg);
|
2021-02-05 20:46:59 +03:00
|
|
|
int sc_thread_join(struct sc_thread *thread, void **ret);
|
2020-11-25 09:25:38 +03:00
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void sc_thread_init(struct sc_thread *thread)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
|
|
|
thread->id = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
#if defined(_WIN32) || defined(_WIN64)
|
2021-02-01 07:11:12 +03:00
|
|
|
#include <process.h>
|
2020-11-25 09:25:38 +03:00
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
unsigned int __stdcall sc_thread_fn(void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-01 07:11:12 +03:00
|
|
|
struct sc_thread *thread = arg;
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
thread->ret = thread->fn(thread->arg);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
int sc_thread_start(struct sc_thread *thread, void *(*fn)(void *), void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
thread->fn = fn;
|
|
|
|
thread->arg = arg;
|
|
|
|
|
|
|
|
thread->id =
|
2021-02-01 07:11:12 +03:00
|
|
|
(HANDLE) _beginthreadex(NULL, 0, sc_thread_fn, thread, 0, NULL);
|
2020-11-25 09:25:38 +03:00
|
|
|
rc = thread->id == 0 ? -1 : 0;
|
|
|
|
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
2021-02-05 20:46:59 +03:00
|
|
|
int sc_thread_join(struct sc_thread *thread, void **ret)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
|
|
|
int rc = 0;
|
|
|
|
DWORD rv;
|
|
|
|
BOOL brc;
|
|
|
|
|
|
|
|
if (thread->id == 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
rv = WaitForSingleObject(thread->id, INFINITE);
|
|
|
|
if (rv == WAIT_FAILED) {
|
|
|
|
rc = -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
brc = CloseHandle(thread->id);
|
|
|
|
if (!brc) {
|
|
|
|
rc = -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
thread->id = 0;
|
|
|
|
if (ret != NULL) {
|
|
|
|
*ret = thread->ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
int sc_thread_start(struct sc_thread *thread, void *(*fn)(void *), void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
|
|
|
int rc;
|
|
|
|
pthread_attr_t hndl;
|
|
|
|
|
|
|
|
rc = pthread_attr_init(&hndl);
|
|
|
|
if (rc != 0) {
|
2020-12-28 06:18:18 +03:00
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
// This may only fail with EINVAL.
|
|
|
|
pthread_attr_setdetachstate(&hndl, PTHREAD_CREATE_JOINABLE);
|
|
|
|
|
|
|
|
rc = pthread_create(&thread->id, &hndl, fn, arg);
|
|
|
|
|
|
|
|
// This may only fail with EINVAL.
|
|
|
|
pthread_attr_destroy(&hndl);
|
|
|
|
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
2021-02-05 20:46:59 +03:00
|
|
|
int sc_thread_join(struct sc_thread *thread, void **ret)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
|
|
|
int rc;
|
2021-02-01 07:11:12 +03:00
|
|
|
void *val;
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
if (thread->id == 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
rc = pthread_join(thread->id, &val);
|
|
|
|
thread->id = 0;
|
|
|
|
|
|
|
|
if (ret != NULL) {
|
|
|
|
*ret = val;
|
|
|
|
}
|
|
|
|
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
int sc_thread_term(struct sc_thread *thread)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-05 20:46:59 +03:00
|
|
|
return sc_thread_join(thread, NULL);
|
2020-11-25 09:25:38 +03:00
|
|
|
}
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void *server_ip4(void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-07 22:31:04 +03:00
|
|
|
(void) arg;
|
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
char buf[5];
|
|
|
|
char tmp[128];
|
|
|
|
struct sc_sock sock, accepted;
|
|
|
|
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8004") == 0);
|
|
|
|
sc_sock_print(&sock, tmp, sizeof(tmp));
|
|
|
|
assert(strcmp(tmp, "Local(127.0.0.1:8004), Remote() ") == 0);
|
|
|
|
|
|
|
|
assert(sc_sock_accept(&sock, &accepted) == 0);
|
2020-12-28 11:03:24 +03:00
|
|
|
assert(sc_sock_recv(&accepted, buf, 5, 0) == 5);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(strcmp("test", buf) == 0);
|
|
|
|
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&accepted) == 0);
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void *client_ip4(void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-07 22:31:04 +03:00
|
|
|
(void) arg;
|
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
int rc;
|
|
|
|
char tmp[128];
|
|
|
|
struct sc_sock sock;
|
|
|
|
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_set_sndtimeo(&sock, 10000) != 0);
|
|
|
|
printf("%s \n", sc_sock_error(&sock));
|
|
|
|
assert(sc_sock_set_rcvtimeo(&sock, 10000) != 0);
|
|
|
|
printf("%s \n", sc_sock_error(&sock));
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
for (int i = 0; i < 5; i++) {
|
|
|
|
rc = sc_sock_connect(&sock, "127.0.0.1", "8004", NULL, NULL);
|
|
|
|
if (rc == 0) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
sleep(1);
|
|
|
|
}
|
|
|
|
|
|
|
|
assert(rc == 0);
|
|
|
|
sc_sock_print(&sock, tmp, sizeof(tmp));
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_set_sndtimeo(&sock, 10000) == 0);
|
|
|
|
assert(sc_sock_set_rcvtimeo(&sock, 10000) == 0);
|
2020-12-28 11:03:24 +03:00
|
|
|
assert(sc_sock_send(&sock, "test", 5, 0) == 5);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
void test_ip4()
|
|
|
|
{
|
|
|
|
struct sc_thread thread1;
|
|
|
|
struct sc_thread thread2;
|
|
|
|
|
|
|
|
sc_thread_init(&thread1);
|
|
|
|
sc_thread_init(&thread2);
|
|
|
|
|
|
|
|
assert(sc_thread_start(&thread1, server_ip4, NULL) == 0);
|
|
|
|
assert(sc_thread_start(&thread2, client_ip4, NULL) == 0);
|
|
|
|
|
|
|
|
assert(sc_thread_term(&thread1) == 0);
|
|
|
|
assert(sc_thread_term(&thread2) == 0);
|
|
|
|
}
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void *server_ip6(void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-07 22:31:04 +03:00
|
|
|
(void) arg;
|
2020-11-25 09:25:38 +03:00
|
|
|
char buf[5];
|
|
|
|
char tmp[128];
|
|
|
|
struct sc_sock sock, accepted;
|
|
|
|
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET6);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(sc_sock_listen(&sock, "::1", "8006") == 0);
|
|
|
|
sc_sock_print(&sock, tmp, sizeof(tmp));
|
|
|
|
assert(strcmp(tmp, "Local(::1:8006), Remote() ") == 0);
|
|
|
|
assert(sc_sock_accept(&sock, &accepted) == 0);
|
2020-12-28 11:03:24 +03:00
|
|
|
assert(sc_sock_recv(&accepted, buf, 5, 0) == 5);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(strcmp("test", buf) == 0);
|
|
|
|
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&accepted) == 0);
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void *client_ip6(void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-07 22:31:04 +03:00
|
|
|
(void) arg;
|
2020-11-25 09:25:38 +03:00
|
|
|
int rc;
|
|
|
|
struct sc_sock sock;
|
|
|
|
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET6);
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
for (int i = 0; i < 5; i++) {
|
|
|
|
rc = sc_sock_connect(&sock, "::1", "8006", NULL, NULL);
|
|
|
|
if (rc == 0) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
sleep(1);
|
|
|
|
}
|
|
|
|
assert(rc == 0);
|
2020-12-28 11:03:24 +03:00
|
|
|
assert(sc_sock_send(&sock, "test", 5, 0) == 5);
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
void test_ip6()
|
|
|
|
{
|
|
|
|
struct sc_thread thread1;
|
|
|
|
struct sc_thread thread2;
|
|
|
|
|
|
|
|
sc_thread_init(&thread1);
|
|
|
|
sc_thread_init(&thread2);
|
|
|
|
|
|
|
|
assert(sc_thread_start(&thread1, server_ip6, NULL) == 0);
|
|
|
|
assert(sc_thread_start(&thread2, client_ip6, NULL) == 0);
|
|
|
|
|
|
|
|
assert(sc_thread_term(&thread1) == 0);
|
|
|
|
assert(sc_thread_term(&thread2) == 0);
|
|
|
|
}
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void *server_unix(void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-07 22:31:04 +03:00
|
|
|
(void) arg;
|
2020-11-25 09:25:38 +03:00
|
|
|
char buf[5];
|
|
|
|
char tmp[128];
|
|
|
|
struct sc_sock sock, accepted;
|
|
|
|
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_UNIX);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(sc_sock_listen(&sock, "x.sock", NULL) == 0);
|
|
|
|
sc_sock_print(&sock, tmp, sizeof(tmp));
|
|
|
|
assert(strcmp(tmp, "Local(x.sock), Remote() ") == 0);
|
|
|
|
int rc = sc_sock_accept(&sock, &accepted);
|
|
|
|
if (rc != 0) {
|
|
|
|
printf("error(%d) : %s\n", errno, sock.err);
|
|
|
|
}
|
|
|
|
|
|
|
|
assert(rc == 0);
|
2020-12-28 11:03:24 +03:00
|
|
|
assert(sc_sock_recv(&accepted, buf, 5, 0) == 5);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(strcmp("test", buf) == 0);
|
|
|
|
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&accepted) == 0);
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void *client_unix(void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-07 22:31:04 +03:00
|
|
|
(void) arg;
|
2020-11-25 09:25:38 +03:00
|
|
|
int rc;
|
|
|
|
struct sc_sock sock;
|
|
|
|
|
|
|
|
sleep(3);
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_UNIX);
|
2020-11-25 09:25:38 +03:00
|
|
|
for (int i = 0; i < 10; i++) {
|
|
|
|
printf("Will connect to x.sock \n");
|
|
|
|
rc = sc_sock_connect(&sock, "x.sock", NULL, NULL, NULL);
|
|
|
|
if (rc == 0) {
|
|
|
|
printf("Connected to x.sock \n");
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
printf("Failed to connect to x.sock \n");
|
|
|
|
sleep(1);
|
|
|
|
}
|
|
|
|
|
|
|
|
assert(rc == 0);
|
2020-12-28 11:03:24 +03:00
|
|
|
assert(sc_sock_send(&sock, "test", 5, 0) == 5);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void test_unix()
|
|
|
|
{
|
|
|
|
struct sc_thread thread1;
|
|
|
|
struct sc_thread thread2;
|
|
|
|
|
|
|
|
sc_thread_init(&thread1);
|
|
|
|
sc_thread_init(&thread2);
|
|
|
|
|
|
|
|
assert(sc_thread_start(&thread1, server_unix, NULL) == 0);
|
|
|
|
assert(sc_thread_start(&thread2, client_unix, NULL) == 0);
|
|
|
|
|
|
|
|
assert(sc_thread_term(&thread1) == 0);
|
|
|
|
assert(sc_thread_term(&thread2) == 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void test1()
|
|
|
|
{
|
2021-02-08 23:01:44 +03:00
|
|
|
int rc;
|
2020-11-25 09:25:38 +03:00
|
|
|
char tmp[5];
|
2021-02-08 23:01:44 +03:00
|
|
|
struct sc_sock sock, client, in;
|
2020-11-25 09:25:38 +03:00
|
|
|
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, false, SC_SOCK_INET);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(sc_sock_connect(&sock, "3127.0.0.1", "2131", NULL, NULL) != 0);
|
|
|
|
assert(sc_sock_finish_connect(&sock) != 0);
|
|
|
|
assert(sc_sock_connect(&sock, "3127.0.0.1", "2131", "127.90.1.1", "50") !=
|
|
|
|
0);
|
|
|
|
assert(sc_sock_finish_connect(&sock) != 0);
|
2020-12-28 11:03:24 +03:00
|
|
|
assert(sc_sock_send(&sock, "test", 5, 0) == -1);
|
|
|
|
assert(sc_sock_recv(&sock, tmp, sizeof(tmp), 0) == -1);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(sc_sock_connect(&sock, "131s::1", "2131", "::1", "50") != 0);
|
|
|
|
assert(sc_sock_finish_connect(&sock) != 0);
|
|
|
|
assert(sc_sock_connect(&sock, "d::1", "2131", "dsad", "50") != 0);
|
|
|
|
assert(sc_sock_finish_connect(&sock) != 0);
|
|
|
|
assert(sc_sock_connect(&sock, "dsadas", "2131", "::1", "50") != 0);
|
2021-02-17 20:04:08 +03:00
|
|
|
assert(sc_sock_connect(&sock, "dsadas", "2131", NULL, "50") != 0);
|
|
|
|
assert(sc_sock_connect(&sock, "dsadas", "2131", "s", NULL) != 0);
|
|
|
|
assert(sc_sock_connect(&sock, "127.0.01", "2131", "100.0.0.0", NULL) != 0);
|
|
|
|
assert(sc_sock_connect(&sock, "127.0.01", "2131", NULL, "9000") != 0);
|
2020-11-25 09:25:38 +03:00
|
|
|
sc_sock_term(&sock);
|
2021-02-08 23:01:44 +03:00
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, false, SC_SOCK_INET);
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
rc = sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL);
|
|
|
|
assert(rc != SC_SOCK_ERROR);
|
|
|
|
|
|
|
|
sleep(2);
|
|
|
|
rc = sc_sock_accept(&sock, &in);
|
|
|
|
if (rc != 0) {
|
|
|
|
printf("%d, %s \n", rc, sc_sock_error(&sock));
|
|
|
|
assert(true);
|
|
|
|
}
|
|
|
|
|
|
|
|
assert(sc_sock_finish_connect(&client) == 0);
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
assert(sc_sock_term(&in) == 0);
|
2020-11-25 09:25:38 +03:00
|
|
|
}
|
|
|
|
|
2021-02-07 20:42:08 +03:00
|
|
|
void test_poll_mass(void)
|
|
|
|
{
|
|
|
|
struct sc_sock_poll poll;
|
|
|
|
struct sc_sock_pipe pipe[100];
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
for (int i = 0; i < 100; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
assert(sc_sock_poll_add(&poll, &pipe[i].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
}
|
|
|
|
for (int i = 0; i < 100; i++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[i].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[i]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
for (int i = 0; i < 100; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
assert(sc_sock_poll_add(&poll, &pipe[i].fdt,
|
|
|
|
SC_SOCK_READ | SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
}
|
|
|
|
for (int i = 0; i < 100; i++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[i].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[i].fdt, SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[i]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
for (int i = 0; i < 100; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
assert(sc_sock_poll_add(&poll, &pipe[i].fdt,
|
|
|
|
SC_SOCK_READ | SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
}
|
|
|
|
for (int i = 0; i < 100; i++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[i].fdt, SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[i].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[i]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
for (int i = 0; i < 100; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
assert(sc_sock_poll_add(&poll, &pipe[i].fdt, SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
}
|
|
|
|
for (int i = 0; i < 100; i++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[i].fdt, SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[i].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[i]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
}
|
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
void test_pipe(void)
|
|
|
|
{
|
|
|
|
char buf[5];
|
|
|
|
struct sc_sock_pipe pipe;
|
|
|
|
|
|
|
|
sc_sock_pipe_init(&pipe, 0);
|
|
|
|
sc_sock_pipe_write(&pipe, "test", 5);
|
|
|
|
sc_sock_pipe_read(&pipe, buf, 5);
|
|
|
|
sc_sock_pipe_term(&pipe);
|
|
|
|
|
|
|
|
assert(strcmp("test", buf) == 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef SC_HAVE_WRAP
|
|
|
|
|
2021-02-08 23:01:44 +03:00
|
|
|
struct sc_mutex
|
|
|
|
{
|
|
|
|
pthread_mutex_t mtx;
|
|
|
|
};
|
|
|
|
|
|
|
|
int test_done = 0;
|
|
|
|
struct sc_mutex mutex;
|
|
|
|
|
|
|
|
int sc_mutex_init(struct sc_mutex *mtx)
|
|
|
|
{
|
|
|
|
int rc, rv;
|
|
|
|
pthread_mutexattr_t attr;
|
|
|
|
pthread_mutex_t mut = PTHREAD_MUTEX_INITIALIZER;
|
|
|
|
|
|
|
|
mtx->mtx = mut;
|
|
|
|
|
|
|
|
// May fail on OOM
|
|
|
|
rc = pthread_mutexattr_init(&attr);
|
|
|
|
if (rc != 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
// This won't fail as long as we pass correct params.
|
|
|
|
rc = pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_NORMAL);
|
|
|
|
assert (rc == 0);
|
|
|
|
|
|
|
|
// May fail on OOM
|
|
|
|
rc = pthread_mutex_init(&mtx->mtx, &attr);
|
|
|
|
|
|
|
|
// This won't fail as long as we pass correct param.
|
|
|
|
rv = pthread_mutexattr_destroy(&attr);
|
|
|
|
assert(rv == 0);
|
|
|
|
|
|
|
|
return rc != 0 ? -1 : 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int sc_mutex_term(struct sc_mutex *mtx)
|
|
|
|
{
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
rc = pthread_mutex_destroy(&mtx->mtx);
|
|
|
|
return rc != 0 ? -1 : 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
void sc_mutex_lock(struct sc_mutex *mtx)
|
|
|
|
{
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
// This won't fail as long as we pass correct param.
|
|
|
|
rc = pthread_mutex_lock(&mtx->mtx);
|
|
|
|
if (rc != 0) {
|
|
|
|
printf("%s \n", strerror(rc));
|
|
|
|
fprintf(stderr, "%s \n", strerror(rc));
|
|
|
|
fflush(stderr);
|
|
|
|
fflush(stdout);
|
|
|
|
}
|
|
|
|
assert(rc == 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
void sc_mutex_unlock(struct sc_mutex *mtx)
|
|
|
|
{
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
// This won't fail as long as we pass correct param.
|
|
|
|
rc = pthread_mutex_unlock(&mtx->mtx);
|
|
|
|
assert(rc == 0);
|
|
|
|
}
|
|
|
|
|
2021-02-07 20:42:08 +03:00
|
|
|
bool fail_malloc = false;
|
|
|
|
void *__real_malloc(size_t n);
|
|
|
|
void *__wrap_malloc(size_t n)
|
|
|
|
{
|
|
|
|
if (fail_malloc) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_malloc(n);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool fail_realloc = false;
|
|
|
|
void *__real_realloc(void *p, size_t size);
|
|
|
|
void *__wrap_realloc(void *p, size_t n)
|
|
|
|
{
|
|
|
|
if (fail_realloc) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_realloc(p, n);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool fail_epollcreate1 = false;
|
|
|
|
int __real_epoll_create1(int flags);
|
|
|
|
int __wrap_epoll_create1(int flags)
|
|
|
|
{
|
|
|
|
if (fail_epollcreate1) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_epoll_create1(flags);
|
|
|
|
}
|
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
bool fail_close = false;
|
|
|
|
int __real_close(int fd);
|
|
|
|
int __wrap_close(int fd)
|
|
|
|
{
|
|
|
|
if (fail_close) {
|
2021-02-08 23:01:44 +03:00
|
|
|
__real_close(fd);
|
2020-11-25 09:25:38 +03:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_close(fd);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool fail_pipe = false;
|
|
|
|
int __real_pipe(int __pipedes[2]);
|
|
|
|
int __wrap_pipe(int __pipedes[2])
|
|
|
|
{
|
|
|
|
if (fail_pipe) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_pipe(__pipedes);
|
|
|
|
}
|
|
|
|
|
2021-02-08 23:01:44 +03:00
|
|
|
int fail_setsockopt = INT32_MAX;
|
|
|
|
int __real_setsockopt(int fd, int level, int optname, const void *optval,
|
|
|
|
socklen_t optlen);
|
|
|
|
int __wrap_setsockopt(int fd, int level, int optname, const void *optval,
|
|
|
|
socklen_t optlen)
|
|
|
|
{
|
|
|
|
sc_mutex_lock(&mutex);
|
|
|
|
if (--fail_setsockopt <= 0) {
|
|
|
|
sc_mutex_unlock(&mutex);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
sc_mutex_unlock(&mutex);
|
|
|
|
return __real_setsockopt(fd, level, optname, optval, optlen);
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_listen = false;
|
|
|
|
int __real_listen(int fd, int n);
|
|
|
|
int __wrap_listen(int fd, int n)
|
|
|
|
{
|
|
|
|
if (fail_listen) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_listen(fd, n);
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_fcntl = INT32_MAX;
|
|
|
|
int __real_fcntl(int fd, int cmd, ...);
|
|
|
|
int __wrap_fcntl(int fd, int cmd, ...)
|
|
|
|
{
|
|
|
|
(void) fd;
|
|
|
|
(void) cmd;
|
|
|
|
|
|
|
|
if (test_done) {
|
|
|
|
va_list va;
|
|
|
|
va_start(va, cmd);
|
|
|
|
void* t = va_arg(va, void*);
|
|
|
|
return __real_fcntl(fd, cmd, t);
|
|
|
|
}
|
|
|
|
|
|
|
|
sc_mutex_lock(&mutex);
|
|
|
|
if (--fail_fcntl <= 0) {
|
|
|
|
sc_mutex_unlock(&mutex);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (cmd == F_GETFL) {
|
|
|
|
sc_mutex_unlock(&mutex);
|
|
|
|
return __real_fcntl(fd, F_GETFL, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (cmd == F_SETFL) {
|
|
|
|
va_list va;
|
|
|
|
va_start(va, cmd);
|
|
|
|
int x = va_arg(va, int);
|
|
|
|
sc_mutex_unlock(&mutex);
|
|
|
|
return __real_fcntl(fd, F_SETFL, x);
|
|
|
|
}
|
|
|
|
|
|
|
|
sc_mutex_unlock(&mutex);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_socket = false;
|
|
|
|
int __real_socket(int domain, int type, int protocol);
|
|
|
|
int __wrap_socket(int domain, int type, int protocol)
|
|
|
|
{
|
|
|
|
if (fail_socket) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_socket(domain, type, protocol);
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_inet_ntop = INT32_MAX;
|
|
|
|
const char *__real_inet_ntop(int af, const void *restrict cp,
|
|
|
|
char *restrict buf, socklen_t len);
|
|
|
|
const char *__wrap_inet_ntop(int af, const void *restrict cp,
|
|
|
|
char *restrict buf, socklen_t len)
|
|
|
|
{
|
|
|
|
sc_mutex_lock(&mutex);
|
|
|
|
if (--fail_inet_ntop <= 0) {
|
|
|
|
sc_mutex_unlock(&mutex);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
sc_mutex_unlock(&mutex);
|
|
|
|
return __real_inet_ntop(af, cp, buf, len);
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_connect = 0;
|
|
|
|
int fail_connect_err = 0;
|
|
|
|
int __real_connect(int fd, const struct sockaddr *addr, socklen_t len);
|
|
|
|
int __wrap_connect(int fd, const struct sockaddr *addr, socklen_t len)
|
|
|
|
{
|
|
|
|
if (fail_connect) {
|
|
|
|
errno = fail_connect_err;
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_connect(fd, addr, len);
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_send = INT32_MAX;
|
|
|
|
int fail_send_err = 0;
|
|
|
|
int fail_send_errno = 0;
|
|
|
|
ssize_t __real_send(int fd, const void *buf, size_t n, int flags);
|
|
|
|
ssize_t __wrap_send(int fd, const void *buf, size_t n, int flags)
|
|
|
|
{
|
|
|
|
if (--fail_send <= 0) {
|
|
|
|
errno = (errno == EINTR) ? EINVAL : fail_send_errno;
|
|
|
|
return fail_send_err;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_send(fd, buf, n, flags);
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_write = INT32_MAX;
|
|
|
|
int fail_write_err = 0;
|
|
|
|
int fail_write_errno = 0;
|
|
|
|
ssize_t __real_write(int fd, const void *buf, size_t n);
|
|
|
|
ssize_t __wrap_write(int fd, const void *buf, size_t n)
|
|
|
|
{
|
|
|
|
if (--fail_write <= 0) {
|
|
|
|
errno = (errno == EINTR) ? EINVAL : fail_write_errno;
|
|
|
|
return fail_write_err;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_write(fd, buf, n);
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_recv = INT32_MAX;
|
|
|
|
int fail_recv_err = 0;
|
|
|
|
int fail_recv_errno = 0;
|
|
|
|
ssize_t __real_recv(int fd, void *buf, size_t n, int flags);
|
|
|
|
ssize_t __wrap_recv(int fd, void *buf, size_t n, int flags)
|
|
|
|
{
|
|
|
|
if (--fail_recv <= 0) {
|
|
|
|
errno = (errno == EINTR) ? EINVAL : fail_recv_errno;
|
|
|
|
return fail_recv_err;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_recv(fd, buf, n, flags);
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_read = INT32_MAX;
|
|
|
|
int fail_read_err = 0;
|
|
|
|
int fail_read_errno = 0;
|
|
|
|
ssize_t __real_read(int fd, void *buf, size_t n, int flags);
|
|
|
|
ssize_t __wrap_read(int fd, void *buf, size_t n, int flags)
|
|
|
|
{
|
|
|
|
if (--fail_read <= 0) {
|
|
|
|
errno = (errno == EINTR) ? EINVAL : fail_read_errno;
|
|
|
|
return fail_read_err;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_read(fd, buf, n, flags);
|
|
|
|
}
|
|
|
|
|
|
|
|
int fail_epoll_ctl = false;
|
|
|
|
int __real_epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);
|
|
|
|
int __wrap_epoll_ctl(int epfd, int op, int fd, struct epoll_event *event)
|
|
|
|
{
|
|
|
|
if (fail_epoll_ctl) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return __real_epoll_ctl(epfd, op, fd, event);
|
|
|
|
}
|
|
|
|
|
2021-02-07 20:42:08 +03:00
|
|
|
void poll_fail_test()
|
|
|
|
{
|
|
|
|
bool fail;
|
|
|
|
int i;
|
2021-02-08 23:01:44 +03:00
|
|
|
struct sc_sock sock;
|
2021-02-07 20:42:08 +03:00
|
|
|
struct sc_sock_poll poll;
|
2021-02-08 23:01:44 +03:00
|
|
|
struct sc_sock_pipe pipe[301];
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
sc_sock_init(&sock, 0, true, AF_INET);
|
|
|
|
fail_epoll_ctl = true;
|
|
|
|
assert(sc_sock_poll_add(&poll, &sock.fdt, SC_SOCK_READ, NULL) == -1);
|
|
|
|
fail_epoll_ctl = false;
|
|
|
|
sc_sock_term(&sock);
|
|
|
|
sc_sock_poll_term(&poll);
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
sc_sock_init(&sock, 0, true, AF_INET);
|
|
|
|
sock.fdt.op = SC_SOCK_READ;
|
|
|
|
assert(sc_sock_poll_del(&poll, &sock.fdt, SC_SOCK_READ, NULL) == -1);
|
|
|
|
sc_sock_term(&sock);
|
|
|
|
sc_sock_poll_term(&poll);
|
2021-02-07 20:42:08 +03:00
|
|
|
|
|
|
|
fail_malloc = true;
|
|
|
|
assert(sc_sock_poll_init(&poll) == -1);
|
|
|
|
fail_malloc = false;
|
|
|
|
|
|
|
|
fail_epollcreate1 = true;
|
|
|
|
assert(sc_sock_poll_init(&poll) == -1);
|
|
|
|
fail_epollcreate1 = false;
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
fail = false;
|
|
|
|
fail_realloc = true;
|
|
|
|
for (i = 0; i < 200; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
fail = sc_sock_poll_add(&poll, &pipe[i].fdt, SC_SOCK_READ, NULL);
|
|
|
|
if (fail) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
assert(fail);
|
|
|
|
for (int j = 0; j <= i; j++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[j]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
fail_realloc = false;
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
fail = false;
|
|
|
|
fail_realloc = true;
|
|
|
|
for (i = 0; i < 200; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
fail = sc_sock_poll_add(&poll, &pipe[i].fdt,
|
|
|
|
SC_SOCK_READ | SC_SOCK_WRITE, NULL);
|
|
|
|
if (fail) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
assert(fail);
|
|
|
|
for (int j = 0; j <= i; j++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt,
|
|
|
|
SC_SOCK_READ | SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[j]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
fail_realloc = false;
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
fail = false;
|
|
|
|
fail_realloc = true;
|
|
|
|
for (i = 0; i < 200; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
fail = sc_sock_poll_add(&poll, &pipe[i].fdt, SC_SOCK_WRITE, NULL);
|
|
|
|
if (fail) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
assert(fail);
|
|
|
|
for (int j = 0; j <= i; j++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt, SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[j]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
fail_realloc = false;
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
fail = false;
|
|
|
|
fail_realloc = true;
|
|
|
|
for (i = 0; i < 200; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
fail = sc_sock_poll_add(&poll, &pipe[i].fdt,
|
|
|
|
SC_SOCK_READ | SC_SOCK_WRITE, NULL);
|
|
|
|
if (fail) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
assert(fail);
|
|
|
|
for (int j = 0; j <= i; j++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt, SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[j]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
fail_realloc = false;
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
fail = false;
|
|
|
|
fail_realloc = true;
|
|
|
|
for (i = 0; i < 200; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
fail = sc_sock_poll_add(&poll, &pipe[i].fdt,
|
|
|
|
SC_SOCK_READ | SC_SOCK_WRITE, NULL);
|
|
|
|
if (fail) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
assert(fail);
|
|
|
|
for (int j = 0; j <= i; j++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt, SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[j]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
fail_realloc = false;
|
|
|
|
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
fail = false;
|
|
|
|
fail_realloc = true;
|
|
|
|
for (i = 0; i < 200; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
fail = sc_sock_poll_add(&poll, &pipe[i].fdt,
|
|
|
|
SC_SOCK_READ | SC_SOCK_WRITE, NULL);
|
|
|
|
if (fail) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
assert(sc_sock_poll_add(&poll, &pipe[i].fdt,
|
|
|
|
SC_SOCK_READ | SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
}
|
|
|
|
assert(fail);
|
|
|
|
for (int j = 0; j <= i; j++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt, SC_SOCK_WRITE, NULL) == 0);
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[j]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
fail_realloc = false;
|
2021-02-08 23:01:44 +03:00
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
for (i = 0; i < SC_SIZE_MAX + 1; i++) {
|
|
|
|
assert(sc_sock_pipe_init(&pipe[i], 0) == 0);
|
|
|
|
fail = sc_sock_poll_add(&poll, &pipe[i].fdt, SC_SOCK_READ, NULL);
|
|
|
|
if (fail) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
assert(fail);
|
|
|
|
for (int j = 0; j <= i; j++) {
|
|
|
|
assert(sc_sock_poll_del(&poll, &pipe[j].fdt, SC_SOCK_READ, NULL) == 0);
|
|
|
|
assert(sc_sock_pipe_term(&pipe[j]) == 0);
|
|
|
|
}
|
|
|
|
assert(poll.count == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
2021-02-07 20:42:08 +03:00
|
|
|
}
|
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
void pipe_fail_test()
|
|
|
|
{
|
|
|
|
struct sc_sock_pipe pipe;
|
|
|
|
fail_pipe = true;
|
|
|
|
assert(sc_sock_pipe_init(&pipe, 0) != 0);
|
2021-02-04 02:35:34 +03:00
|
|
|
assert(*sc_sock_pipe_err(&pipe) != '\0');
|
2020-11-25 09:25:38 +03:00
|
|
|
fail_pipe = false;
|
|
|
|
assert(sc_sock_pipe_init(&pipe, 0) == 0);
|
|
|
|
fail_close = true;
|
|
|
|
assert(sc_sock_pipe_term(&pipe) == -1);
|
|
|
|
fail_close = false;
|
2021-02-08 23:01:44 +03:00
|
|
|
|
|
|
|
assert(sc_sock_pipe_init(&pipe, 0) == 0);
|
|
|
|
fail_write = 1;
|
|
|
|
fail_write_err = -1;
|
|
|
|
fail_write_errno = EINTR;
|
|
|
|
assert(sc_sock_pipe_write(&pipe, NULL, 10) == -1);
|
|
|
|
sc_sock_pipe_term(&pipe);
|
|
|
|
|
|
|
|
fail_write = INT32_MAX;
|
|
|
|
fail_write_err = 0;
|
|
|
|
fail_write_errno = 0;
|
|
|
|
|
|
|
|
assert(sc_sock_pipe_init(&pipe, 0) == 0);
|
|
|
|
fail_read = 1;
|
|
|
|
fail_read_err = -1;
|
|
|
|
fail_read_errno = EINTR;
|
|
|
|
assert(sc_sock_pipe_read(&pipe, NULL, 10) == -1);
|
|
|
|
sc_sock_pipe_term(&pipe);
|
|
|
|
|
|
|
|
fail_read = INT32_MAX;
|
|
|
|
fail_read_err = 0;
|
|
|
|
fail_read_errno = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
void sock_fail_test()
|
|
|
|
{
|
|
|
|
struct sc_sock sock;
|
|
|
|
struct sc_sock sock2;
|
|
|
|
struct sc_sock sock3;
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
|
|
|
sc_sock_listen(&sock, "127.0.0.1", "8080");
|
|
|
|
fail_close = true;
|
|
|
|
assert(sc_sock_term(&sock) == -1);
|
|
|
|
assert(*sc_sock_error(&sock));
|
|
|
|
fail_close = false;
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
|
|
|
|
|
|
|
fail_setsockopt = 1;
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == -1);
|
|
|
|
fail_setsockopt = 2;
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == -1);
|
|
|
|
fail_setsockopt = 3;
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == -1);
|
|
|
|
fail_setsockopt = 4;
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == -1);
|
|
|
|
fail_setsockopt = 5;
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == 0);
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_UNIX);
|
|
|
|
fail_setsockopt = 1;
|
|
|
|
assert(sc_sock_listen(&sock, "/tmp/x", NULL) == -1);
|
|
|
|
fail_setsockopt = 2;
|
|
|
|
assert(sc_sock_listen(&sock, "/tmp/x", NULL) == -1);
|
|
|
|
fail_setsockopt = 3;
|
|
|
|
assert(sc_sock_listen(&sock, "/tmp/x", NULL) == 0);
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
|
|
|
fail_listen = 1;
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == -1);
|
|
|
|
fail_listen = 0;
|
|
|
|
assert(sc_sock_accept(&sock, &sock2) == -1);
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == 0);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
|
|
|
|
sc_sock_init(&sock2, 0, true, SC_SOCK_INET);
|
|
|
|
assert(sc_sock_connect(&sock2, "127.0.0.1", "8080", NULL, NULL) == 0);
|
|
|
|
fail_setsockopt = 1;
|
|
|
|
assert(sc_sock_accept(&sock, &sock3) == -1);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
sc_sock_term(&sock);
|
|
|
|
sc_sock_term(&sock2);
|
|
|
|
|
|
|
|
assert(sc_sock_recv(&sock, NULL, -33, 0) == -33);
|
|
|
|
assert(sc_sock_send(&sock, NULL, -33, 0) == -33);
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET6);
|
|
|
|
fail_setsockopt = 1;
|
|
|
|
assert(sc_sock_listen(&sock, "::1", "8080") == -1);
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
|
|
|
|
assert(sc_sock_set_blocking(&sock, true) == -1);
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == 0);
|
|
|
|
fail_fcntl = 2;
|
|
|
|
assert(sc_sock_set_blocking(&sock, true) == -1);
|
|
|
|
fail_fcntl = INT32_MAX;
|
|
|
|
sc_sock_term(&sock);
|
|
|
|
|
2021-02-08 23:10:45 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
|
|
|
fail_fcntl = 1;
|
|
|
|
assert(sc_sock_connect(&sock, "127.0.0.1", "8080", NULL, NULL) == -1);
|
|
|
|
assert(sc_sock_set_blocking(&sock, true) == -1);
|
|
|
|
fail_fcntl = INT32_MAX;
|
|
|
|
sc_sock_term(&sock);
|
|
|
|
|
2021-02-08 23:01:44 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
|
|
|
fail_fcntl = 1;
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == -1);
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
fail_fcntl = INT32_MAX;
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_UNIX);
|
|
|
|
assert(sc_sock_connect(
|
|
|
|
&sock,
|
|
|
|
"/tmp/"
|
|
|
|
"longlonglonglolonglonglonglonglonglonglonglonglonglonglongl"
|
|
|
|
"onglonglonglonglonglonglongnglonglonglonglonglong",
|
|
|
|
"", NULL, NULL) == -1);
|
|
|
|
sc_sock_term(&sock);
|
|
|
|
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
|
|
|
fail_socket = 1;
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == -1);
|
|
|
|
assert(sc_sock_connect(&sock, "127.0.0.1", "8080", NULL, NULL) == -1);
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_UNIX);
|
|
|
|
fail_socket = 1;
|
|
|
|
assert(sc_sock_listen(&sock, "/tmp/x", "8080") == -1);
|
|
|
|
assert(sc_sock_connect(&sock, "/tmp/x", "8080", NULL, NULL) == -1);
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
|
|
|
|
fail_socket = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
void sock_fail_test2()
|
|
|
|
{
|
|
|
|
int rc;
|
|
|
|
struct sc_sock sock, client, in;
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, false, SC_SOCK_INET);
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == 0);
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
rc = sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL);
|
|
|
|
assert(rc != SC_SOCK_ERROR);
|
|
|
|
sleep(2);
|
|
|
|
fail_fcntl = 1;
|
|
|
|
assert(sc_sock_accept(&sock, &in) == -1);
|
|
|
|
fail_fcntl = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
assert(sc_sock_term(&in) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, false, SC_SOCK_INET);
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == 0);
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
rc = sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL);
|
|
|
|
assert(rc != SC_SOCK_ERROR);
|
|
|
|
sleep(2);
|
|
|
|
fail_setsockopt = 1;
|
|
|
|
assert(sc_sock_accept(&sock, &in) == -1);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
assert(sc_sock_term(&in) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, false, SC_SOCK_INET);
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == 0);
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
rc = sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL);
|
|
|
|
assert(rc != SC_SOCK_ERROR);
|
|
|
|
sleep(2);
|
|
|
|
fail_setsockopt = 2;
|
|
|
|
assert(sc_sock_accept(&sock, &in) == -1);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
assert(sc_sock_term(&in) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, false, SC_SOCK_INET);
|
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1", "8080") == 0);
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
rc = sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL);
|
|
|
|
assert(rc != SC_SOCK_ERROR);
|
|
|
|
sleep(2);
|
|
|
|
fail_setsockopt = 3;
|
|
|
|
assert(sc_sock_accept(&sock, &in) == -1);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
assert(sc_sock_term(&in) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, false, SC_SOCK_UNIX);
|
|
|
|
assert(sc_sock_listen(&sock, "/tmp/x", "8080") == 0);
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_UNIX);
|
|
|
|
fail_setsockopt = 1;
|
|
|
|
rc = sc_sock_connect(&client, "/tmp/x", "8080", NULL, NULL);
|
|
|
|
assert(rc == SC_SOCK_ERROR);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
assert(sc_sock_term(&in) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&sock, 0, false, SC_SOCK_UNIX);
|
|
|
|
assert(sc_sock_listen(&sock, "/tmp/x", "8080") == 0);
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_UNIX);
|
|
|
|
fail_setsockopt = 2;
|
|
|
|
rc = sc_sock_connect(&client, "/tmp/x", "8080", NULL, NULL);
|
|
|
|
assert(rc == SC_SOCK_ERROR);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
assert(sc_sock_term(&in) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_setsockopt = 1;
|
|
|
|
rc = sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL);
|
|
|
|
assert(rc == SC_SOCK_ERROR);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_setsockopt = 2;
|
|
|
|
rc = sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL);
|
|
|
|
assert(rc == SC_SOCK_ERROR);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_setsockopt = 3;
|
|
|
|
rc = sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL);
|
|
|
|
assert(rc == SC_SOCK_ERROR);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_setsockopt = 4;
|
|
|
|
rc = sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL);
|
|
|
|
assert(rc == SC_SOCK_ERROR);
|
|
|
|
fail_setsockopt = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
sc_sock_listen(&client, "127.0.0.1", "8080");
|
|
|
|
fail_inet_ntop = 1;
|
|
|
|
sc_sock_print(&client, NULL, 0);
|
|
|
|
fail_inet_ntop = INT32_MAX;
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
void sock_fail_test3()
|
|
|
|
{
|
|
|
|
struct sc_sock client;
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_send = 1;
|
|
|
|
fail_send_errno = EAGAIN;
|
|
|
|
fail_send_err = -1;
|
|
|
|
assert(sc_sock_send(&client, NULL, 10, 0) == SC_SOCK_WANT_WRITE);
|
|
|
|
sc_sock_term(&client);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_send = 1;
|
|
|
|
fail_send_err = -1;
|
|
|
|
fail_send_errno = EINTR;
|
|
|
|
assert(sc_sock_send(&client, NULL, 10, 0) == SC_SOCK_ERROR);
|
|
|
|
sc_sock_term(&client);
|
|
|
|
|
|
|
|
fail_send = INT32_MAX;
|
|
|
|
fail_send_err = 0;
|
|
|
|
fail_send_errno = 0;
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_recv = 1;
|
|
|
|
fail_recv_errno = EAGAIN;
|
|
|
|
fail_recv_err = -1;
|
|
|
|
assert(sc_sock_recv(&client, NULL, 10, 0) == SC_SOCK_WANT_READ);
|
|
|
|
sc_sock_term(&client);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_recv = 1;
|
|
|
|
fail_recv_err = -1;
|
|
|
|
fail_recv_errno = EINTR;
|
|
|
|
assert(sc_sock_recv(&client, NULL, 10, 0) == SC_SOCK_ERROR);
|
|
|
|
sc_sock_term(&client);
|
|
|
|
|
|
|
|
fail_recv = INT32_MAX;
|
|
|
|
fail_recv_err = 0;
|
|
|
|
fail_recv_errno = 0;
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_connect_err = EINPROGRESS;
|
|
|
|
fail_connect = -1;
|
|
|
|
assert(sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL) ==
|
|
|
|
SC_SOCK_WANT_WRITE);
|
|
|
|
fail_connect_err = 0;
|
|
|
|
fail_connect = 0;
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_connect_err = -999;
|
|
|
|
fail_connect = -1;
|
|
|
|
assert(sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL) == -1);
|
|
|
|
fail_connect_err = 0;
|
|
|
|
fail_connect = 0;
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_INET);
|
|
|
|
fail_connect_err = EAGAIN;
|
|
|
|
fail_connect = -1;
|
|
|
|
assert(sc_sock_connect(&client, "127.0.0.1", "8080", NULL, NULL) ==
|
|
|
|
SC_SOCK_WANT_WRITE);
|
|
|
|
fail_connect_err = 0;
|
|
|
|
fail_connect = 0;
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&client, 0, false, SC_SOCK_UNIX);
|
|
|
|
fail_connect_err = EINPROGRESS;
|
|
|
|
fail_connect = -1;
|
|
|
|
assert(sc_sock_connect(&client, "/tmp/x", "8080", NULL, NULL) == -1);
|
|
|
|
fail_connect_err = 0;
|
|
|
|
fail_connect = 0;
|
|
|
|
assert(sc_sock_term(&client) == 0);
|
2020-11-25 09:25:38 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
#else
|
2021-02-08 23:01:44 +03:00
|
|
|
void sock_fail_test()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
void sock_fail_test2()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
void sock_fail_test3()
|
|
|
|
{
|
|
|
|
}
|
2021-02-07 20:42:08 +03:00
|
|
|
void poll_fail_test()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
void pipe_fail_test()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void *server(void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-07 22:31:04 +03:00
|
|
|
(void) arg;
|
2020-11-25 09:25:38 +03:00
|
|
|
int write_triggered = 2;
|
|
|
|
int rc, received = 0;
|
|
|
|
struct sc_sock server;
|
|
|
|
struct sc_sock accepted;
|
|
|
|
struct sc_sock_poll poll;
|
|
|
|
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
2021-02-04 02:35:34 +03:00
|
|
|
assert(*sc_sock_poll_err(&poll) == '\0');
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
assert(sc_sock_poll_init(&poll) == 0);
|
|
|
|
|
|
|
|
sc_sock_init(&server, 0, true, SC_SOCK_INET);
|
|
|
|
rc = sc_sock_listen(&server, "127.0.0.1", "11000");
|
|
|
|
assert(rc == 0);
|
|
|
|
|
|
|
|
rc = sc_sock_poll_add(&poll, &server.fdt, SC_SOCK_READ, &server);
|
|
|
|
assert(rc == 0);
|
|
|
|
bool done = false;
|
|
|
|
|
|
|
|
while (!done) {
|
|
|
|
int count = sc_sock_poll_wait(&poll, -1);
|
|
|
|
assert(rc != -1);
|
|
|
|
|
|
|
|
for (int i = 0; i < count; i++) {
|
|
|
|
int ev = sc_sock_poll_event(&poll, i);
|
2021-02-01 07:11:12 +03:00
|
|
|
struct sc_sock *sock = sc_sock_poll_data(&poll, i);
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
if (ev == 0) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (sock == &server) {
|
|
|
|
if (ev & SC_SOCK_READ) {
|
|
|
|
rc = sc_sock_accept(&server, &accepted);
|
|
|
|
printf("accepted \n");
|
|
|
|
|
|
|
|
assert(rc == 0);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_poll_add(&poll, &accepted.fdt, SC_SOCK_WRITE,
|
2020-11-25 09:25:38 +03:00
|
|
|
&accepted) == 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ev & SC_SOCK_WRITE) {
|
|
|
|
assert(false);
|
|
|
|
}
|
2021-02-01 07:11:12 +03:00
|
|
|
} else {
|
2020-11-25 09:25:38 +03:00
|
|
|
if (ev & SC_SOCK_READ) {
|
|
|
|
char buf[8];
|
2020-12-28 11:03:24 +03:00
|
|
|
rc = sc_sock_recv(&accepted, buf, sizeof(buf), 0);
|
2020-11-25 09:25:38 +03:00
|
|
|
if (rc == 8) {
|
|
|
|
assert(strcmp(buf, "dataxxx") == 0);
|
|
|
|
received++;
|
2021-02-01 07:11:12 +03:00
|
|
|
} else if (rc == 0 || rc < 0) {
|
2020-11-25 09:25:38 +03:00
|
|
|
rc = sc_sock_poll_del(&poll, &accepted.fdt,
|
2021-02-01 07:11:12 +03:00
|
|
|
SC_SOCK_READ | SC_SOCK_WRITE,
|
2020-11-25 09:25:38 +03:00
|
|
|
&accepted);
|
|
|
|
assert(rc == 0);
|
|
|
|
rc = sc_sock_term(&accepted);
|
|
|
|
assert(rc == 0);
|
|
|
|
done = true;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (write_triggered > 0) {
|
|
|
|
write_triggered--;
|
|
|
|
if (write_triggered == 0) {
|
2021-02-01 07:11:12 +03:00
|
|
|
rc = sc_sock_poll_del(&poll, &accepted.fdt,
|
|
|
|
SC_SOCK_WRITE, &accepted);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(rc == 0);
|
2021-02-01 07:11:12 +03:00
|
|
|
} else {
|
|
|
|
rc = sc_sock_poll_add(&poll, &accepted.fdt,
|
|
|
|
SC_SOCK_READ, &accepted);
|
2020-11-25 09:25:38 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
assert(write_triggered == 0);
|
|
|
|
assert(sc_sock_term(&server) == 0);
|
|
|
|
assert(sc_sock_poll_term(&poll) == 0);
|
|
|
|
|
|
|
|
printf("Received :%d \n", received);
|
|
|
|
assert(received == 10000);
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void *client(void *arg)
|
2020-11-25 09:25:38 +03:00
|
|
|
{
|
2021-02-07 22:31:04 +03:00
|
|
|
(void) arg;
|
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
int rc;
|
|
|
|
struct sc_sock sock;
|
|
|
|
|
|
|
|
for (int i = 0; i < 10; i++) {
|
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
|
|
|
|
|
|
|
rc = sc_sock_connect(&sock, "127.0.0.1", "11000", NULL, NULL);
|
|
|
|
if (rc != 0) {
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
|
|
|
sleep(1);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (int j = 0; j < 10000; j++) {
|
2020-12-28 11:03:24 +03:00
|
|
|
rc = sc_sock_send(&sock, "dataxxx", 8, 0);
|
2020-11-25 09:25:38 +03:00
|
|
|
assert(rc == 8);
|
|
|
|
}
|
|
|
|
|
|
|
|
rc = sc_sock_term(&sock);
|
|
|
|
assert(rc == 0);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
void test_poll()
|
|
|
|
{
|
|
|
|
struct sc_thread thread1;
|
|
|
|
struct sc_thread thread2;
|
|
|
|
|
|
|
|
sc_thread_init(&thread1);
|
|
|
|
sc_thread_init(&thread2);
|
|
|
|
|
|
|
|
assert(sc_thread_start(&thread1, server, NULL) == 0);
|
|
|
|
assert(sc_thread_start(&thread2, client, NULL) == 0);
|
|
|
|
|
|
|
|
assert(sc_thread_term(&thread1) == 0);
|
|
|
|
assert(sc_thread_term(&thread2) == 0);
|
|
|
|
}
|
|
|
|
|
2021-02-01 07:11:12 +03:00
|
|
|
void test_err()
|
|
|
|
{
|
|
|
|
struct sc_sock sock;
|
|
|
|
char buf[128];
|
|
|
|
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_listen(&sock, "127.0.0.1x", "8004") != 0);
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_UNIX);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_listen(&sock, "/", "8004") != 0);
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET6);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_listen(&sock, "/", "8004") != 0);
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET6);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_listen(&sock, "0.0.0.0", "99999") != 0);
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_listen(&sock, "0.0.0.3", "99999") != 0);
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET6);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_connect(&sock, "::1", "8006", NULL, NULL) != 0);
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_UNIX);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_connect(&sock, "/", "8006", NULL, NULL) != 0);
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_connect(&sock, "0.0.0.0", "8006", NULL, NULL) != 0);
|
|
|
|
sc_sock_print(&sock, buf, 128);
|
|
|
|
sc_sock_local_str(&sock, buf, 128);
|
|
|
|
assert(*buf == '\0');
|
|
|
|
sc_sock_remote_str(&sock, buf, 128);
|
|
|
|
assert(*buf == '\0');
|
|
|
|
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_connect(&sock, "0.0.0.0", "8006", "127.0.0.1", "8080") != 0);
|
2021-02-04 11:17:43 +03:00
|
|
|
sc_sock_init(&sock, 0, true, SC_SOCK_INET6);
|
2021-02-01 07:11:12 +03:00
|
|
|
assert(sc_sock_connect(&sock, "0.0.0.0", "8006", "::1", "8080") != 0);
|
|
|
|
assert(sc_sock_term(&sock) == 0);
|
2021-02-04 11:17:43 +03:00
|
|
|
|
|
|
|
struct sc_sock_poll p = {0};
|
|
|
|
assert(sc_sock_poll_wait(&p, 100) != 0);
|
|
|
|
assert(*sc_sock_poll_err(&p) != '\0');
|
2021-02-01 07:11:12 +03:00
|
|
|
}
|
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
int main()
|
|
|
|
{
|
2021-02-08 23:01:44 +03:00
|
|
|
#ifdef SC_HAVE_WRAP
|
|
|
|
assert(sc_mutex_init(&mutex) == 0);
|
|
|
|
#endif
|
|
|
|
|
2021-02-14 16:29:50 +03:00
|
|
|
assert(sc_sock_startup() == 0);
|
2020-11-25 09:25:38 +03:00
|
|
|
|
|
|
|
test1();
|
|
|
|
test_ip4();
|
|
|
|
|
|
|
|
#if defined(__x86_64__)
|
|
|
|
test_ip6();
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#if !defined(__APPLE__)
|
|
|
|
test_unix();
|
|
|
|
#endif
|
|
|
|
test_pipe();
|
|
|
|
pipe_fail_test();
|
2021-02-07 20:42:08 +03:00
|
|
|
poll_fail_test();
|
2021-02-08 23:01:44 +03:00
|
|
|
sock_fail_test();
|
|
|
|
sock_fail_test2();
|
|
|
|
sock_fail_test3();
|
2020-11-25 09:25:38 +03:00
|
|
|
test_poll();
|
2021-02-01 07:11:12 +03:00
|
|
|
test_err();
|
2021-02-07 20:42:08 +03:00
|
|
|
test_poll_mass();
|
2020-11-25 09:25:38 +03:00
|
|
|
|
2021-02-14 16:29:50 +03:00
|
|
|
assert(sc_sock_cleanup() == 0);
|
2021-02-08 23:01:44 +03:00
|
|
|
|
|
|
|
#ifdef SC_HAVE_WRAP
|
|
|
|
assert(sc_mutex_term(&mutex) == 0);
|
|
|
|
test_done = 1;
|
|
|
|
#endif
|
|
|
|
|
2020-11-25 09:25:38 +03:00
|
|
|
return 0;
|
|
|
|
}
|