xref: /freebsd/contrib/ntp/libntp/ntp_lineedit.c (revision 39ee7a7a6bdd1557b1c3532abf60d139798ac88b)
1 /*
2  * ntp_lineedit.c - generic interface to various line editing libs
3  */
4 #ifdef HAVE_CONFIG_H
5 # include <config.h>
6 #endif
7 
8 #include <errno.h>
9 #include <string.h>
10 #include <stdlib.h>
11 #include <stdio.h>
12 
13 #if defined(HAVE_READLINE_HISTORY) &&		\
14     (!defined(HAVE_READLINE_HISTORY_H) ||	\
15      !defined(HAVE_READLINE_READLINE_H))
16 # undef HAVE_READLINE_HISTORY
17 #endif
18 #if defined(HAVE_READLINE_HISTORY)
19 # include <readline/readline.h>
20 # include <readline/history.h>
21 # define LE_READLINE
22 #elif defined(HAVE_HISTEDIT_H)
23 # include <histedit.h>
24 # define LE_EDITLINE
25 #else
26 # define LE_NONE
27 #endif
28 
29 #include "ntp.h"
30 #include "ntp_stdlib.h"
31 #include "ntp_lineedit.h"
32 
33 #define MAXEDITLINE	512
34 
35 /*
36  * external references
37  */
38 
39 extern char const *	progname;
40 
41 /*
42  * globals, private prototypes
43  */
44 
45 static int	ntp_readline_initted;
46 static char *	lineedit_prompt;
47 
48 
49 #ifdef LE_EDITLINE
50 # ifndef H_SETSIZE
51 #  define H_SETSIZE H_EVENT
52 # endif
53 static EditLine *	ntp_el;
54 static History *	ntp_hist;
55 static HistEvent	hev;
56 
57 char *	ntp_prompt_callback(EditLine *);
58 #endif	/* LE_EDITLINE */
59 
60 
61 /*
62  * ntp_readline_init - setup, set or reset prompt string
63  */
64 int
65 ntp_readline_init(
66 	const char *	prompt
67 	)
68 {
69 	int	success;
70 
71 	success = 1;
72 
73 	if (prompt) {
74 		if (lineedit_prompt)
75 			free(lineedit_prompt);
76 		lineedit_prompt = estrdup(prompt);
77 	}
78 
79 #ifdef LE_EDITLINE
80 	if (NULL == ntp_el) {
81 
82 # if 4 == EL_INIT_ARGS
83 		ntp_el = el_init(progname, stdin, stdout, stderr);
84 # else
85 		ntp_el = el_init(progname, stdin, stdout);
86 # endif
87 		if (ntp_el) {
88 
89 			el_set(ntp_el, EL_PROMPT, ntp_prompt_callback);
90 			el_set(ntp_el, EL_EDITOR, "emacs");
91 
92 			ntp_hist = history_init();
93 
94 			if (NULL == ntp_hist) {
95 
96 				mfprintf(stderr, "history_init(): %m\n");
97 				fflush(stderr);
98 
99 				el_end(ntp_el);
100 				ntp_el = NULL;
101 
102 				success = 0;
103 
104 			} else {
105 				ZERO(hev);
106 #ifdef H_SETSIZE
107 				history(ntp_hist, &hev, H_SETSIZE, 128);
108 #endif
109 				el_set(ntp_el, EL_HIST, history,
110 				       ntp_hist);
111 				/* use any .editrc */
112 				el_source(ntp_el, NULL);
113 			}
114 		} else
115 			success = 0;
116 	}
117 #endif	/* LE_EDITLINE */
118 
119 	ntp_readline_initted = success;
120 
121 	return success;
122 }
123 
124 
125 /*
126  * ntp_readline_uninit - release resources
127  */
128 void
129 ntp_readline_uninit(
130 	void
131 	)
132 {
133 #ifdef LE_EDITLINE
134 	if (ntp_el) {
135 		el_end(ntp_el);
136 		ntp_el = NULL;
137 
138 		history_end(ntp_hist);
139 		ntp_hist = NULL;
140 	}
141 #endif	/* LE_EDITLINE */
142 
143 	if (lineedit_prompt) {
144 		free(lineedit_prompt);
145 		lineedit_prompt = NULL;
146 	}
147 
148 	ntp_readline_initted = 0;
149 }
150 
151 
152 /*
153  * ntp_readline - read a line with the line editor available
154  *
155  * The string returned must be released with free()
156  */
157 
158 char *
159 ntp_readline(
160 	int *	pcount
161 	)
162 {
163 	char *		line;
164 #ifdef LE_NONE
165 	char		line_buf[MAXEDITLINE];
166 #endif
167 #ifdef LE_EDITLINE
168 	const char *	cline;
169 #endif
170 
171 	if (!ntp_readline_initted)
172 		return NULL;
173 
174 	*pcount = 0;
175 
176 #ifdef LE_READLINE
177 	line = readline(lineedit_prompt ? lineedit_prompt : "");
178 	if (NULL != line) {
179 		if (*line) {
180 			add_history(line);
181 		}
182 		*pcount = strlen(line);
183 	}
184 #endif	/* LE_READLINE */
185 
186 #ifdef LE_EDITLINE
187 	cline = el_gets(ntp_el, pcount);
188 
189 	if (NULL != cline) {
190 		history(ntp_hist, &hev, H_ENTER, cline);
191 		line = estrdup(cline);
192 	} else if (*pcount == -1) {
193 		line = NULL;
194 	} else {
195 		line = estrdup("");
196 	}
197 #endif	/* LE_EDITLINE */
198 
199 #ifdef LE_NONE
200 					/* stone hammers */
201 	if (lineedit_prompt) {
202 # ifdef VMS
203 			/*
204 			 * work around problem mixing
205 			 * stdout & stderr
206 			 */
207 			fputs("", stdout);
208 # endif	/* VMS */
209 
210 		fputs(lineedit_prompt, stderr);
211 		fflush(stderr);
212 	}
213 
214 	line = fgets(line_buf, sizeof(line_buf), stdin);
215 	if (NULL != line && *line) {
216 		*pcount = strlen(line);
217 		line = estrdup(line);
218 	} else
219 		line = NULL;
220 
221 #endif	/* LE_NONE */
222 
223 
224 	if (!line)			/* EOF */
225 		fputs("\n", stderr);
226 
227 	return line;
228 }
229 
230 
231 #ifdef LE_EDITLINE
232 /*
233  * ntp_prompt_callback - return prompt string to el_gets()
234  */
235 char *
236 ntp_prompt_callback(
237 	EditLine *el
238 	)
239 {
240 	UNUSED_ARG(el);
241 
242 	return lineedit_prompt;
243 }
244 #endif /* LE_EDITLINE */
245 
246