summaryrefslogtreecommitdiffstats
path: root/tests/test_client.c
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_client.c')
-rw-r--r--[-rwxr-xr-x]tests/test_client.c546
1 files changed, 404 insertions, 142 deletions
diff --git a/tests/test_client.c b/tests/test_client.c
index f8ef157..c469c74 100755..100644
--- a/tests/test_client.c
+++ b/tests/test_client.c
@@ -1,47 +1,92 @@
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
+#include <sys/select.h>
+#include <unistd.h>
#include "../PICA_client.h"
-
+#include "../PICA_proto.h"
struct sockaddr_in a;
-struct PICA_conninfo *c;
-struct PICA_chaninfo *chn;
+struct PICA_acc *acc;
+struct PICA_c2n *c;
+struct PICA_c2c *chn;
+enum PICA_directc2c_config directc2c_cfg = PICA_DIRECTC2C_CFG_DISABLED;
+struct PICA_listener *l = NULL;
+
+
+int _outgoing_chnl = 0;
-int _outgoing_chnl=0;
+int connected_to_node = 0;
+int c2c_active = 0;
+int c2c_in_progress = 0;
-unsigned char buf[64*1024];
+int echo_mode = 0;
+int random_message_mode = 0;
+int delivery_acks_count = 0;
+int rmm_wait_for_reply = 0;
+
+unsigned char buf[64 * 1024];
int accept_cb(const unsigned char *id)
{
-printf("accept_cb: %s\n",PICA_id_to_base64(id, NULL));
-return 1;
+ printf("accept_cb: %s\n", PICA_id_to_base64(id, NULL));
+ return 1;
}
//получение сообщения.
-void newmsg_cb(const unsigned char *peer_id,char* msgbuf,unsigned int nb,int type)
+void newmsg_cb(const unsigned char *peer_id, const char* msgbuf, unsigned int nb, int type)
{
-memcpy(buf,msgbuf,nb);
-buf[nb]=0;
-printf(" %s: ",PICA_id_to_base64(peer_id, NULL));
-puts(buf);
+
+ if (echo_mode)
+ {
+ int ret;
+
+ ret = PICA_send_msg(chn, msgbuf, nb);
+
+ if (ret != PICA_OK)
+ printf("failed to send message, error = %i\n", ret);
+
+ return;
+ }
+
+ if (random_message_mode)
+ {
+ //compare received & sent messages, fail if they do not match exactly
+ if (memcmp(msgbuf, buf, nb) != 0)
+ {
+ printf("sent and returned messages are not equal!\n");
+ exit(1);
+ }
+
+ rmm_wait_for_reply = 0;
+ }
+
+ memcpy(buf, msgbuf, nb);
+ buf[nb] = 0;
+ printf(" %s: ", PICA_id_to_base64(peer_id, NULL));
+ puts(buf);
}
//получение подтверждения о доставке сообщения
void msgok_cb(const unsigned char *peer_id)
{
-puts("[V]");
+ delivery_acks_count++;
+ puts("[V]");
}
//создание канала с собеседником
-void channel_established_cb(const unsigned char *peer_id)
+void c2c_established_cb(const unsigned char *peer_id)
{
-puts("channel_established_cb");
+ c2c_active = 1;
+ c2c_in_progress = 0;
+ puts("c2c_established_cb");
+ chn = c->chan_list_head;//hack, its better to pass new c2c pointer as callback arg
}
-void channel_failed_cb(const unsigned char *peer_id)
+void c2c_failed_cb(const unsigned char *peer_id)
{
-printf("Failed to create channel to %s\n",PICA_id_to_base64(peer_id, NULL));
-chn=0;
+ c2c_in_progress = 0;
+ printf("Failed to create c2c to %s\n", PICA_id_to_base64(peer_id, NULL));
+ chn = 0;
}
//входящий запрос на создание канала от пользователя с номером caller_id
@@ -49,191 +94,408 @@ chn=0;
//запрошенный пользователь не найден, в оффлайне или отказался от общения
void notfound_cb(const unsigned char *callee_id)
{
-puts("notfound_cb");
+ puts("notfound_cb");
}
-void channel_closed_cb(const unsigned char *peer_id, int reason)
+void c2c_closed_cb(const unsigned char *peer_id, int reason)
{
-printf("channel_closed_cb( peer_id = %s, reason = %i)\n", PICA_id_to_base64(peer_id, NULL), reason);
+ c2c_active = 0;
+ c2c_in_progress = 0;
+ printf("c2c_closed_cb( peer_id = %s, reason = %i)\n", PICA_id_to_base64(peer_id, NULL), reason);
}
void nodelist_cb(int type, void *addr_bin, const char *addr_str, unsigned int port)
{
-switch(type)
- {
- case 0xA0:
+ switch(type)
+ {
+ case 0xA0:
printf("node (IP %s port %u)\n", addr_str, port);
break;
-
- case 0xA1:
+
+ case 0xA1:
printf("node (IPv6 %s port %u)", addr_str, port);
break;
-
- case 0xA2:
+
+ case 0xA2:
printf("node (hostname %s port %u)\n", addr_str, port);
break;
-
- default:
+
+ default:
puts("unknown node address type");
return;
- }
+ }
}
int peer_cert_verify_cb(const unsigned char *peer_id, const char *cert_pem, unsigned int nb)
{
- printf("peer's certificate (%s): \n%.*s\n", PICA_id_to_base64(peer_id, NULL), nb,cert_pem);
- return 1;
+ printf("peer's certificate (%s): \n%.*s\n", PICA_id_to_base64(peer_id, NULL), nb, cert_pem);
+ return 1;
}
-struct PICA_client_callbacks cbs = {
- newmsg_cb,
- msgok_cb,
- channel_established_cb,
- channel_failed_cb,
- accept_cb,
- notfound_cb,
- channel_closed_cb,
- nodelist_cb,
- peer_cert_verify_cb
-};
-
-int main(int argc,char** argv)
+int accept_file_cb(const unsigned char *peer_id, uint64_t file_size, const char *filename, unsigned int filename_size)
{
-int ret;
-unsigned char peer_id[PICA_ID_SIZE];
+ int ret;
+ char namebuf[256];
+
+ printf("accept_file_cb: peer_id %s filename: %.*s\n", PICA_id_to_base64(peer_id, NULL), filename_size, filename);
+
+ sprintf(namebuf, "/tmp/received_%.*s", filename_size, filename);
-if (argc<4)
+ ret = PICA_accept_file(chn, namebuf, strlen(namebuf));
+
+ if (ret != PICA_OK)
{
- puts("usage: test_client address port cert_filename [peer id]");
- return 0;
+ printf("faild to accept file, errror %i\n", ret);
+ return 0;
}
-puts(SSLeay_version(SSLEAY_VERSION));
-printf("Calling PICA_init...\n");
+ return 2;
+}
+
+void accepted_file_cb(const unsigned char *peer_id)
+{
+ printf("accepted_file_cb: peer_id %s \n", PICA_id_to_base64(peer_id, NULL));
+}
+
+void denied_file_cb(const unsigned char *peer_id)
+{
+ printf("denied_file_cb: peer_id %s \n", PICA_id_to_base64(peer_id, NULL));
+}
-PICA_client_init(&cbs);
+void file_progress(const unsigned char *peer_id, uint64_t sent, uint64_t received)
+{
+ printf("file_progress: peer_id %s\n", PICA_id_to_base64(peer_id, NULL));
+}
-// printf("setting address...\n");
-//
-// memset(&a,0,sizeof(a));
-//
-// a.sin_family = AF_INET;
-// a.sin_addr.s_addr = inet_addr (argv[1]);//("127.0.0.1");
-// a.sin_port = htons(51914);
+void file_control(const unsigned char *peer_id, unsigned int sender_cmd, unsigned int receiver_cmd)
+{
+ printf("file_control: peer_id %s sender_cmd %u receiver_cmd %u \n", PICA_id_to_base64(peer_id, NULL), sender_cmd, receiver_cmd);
+}
+void file_finished(const unsigned char *peer_id, int sending)
+{
+ printf("file_finished: peer_id %s\n", PICA_id_to_base64(peer_id, NULL));
+}
-printf("making connection...\n");
+void c2n_established_cb(struct PICA_c2n *c2n)
+{
+ connected_to_node = 1;
+ printf("c2n_established_cb: %p\n my_id %s\n", c2n, PICA_id_to_base64(c2n->acc->id, NULL));
+}
-//PICA_new_connection(const char *nodeaddr, unsigned int port, const char *CA_file, const char *cert_file, const char *pkey_file, const char* password, struct PICA_conninfo **ci)
-ret=PICA_new_connection(argv[1], atoi(argv[2]), /*"trusted_CA.pem"*/ argv[3], argv[3],argv[3],NULL,&c);
+void c2n_failed_cb(struct PICA_c2n *c2n, int error)
+{
+ printf("c2n_failed_cb: %p error_code %i\n", c2n, error);
+ c = NULL;
+}
-ERR_print_errors_fp(stdout);
+void c2n_closed_cb(struct PICA_c2n *c2n, int error)
+{
+ printf("c2n_closed_cb: %p error_code %i\n", c2n, error);
+ c = NULL;
+}
-if (ret==PICA_OK)
-printf("PICA_OK Connected to server...\n");
-else
+void listener_error(struct PICA_listener *lst, int errorcode)
{
-printf("PICA_new_connection: ret=%i\n",ret);
-return 1;
+ printf("listener_error: %p error code %i\n", lst, errorcode);
+ l = NULL;
}
-if (argc==5)
+struct PICA_client_callbacks cbs =
+{
+ newmsg_cb,
+ msgok_cb,
+ c2c_established_cb,
+ c2c_failed_cb,
+ accept_cb,
+ notfound_cb,
+ c2c_closed_cb,
+ nodelist_cb,
+ peer_cert_verify_cb,
+ accept_file_cb,
+ accepted_file_cb,
+ denied_file_cb,
+ file_progress,
+ file_control,
+ file_finished,
+ c2n_established_cb,
+ c2n_failed_cb,
+ c2n_closed_cb,
+ listener_error
+};
+
+int main(int argc, char** argv)
+{
+ int ret, opt;
+ unsigned char peer_id[PICA_ID_SIZE];
+ struct PICA_c2n* c2ns[] = {NULL, NULL};
+
+ fd_set stdinfds;
+ struct timeval tv;
+
+ const char *nodeaddr = NULL, *nodeport = NULL,
+ *certfile = NULL, *peerid = NULL,
+ *localdirectaddr = NULL, *extport = NULL,
+ *filetosend = NULL,
+ *locport = NULL;
+
+ while ((opt = getopt(argc, argv, "a:p:c:i:d:e:l:f:tmr")) != -1)
{
- if (PICA_id_from_base64(argv[4], peer_id) == NULL)
+ switch(opt)
+ {
+ case 'a':
+ nodeaddr = optarg;
+ break;
+
+ case 'p':
+ nodeport = optarg;
+ break;
+
+ case 'c':
+ certfile = optarg;
+ break;
+
+ case 'i':
+ peerid = optarg;
+ break;
+
+ case 'd':
+ localdirectaddr = optarg;
+ break;
+
+ case 'e':
+ extport = optarg;
+ break;
+
+ case 'f':
+ filetosend = optarg;
+ break;
+
+ case 'l':
+ locport = optarg;
+ break;
+
+ case 't':
+ directc2c_cfg = PICA_DIRECTC2C_CFG_CONNECTONLY;
+ break;
+
+ case 'm':
+ echo_mode = 1;
+ break;
+
+ case 'r':
+ random_message_mode = 1;
+ break;
+
+ default:
+ fprintf(stderr, "invalid arg %s\n", optarg);
+ return -1;
+ }
+ }
+
+ if (echo_mode && random_message_mode)
{
- puts("invalid peer_id");
- return 1;
+ puts("echo mode and random test messages mode are mutually exclusive, please select only one of them");
+ return -1;
}
- printf("Creating channel to %s...\n",PICA_id_to_base64(peer_id, NULL));
-
- ret=PICA_create_channel(c,peer_id,&chn);
- //sleep(17);//timeout test
- if (ret!=PICA_OK)
- {
- printf("error: ret=%i\n",ret);
+ if (argc < 4 || !nodeaddr || !nodeport || !certfile)
+ {
+ puts("usage: test_client -a node address -p node port -c cert_filename [-i peer id]\n\
+ -d local address for direct incoming c2c connections\n\
+ -e external port for incoming direct connections\n\
+ -l internal port to listen for incoming direct connections\n\
+ -t enable connect-only mode for direct c2c connections\n\
+ -m start in echo mode, received messages are sent back, input from stdin is not accepted\n\
+ -r send random test messages and exit, expects other side to be run in echo mode\n\
+cert_filename should point to file that contains client certificate, private key and Diffie-Hellman parameters in PEM format");
+ return 0;
+ }
+
+
+
+ puts(SSLeay_version(SSLEAY_VERSION));
+ printf("Calling PICA_init...\n");
+
+ PICA_client_init(&cbs);
+
+ printf("opening account...");
+ ret = PICA_open_acc(certfile, certfile, certfile, NULL, &acc);
+
+ if (ret != PICA_OK)
+ {
+ fprintf(stderr, "failed to open account, ret = %i\n", ret);
return 1;
- }
-
}
-do
+ if (localdirectaddr)
{
- if (chn && chn->state==PICA_CHANSTATE_ACTIVE)
+ int intp = 2300, extp = 2300;
+
+ directc2c_cfg = PICA_DIRECTC2C_CFG_ALLOWINCOMING;
+
+ if (extport)
+ extp = atoi(extport);
+ if (locport)
+ intp = atoi(locport);
+
+ printf("new listener on %s port %i external announced port %i\n", localdirectaddr, intp, extp);
+
+ ret = PICA_new_listener(acc, localdirectaddr, extp, intp, &l);
+
+ if (ret != PICA_OK)
{
- unsigned int msglen;
- gets(buf);
- msglen=strlen(buf);
-
- ret=PICA_send_msg(chn,buf,msglen);
-
- if (ret!=PICA_OK)
- printf("PICA_send_msg ret=%i\n",ret);
+ fprintf(stderr, "failed to create listener, error code: %i\n", ret);
+ l = NULL;
}
+
}
-while(PICA_read(c,100000) == PICA_OK && PICA_write(c) == PICA_OK);
-/*
-puts("Enter peer_id to talk with or . for waiting incoming sessions:");
-gets(buf);
-if (buf[0]=='.')
+ printf("making connection...\n");
+
+//PICA_new_c2n(const char *nodeaddr, unsigned int port, const char *CA_file, const char *cert_file, const char *pkey_file, const char* password, struct PICA_c2n **ci)
+ ret = PICA_new_c2n(acc, nodeaddr, atoi(nodeport), directc2c_cfg, l, &c);
+
+ ERR_print_errors_fp(stdout);
+
+ if (ret == PICA_OK)
+ printf("PICA_OK started c2n connection to node...\n");
+ else
{
- unsigned int br;
- ret=PICA_read_conn(c,accept_cb,&chn);
+ printf("PICA_new_c2n: ret=%i\n", ret);
+ return 1;
+ }
+
+ c2ns[0] = c;
+
- if (ret==PICA_NEWCHAN)
+ while(PICA_event_loop(c2ns, 500) == PICA_OK && c != NULL)
+ {
+ //puts("event loop...");
+ if (peerid && connected_to_node == 1 && c2c_active == 0 && c2c_in_progress == 0)
+ {
+ if (PICA_id_from_base64(peerid, peer_id) == NULL)
{
- puts("PICA_NEWCHAN!");
+ puts("invalid peer_id");
+ return 1;
}
- else
- printf("read_conn error: ret=%i\n",ret);
-
- ret=PICA_read_msg(chn,buf,&br);
- switch(ret)
+
+ printf("Creating c2c to %s...\n", PICA_id_to_base64(peer_id, NULL));
+
+ ret = PICA_new_c2c(c, peer_id, l, &chn);
+ //sleep(17);//timeout test
+ if (ret != PICA_OK)
{
- case PICA_MSGUTF8:
- printf("%u (%u bytes): ",c->id,br);
- puts(buf);
- break;
- case PICA_MSGOK:
- puts("[v]");
- break;
+ printf("error: ret=%i\n", ret);
+ return 1;
+ }
+
+ printf("chn = %p\n", chn);
+ c2c_in_progress = 1;
+
+ }
+
+ if (random_message_mode && connected_to_node == 1 && c2c_active == 1 && rmm_wait_for_reply == 0)
+ {
+ static int test_stage = 0;
+
+ switch(test_stage)
+ {
+ //send zero-length message
+ case 0:
+ ret = PICA_send_msg(chn, NULL, 0);
+ test_stage++;
+ rmm_wait_for_reply = 1;
+ break;
+
+ //send max length message
+ case 1:
+ memset(buf, 'X', PICA_PROTO_C2CMSG_MAXDATASIZE);
+ ret = PICA_send_msg(chn, buf, PICA_PROTO_C2CMSG_MAXDATASIZE);
+ test_stage++;
+ rmm_wait_for_reply = 1;
+ break;
+
+ //send random length message
+ case 2:
+ RAND_pseudo_bytes((unsigned char*)&ret, sizeof ret);
+ if (ret < 0)
+ ret = 0 - ret;
+ ret %= PICA_PROTO_C2CMSG_MAXDATASIZE;
+ memset(buf, 'x', ret);
+ ret = PICA_send_msg(chn, buf, ret);
+ test_stage++;
+ rmm_wait_for_reply = 1;
+ break;
+
default:
- printf("PICA_read_msg ret=%i\n",ret);
+ puts("waiting for test messages to be delivered");
}
- }
-else
- {
- peer_id=atoi(buf);
- printf("Creating channel to %u...\n",peer_id);
-
- ret=PICA_create_channel(c,peer_id,&chn,1);
- if (ret==PICA_OK)
+ if (ret != PICA_OK)
+ {
+ printf("failed to send message, error = %i\n", ret);
+ return 1;
+ }
+
+ if (delivery_acks_count == 3)
+ break;//exit main loop on success
+ }
+
+ if (connected_to_node == 1 && c2c_active == 1 && !echo_mode && !random_message_mode)
{
- unsigned int msglen;
- puts("Channel established...");
- gets(buf);
- msglen=strlen(buf);
-
- ret=PICA_send_msg(chn,buf,msglen);
- if (ret!=PICA_OK)
- printf("PICA_send_msg ret=%i\n",ret);
+ FD_ZERO(&stdinfds);
+ FD_SET(STDIN_FILENO, &stdinfds);
+
+ tv.tv_sec = 0;
+ tv.tv_usec = 500000;
+
+ ret = select(STDIN_FILENO + 1, &stdinfds, NULL, NULL, &tv);
+
+ if (ret == -1)
+ {
+ perror("select() on stdin failed");
+ return 1;
+ }
+
+ if (ret > 0 && FD_ISSET(STDIN_FILENO, &stdinfds))
+ {
+ ret = read(STDIN_FILENO, buf, 256);
+
+ if (ret <= 0)
+ {
+ perror("read from stdin");
+ return 1;
+ }
+
+ ret = PICA_send_msg(chn, buf, ret);
+
+ if (ret != PICA_OK)
+ printf("failed to send message, error = %i\n", ret);
+ }
+ }
+
+ if (connected_to_node == 1 && c2c_active == 1 && filetosend)
+ {
+ printf("sending file %s to peer...\n", filetosend);
+ ret = PICA_send_file(chn, filetosend);
+
+ if (ret != PICA_OK)
+ printf("failed to send file %s, error = %i\n", filetosend, ret);
+
+ filetosend = NULL;
}
- else
- printf("error: ret=%i\n",ret);
- }
-*/
-sleep(10);
-if (chn)
- {
- puts("--------");
- PICA_close_channel(chn);
}
-PICA_close_connection(c);
-return 0;
+
+ if (c)
+ PICA_close_c2n(c);
+
+ if(l)
+ PICA_close_listener(l);
+
+ return 0;
}
generated by cgit v1.2.1 (git 2.18.0) at 2026-10-15 23:47:58 +0000