1 /*
2  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include "compat.h"
18 
19 #include <sys/types.h>
20 #include <sys/socket.h>
21 
22 #include <netdb.h>
23 #include <limits.h>
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <syslog.h>
28 #include <tls.h>
29 #include <unistd.h>
30 
31 #include <readline/readline.h>
32 #include <readline/history.h>
33 
34 #include "9pclib.h"
35 #include "kamid.h"
36 #include "utils.h"
37 #include "log.h"
38 
39 /* flags */
40 int		 tls;
41 const char	*crtpath;
42 const char	*keypath;
43 
44 /* state */
45 struct tls_config	*tlsconf;
46 struct tls		*ctx;
47 int			 sock;
48 
49 #define PWDFID		0
50 
51 static void ATTR_DEAD
52 usage(int ret)
53 {
54 	fprintf(stderr, "usage: %s [-c] host[:port] [path]\n",
55 	    getprogname());
56 	fprintf(stderr, PACKAGE_NAME " suite version " PACKAGE VERSION "\n");
57 	exit(ret);
58 }
59 
60 static void
61 do_version(void)
62 {
63 	tversion(VERSION9P, MSIZE9P);
64 	/* TODO: get reply */
65 }
66 
67 static void
68 do_attach(const char *path)
69 {
70 	if (path == NULL)
71 		path = "/";
72 
73 	/* TODO: do attach */
74 }
75 
76 static void
77 do_connect(const char *connspec, const char *path)
78 {
79 	int handshake;
80 	char *host, *colon;
81 	const char *port;
82 
83 	host = xstrdup(connspec);
84 	if ((colon = strchr(host, ':')) != NULL) {
85 		*colon = '\0';
86 		port = ++colon;
87 	} else
88 		port = "1337";
89 
90 	if (!tls)
91 		fatalx("non-tls mode is not supported");
92 
93 	if ((tlsconf = tls_config_new()) == NULL)
94 		fatalx("tls_config_new");
95 	tls_config_insecure_noverifycert(tlsconf);
96 	tls_config_insecure_noverifyname(tlsconf);
97 	if (tls_config_set_keypair_file(tlsconf, crtpath, keypath) == -1)
98 		fatalx("can't load certs (%s, %s)", crtpath, keypath);
99 
100 	if ((ctx = tls_client()) == NULL)
101 		fatal("tls_client");
102 	if (tls_configure(ctx, tlsconf) == -1)
103 		fatalx("tls_configure: %s", tls_error(ctx));
104 
105 	printf("connecting to %s:%s...", host, port);
106 	fflush(stdout);
107 
108 	if (tls_connect(ctx, host, port) == -1)
109 		fatalx("can't connect to %s:%s: %s", host, port,
110 		    tls_error(ctx));
111 
112 	for (handshake = 0; !handshake;) {
113 		switch (tls_handshake(ctx)) {
114 		case -1:
115 			fatalx("tls_handshake: %s", tls_error(ctx));
116 		case 0:
117 			handshake = 1;
118 			break;
119 		}
120 	}
121 
122 	printf(" done!\n");
123 
124 	do_version();
125 	do_attach(path);
126 
127 	free(host);
128 }
129 
130 int
131 main(int argc, char **argv)
132 {
133 	int	ch;
134 
135 	log_init(1, LOG_DAEMON);
136 	log_setverbose(1);
137 	log_procinit(getprogname());
138 
139 	while ((ch = getopt(argc, argv, "C:cK:")) != -1) {
140 		switch (ch) {
141 		case 'C':
142 			crtpath = optarg;
143 			break;
144 		case 'c':
145 			tls = 1;
146 			break;
147 		case 'K':
148 			keypath = optarg;
149 			break;
150 		default:
151 			usage(1);
152 		}
153 	}
154 	argc -= optind;
155 	argv += optind;
156 
157 	if (argc == 0)
158 		usage(1);
159 
160 	if ((evb = evbuffer_new()) == NULL)
161 		fatal("evbuffer_new");
162 
163 	do_connect(argv[0], argv[1]);
164 
165 	for (;;) {
166 		char *line;
167 
168 		if ((line = readline("ftp> ")) == NULL)
169 			break;
170 		/* XXX: trim spaces */
171 		if (*line == '\0')
172 			continue;
173 		add_history(line);
174 	}
175 
176 	printf("\n");
177 }