|
Packit Service |
4684c1 |
/* This example code is placed in the public domain. */
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
#ifdef HAVE_CONFIG_H
|
|
Packit Service |
4684c1 |
#include <config.h>
|
|
Packit Service |
4684c1 |
#endif
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
#include <stdio.h>
|
|
Packit Service |
4684c1 |
#include <stdlib.h>
|
|
Packit Service |
4684c1 |
#include <string.h>
|
|
Packit Service |
4684c1 |
#include <sys/types.h>
|
|
Packit Service |
4684c1 |
#include <sys/socket.h>
|
|
Packit Service |
4684c1 |
#include <arpa/inet.h>
|
|
Packit Service |
4684c1 |
#include <unistd.h>
|
|
Packit Service |
4684c1 |
#include <assert.h>
|
|
Packit Service |
4684c1 |
#include <gnutls/gnutls.h>
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
/* A very basic TLS client, with PSK authentication.
|
|
Packit Service |
4684c1 |
*/
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
#define CHECK(x) assert((x)>=0)
|
|
Packit Service |
4684c1 |
#define LOOP_CHECK(rval, cmd) \
|
|
Packit Service |
4684c1 |
do { \
|
|
Packit Service |
4684c1 |
rval = cmd; \
|
|
Packit Service |
4684c1 |
} while(rval == GNUTLS_E_AGAIN || rval == GNUTLS_E_INTERRUPTED); \
|
|
Packit Service |
4684c1 |
assert(rval >= 0)
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
#define MAX_BUF 1024
|
|
Packit Service |
4684c1 |
#define MSG "GET / HTTP/1.0\r\n\r\n"
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
extern int tcp_connect(void);
|
|
Packit Service |
4684c1 |
extern void tcp_close(int sd);
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
int main(void)
|
|
Packit Service |
4684c1 |
{
|
|
Packit Service |
4684c1 |
int ret, sd, ii;
|
|
Packit Service |
4684c1 |
gnutls_session_t session;
|
|
Packit Service |
4684c1 |
char buffer[MAX_BUF + 1];
|
|
Packit Service |
4684c1 |
const char *err;
|
|
Packit Service |
4684c1 |
gnutls_psk_client_credentials_t pskcred;
|
|
Packit Service |
4684c1 |
const gnutls_datum_t key = { (void *) "DEADBEEF", 8 };
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
if (gnutls_check_version("3.6.3") == NULL) {
|
|
Packit Service |
4684c1 |
fprintf(stderr, "GnuTLS 3.6.3 or later is required for this example\n");
|
|
Packit Service |
4684c1 |
exit(1);
|
|
Packit Service |
4684c1 |
}
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
CHECK(gnutls_global_init());
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
CHECK(gnutls_psk_allocate_client_credentials(&pskcred));
|
|
Packit Service |
4684c1 |
CHECK(gnutls_psk_set_client_credentials(pskcred, "test", &key,
|
|
Packit Service |
4684c1 |
GNUTLS_PSK_KEY_HEX));
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
/* Initialize TLS session
|
|
Packit Service |
4684c1 |
*/
|
|
Packit Service |
4684c1 |
CHECK(gnutls_init(&session, GNUTLS_CLIENT));
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
ret =
|
|
Packit Service |
4684c1 |
gnutls_set_default_priority_append(session,
|
|
Packit Service |
4684c1 |
"-KX-ALL:+ECDHE-PSK:+DHE-PSK:+PSK",
|
|
Packit Service |
4684c1 |
&err, 0);
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
/* Alternative for pre-3.6.3 versions:
|
|
Packit Service |
4684c1 |
* gnutls_priority_set_direct(session, "NORMAL:+ECDHE-PSK:+DHE-PSK:+PSK", &err)
|
|
Packit Service |
4684c1 |
*/
|
|
Packit Service |
4684c1 |
if (ret < 0) {
|
|
Packit Service |
4684c1 |
if (ret == GNUTLS_E_INVALID_REQUEST) {
|
|
Packit Service |
4684c1 |
fprintf(stderr, "Syntax error at: %s\n", err);
|
|
Packit Service |
4684c1 |
}
|
|
Packit Service |
4684c1 |
exit(1);
|
|
Packit Service |
4684c1 |
}
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
/* put the x509 credentials to the current session
|
|
Packit Service |
4684c1 |
*/
|
|
Packit Service |
4684c1 |
CHECK(gnutls_credentials_set(session, GNUTLS_CRD_PSK, pskcred));
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
/* connect to the peer
|
|
Packit Service |
4684c1 |
*/
|
|
Packit Service |
4684c1 |
sd = tcp_connect();
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
gnutls_transport_set_int(session, sd);
|
|
Packit Service |
4684c1 |
gnutls_handshake_set_timeout(session,
|
|
Packit Service |
4684c1 |
GNUTLS_DEFAULT_HANDSHAKE_TIMEOUT);
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
/* Perform the TLS handshake
|
|
Packit Service |
4684c1 |
*/
|
|
Packit Service |
4684c1 |
do {
|
|
Packit Service |
4684c1 |
ret = gnutls_handshake(session);
|
|
Packit Service |
4684c1 |
}
|
|
Packit Service |
4684c1 |
while (ret < 0 && gnutls_error_is_fatal(ret) == 0);
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
if (ret < 0) {
|
|
Packit Service |
4684c1 |
fprintf(stderr, "*** Handshake failed\n");
|
|
Packit Service |
4684c1 |
gnutls_perror(ret);
|
|
Packit Service |
4684c1 |
goto end;
|
|
Packit Service |
4684c1 |
} else {
|
|
Packit Service |
4684c1 |
char *desc;
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
desc = gnutls_session_get_desc(session);
|
|
Packit Service |
4684c1 |
printf("- Session info: %s\n", desc);
|
|
Packit Service |
4684c1 |
gnutls_free(desc);
|
|
Packit Service |
4684c1 |
}
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
LOOP_CHECK(ret, gnutls_record_send(session, MSG, strlen(MSG)));
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
LOOP_CHECK(ret, gnutls_record_recv(session, buffer, MAX_BUF));
|
|
Packit Service |
4684c1 |
if (ret == 0) {
|
|
Packit Service |
4684c1 |
printf("- Peer has closed the TLS connection\n");
|
|
Packit Service |
4684c1 |
goto end;
|
|
Packit Service |
4684c1 |
} else if (ret < 0 && gnutls_error_is_fatal(ret) == 0) {
|
|
Packit Service |
4684c1 |
fprintf(stderr, "*** Warning: %s\n", gnutls_strerror(ret));
|
|
Packit Service |
4684c1 |
} else if (ret < 0) {
|
|
Packit Service |
4684c1 |
fprintf(stderr, "*** Error: %s\n", gnutls_strerror(ret));
|
|
Packit Service |
4684c1 |
goto end;
|
|
Packit Service |
4684c1 |
}
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
if (ret > 0) {
|
|
Packit Service |
4684c1 |
printf("- Received %d bytes: ", ret);
|
|
Packit Service |
4684c1 |
for (ii = 0; ii < ret; ii++) {
|
|
Packit Service |
4684c1 |
fputc(buffer[ii], stdout);
|
|
Packit Service |
4684c1 |
}
|
|
Packit Service |
4684c1 |
fputs("\n", stdout);
|
|
Packit Service |
4684c1 |
}
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
CHECK(gnutls_bye(session, GNUTLS_SHUT_RDWR));
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
end:
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
tcp_close(sd);
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
gnutls_deinit(session);
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
gnutls_psk_free_client_credentials(pskcred);
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
gnutls_global_deinit();
|
|
Packit Service |
4684c1 |
|
|
Packit Service |
4684c1 |
return 0;
|
|
Packit Service |
4684c1 |
}
|