WHCandCursor
e853abac5e
Login claims the chair, writes a session ticket, and starts a background service for discovery plus revision-based context sync. Password Realm opens require --login (or CST_REALM_PASSWORD); --service/--service-stop manage the resident agent. Co-authored-by: Cursor <cursoragent@cursor.com>
89 lines
2.5 KiB
C
89 lines
2.5 KiB
C
#define _POSIX_C_SOURCE 200809L
|
|
#include "cst_net.h"
|
|
#include "cst_peer.h"
|
|
#include "cst_realm.h"
|
|
#include "cst_test_util.h"
|
|
|
|
#include <assert.h>
|
|
#include <pthread.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
typedef struct {
|
|
os_socket_t listen;
|
|
cst_realm_cert_t cert;
|
|
uint8_t sk[64];
|
|
uint8_t realm_pk[32];
|
|
int ok;
|
|
char remote_id[64];
|
|
} srv_t;
|
|
|
|
static void *server_main(void *p) {
|
|
srv_t *s = p;
|
|
cst_peer_session_t sess;
|
|
char err[128];
|
|
s->ok = cst_peer_accept((intptr_t)s->listen, &s->cert, s->sk, s->realm_pk,
|
|
5000, &sess, err, sizeof(err)) == 0;
|
|
if (s->ok) {
|
|
snprintf(s->remote_id, sizeof(s->remote_id), "%s", sess.remote.instance_id);
|
|
cst_peer_close(&sess);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
int main(void) {
|
|
os_net_init();
|
|
cst_realm_keys_t realm;
|
|
assert(cst_realm_generate(&realm) == 0);
|
|
|
|
cst_realm_cert_t ca, cb;
|
|
uint8_t sa[32], ska[64], sb[32], skb[64];
|
|
assert(cst_realm_enroll(&realm, "peer-a", "A", 0, &ca, sa, ska) == 0);
|
|
assert(cst_realm_enroll(&realm, "peer-b", "B", 0, &cb, sb, skb) == 0);
|
|
|
|
char err[128];
|
|
os_socket_t ls = os_net_listen("127.0.0.1", 0, err, sizeof(err));
|
|
assert(ls != OS_SOCKET_BAD);
|
|
int port = os_net_port(ls);
|
|
assert(port > 0);
|
|
|
|
srv_t srv = {.listen = ls, .cert = cb};
|
|
memcpy(srv.sk, skb, 64);
|
|
memcpy(srv.realm_pk, realm.pk, 32);
|
|
pthread_t th;
|
|
assert(pthread_create(&th, NULL, server_main, &srv) == 0);
|
|
|
|
cst_peer_session_t sess;
|
|
assert(cst_peer_dial("127.0.0.1", port, &ca, ska, realm.pk, 5000, &sess, err,
|
|
sizeof(err)) == 0);
|
|
assert(!strcmp(sess.remote.instance_id, "peer-b"));
|
|
cst_peer_close(&sess);
|
|
pthread_join(th, NULL);
|
|
assert(srv.ok);
|
|
assert(!strcmp(srv.remote_id, "peer-a"));
|
|
|
|
/* Foreign realm must fail. */
|
|
cst_realm_keys_t other;
|
|
assert(cst_realm_generate(&other) == 0);
|
|
cst_realm_cert_t cf;
|
|
uint8_t sf[32], skf[64];
|
|
assert(cst_realm_enroll(&other, "peer-f", "F", 0, &cf, sf, skf) == 0);
|
|
os_socket_t ls2 = os_net_listen("127.0.0.1", 0, err, sizeof(err));
|
|
assert(ls2 != OS_SOCKET_BAD);
|
|
int port2 = os_net_port(ls2);
|
|
srv_t srv2 = {.listen = ls2, .cert = cb};
|
|
memcpy(srv2.sk, skb, 64);
|
|
memcpy(srv2.realm_pk, realm.pk, 32);
|
|
assert(pthread_create(&th, NULL, server_main, &srv2) == 0);
|
|
assert(cst_peer_dial("127.0.0.1", port2, &cf, skf, other.pk, 3000, &sess, err,
|
|
sizeof(err)) != 0);
|
|
pthread_join(th, NULL);
|
|
assert(!srv2.ok);
|
|
|
|
os_net_close(ls);
|
|
os_net_close(ls2);
|
|
os_net_cleanup();
|
|
puts("test_peer_auth: OK");
|
|
return 0;
|
|
}
|