main.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "client.h"
#include "header.h"
#include "url.h"

#define MAX_BUFFER_LEN	2048
#define MAX_REDIRECTS	5


int
main(int argc, char *argv[])
{
	SSL_library_init();
	OpenSSL_add_all_algorithms();
	SSL_load_error_strings();

	int sockfd;
	int n_bytes = 0;
	int n_redirects = 0;

	SSL *ssl;
	SSL_CTX *ctx;

	struct url *url = create_url();
	struct gemini_header header;

	char req[MAX_BUFFER_LEN];
	char res[MAX_BUFFER_LEN];

	if (argc < 2) {
		fprintf(stderr, "usage: %s [domain]\n", argv[0]);
		exit(EXIT_FAILURE);
	}

	parse_url(argv[1], url);

request:
	sockfd = connect_to_domain(url->authority);
	secure_connection(sockfd, &ctx, &ssl, url->authority);

	memset(req, '\0', sizeof(req));
	memset(res, '\0', sizeof(res));

	sprintf(req, "gemini://%s%s%s\r\n",
		url->authority, url->path, url->query ? url->query : "");
	SSL_write(ssl, req, strlen(req));

	SSL_read(ssl, res, sizeof(res));
	parse_gemini_header(res, &header);

	switch (header.status / 10) {
	case GEMINI_INPUT:
		puts(header.meta);

		char input[MAX_BUFFER_LEN];
		fgets(input, MAX_BUFFER_LEN, stdin);

		append_query(input, url);

		SSL_CTX_free(ctx);
		SSL_shutdown(ssl);
		SSL_free(ssl);
		close(sockfd);

		goto request;
	case GEMINI_REDIRECT:
		if (++n_redirects > MAX_REDIRECTS) {
			fprintf(stderr, "Number of redirects exceeded maximum.\n");
			exit(EXIT_FAILURE);
		}

		parse_url(header.meta, url);

		SSL_CTX_free(ctx);
		SSL_shutdown(ssl);
		SSL_free(ssl);
		close(sockfd);

		goto request;
	case GEMINI_SUCCESS:
		while ((n_bytes = SSL_read(ssl, res, MAX_BUFFER_LEN)) != 0) {
			fwrite(&res, 1, n_bytes, stdout);
			if (ferror(stdout)) {
				fprintf(stderr, "fwrite() failed.\n");
				exit(EXIT_FAILURE);
			}
		}
		break;
	case GEMINI_TEMPORARY_FAILURE:
	case GEMINI_PERMANENT_FAILURE:
		fprintf(stderr, "%s", header.meta);
		break;
	}

	SSL_CTX_free(ctx);
	SSL_shutdown(ssl);
	SSL_free(ssl);

	free_url(url);
	close(sockfd);

	exit(EXIT_SUCCESS);
}