Commit 78b26d7a authored by Jake Cobb's avatar Jake Cobb

Problem: Throughput perf tool doesn't support CURVE

Solution: Add extra argument to enable CURVE, use
fixed keys to make local_thr and remote_thr compatible
with CURVE enabled.
parent 6652a0d7
...@@ -31,6 +31,12 @@ ...@@ -31,6 +31,12 @@
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
// keys are arbitrary but must match remote_lat.cpp
const char server_pubkey[] = "DX4nh=yUn{-9ugra0X3Src4SU-4xTgqxcYY.+<SH";
const char server_prvkey[] = "{X}#>t#jRGaQ}gMhv=30r(Mw+87YGs+5%kh=i@f8";
const char client_pubkey[] = "<n^oA}I:66W+*ds3tAmi1+KJzv-}k&fC2aA5Bj0K";
const char client_prvkey[] = "9R9bV}[6z6DC-%$!jTVTKvWc=LEL{4i4gzUe$@Zx";
int main (int argc, char *argv []) int main (int argc, char *argv [])
{ {
const char *bind_to; const char *bind_to;
...@@ -45,14 +51,18 @@ int main (int argc, char *argv []) ...@@ -45,14 +51,18 @@ int main (int argc, char *argv [])
unsigned long elapsed; unsigned long elapsed;
unsigned long throughput; unsigned long throughput;
double megabits; double megabits;
int curve = 0;
if (argc != 4) { if (argc != 4 && argc != 5) {
printf ("usage: local_thr <bind-to> <message-size> <message-count>\n"); printf ("usage: local_thr <bind-to> <message-size> <message-count> [<enable_curve>]\n");
return 1; return 1;
} }
bind_to = argv [1]; bind_to = argv [1];
message_size = atoi (argv [2]); message_size = atoi (argv [2]);
message_count = atoi (argv [3]); message_count = atoi (argv [3]);
if (argc >= 5 && atoi (argv [4])) {
curve = 1;
}
ctx = zmq_init (1); ctx = zmq_init (1);
if (!ctx) { if (!ctx) {
...@@ -68,6 +78,19 @@ int main (int argc, char *argv []) ...@@ -68,6 +78,19 @@ int main (int argc, char *argv [])
// Add your socket options here. // Add your socket options here.
// For example ZMQ_RATE, ZMQ_RECOVERY_IVL and ZMQ_MCAST_LOOP for PGM. // For example ZMQ_RATE, ZMQ_RECOVERY_IVL and ZMQ_MCAST_LOOP for PGM.
if (curve) {
rc = zmq_setsockopt (s, ZMQ_CURVE_SECRETKEY, server_prvkey, sizeof(server_prvkey));
if (rc != 0) {
printf ("error in zmq_setsockoopt: %s\n", zmq_strerror (errno));
return -1;
}
int server = 1;
rc = zmq_setsockopt (s, ZMQ_CURVE_SERVER, &server, sizeof(int));
if (rc != 0) {
printf ("error in zmq_setsockoopt: %s\n", zmq_strerror (errno));
return -1;
}
}
rc = zmq_bind (s, bind_to); rc = zmq_bind (s, bind_to);
if (rc != 0) { if (rc != 0) {
......
...@@ -32,6 +32,12 @@ ...@@ -32,6 +32,12 @@
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
// keys are arbitrary but must match local_lat.cpp
const char server_pubkey[] = "DX4nh=yUn{-9ugra0X3Src4SU-4xTgqxcYY.+<SH";
const char server_prvkey[] = "{X}#>t#jRGaQ}gMhv=30r(Mw+87YGs+5%kh=i@f8";
const char client_pubkey[] = "<n^oA}I:66W+*ds3tAmi1+KJzv-}k&fC2aA5Bj0K";
const char client_prvkey[] = "9R9bV}[6z6DC-%$!jTVTKvWc=LEL{4i4gzUe$@Zx";
int main (int argc, char *argv []) int main (int argc, char *argv [])
{ {
const char *connect_to; const char *connect_to;
...@@ -42,15 +48,19 @@ int main (int argc, char *argv []) ...@@ -42,15 +48,19 @@ int main (int argc, char *argv [])
int rc; int rc;
int i; int i;
zmq_msg_t msg; zmq_msg_t msg;
int curve = 0;
if (argc != 4) { if (argc != 4 && argc != 5) {
printf ("usage: remote_thr <connect-to> <message-size> " printf ("usage: remote_thr <connect-to> <message-size> "
"<message-count>\n"); "<message-count> [<enable_curve>]\n");
return 1; return 1;
} }
connect_to = argv [1]; connect_to = argv [1];
message_size = atoi (argv [2]); message_size = atoi (argv [2]);
message_count = atoi (argv [3]); message_count = atoi (argv [3]);
if (argc >= 5 && atoi (argv [4])) {
curve = 1;
}
ctx = zmq_init (1); ctx = zmq_init (1);
if (!ctx) { if (!ctx) {
...@@ -66,6 +76,25 @@ int main (int argc, char *argv []) ...@@ -66,6 +76,25 @@ int main (int argc, char *argv [])
// Add your socket options here. // Add your socket options here.
// For example ZMQ_RATE, ZMQ_RECOVERY_IVL and ZMQ_MCAST_LOOP for PGM. // For example ZMQ_RATE, ZMQ_RECOVERY_IVL and ZMQ_MCAST_LOOP for PGM.
if (curve) {
rc = zmq_setsockopt (s, ZMQ_CURVE_SECRETKEY, client_prvkey, sizeof (client_prvkey));
if (rc != 0) {
printf ("error in zmq_setsockoopt: %s\n", zmq_strerror (errno));
return -1;
}
rc = zmq_setsockopt (s, ZMQ_CURVE_PUBLICKEY, client_pubkey, sizeof (client_pubkey));
if (rc != 0) {
printf ("error in zmq_setsockoopt: %s\n", zmq_strerror (errno));
return -1;
}
rc = zmq_setsockopt (s, ZMQ_CURVE_SERVERKEY, server_pubkey, sizeof (server_pubkey));
if (rc != 0) {
printf ("error in zmq_setsockoopt: %s\n", zmq_strerror (errno));
return -1;
}
}
rc = zmq_connect (s, connect_to); rc = zmq_connect (s, connect_to);
if (rc != 0) { if (rc != 0) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment