projects/pwm: io.c annotate

annotate io.c @ 21:ee4d36c85287

Make EOF in interactive mode equivalent to the q command

author	Guido Berhoerster <guido+pwm@berhoerster.name>
date	Wed, 06 Sep 2017 16:41:58 +0200
parents	5c6155c8e9b6
children	ec01c579024a

rev	line source
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	1 /*
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	2 * Copyright (C) 2017 Guido Berhoerster <guido+pwm@berhoerster.name>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	3 *
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	4 * Permission is hereby granted, free of charge, to any person obtaining
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	5 * a copy of this software and associated documentation files (the
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	6 * "Software"), to deal in the Software without restriction, including
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	7 * without limitation the rights to use, copy, modify, merge, publish,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	8 * distribute, sublicense, and/or sell copies of the Software, and to
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	9 * permit persons to whom the Software is furnished to do so, subject to
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	10 * the following conditions:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	11 *
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	12 * The above copyright notice and this permission notice shall be included
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	13 * in all copies or substantial portions of the Software.
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	14 *
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	16 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	17 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	18 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	19 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	20 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	21 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	22 */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	23
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	24 #include "compat.h"
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	25
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	26 #ifdef HAVE_ERR_H
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	27 #include <err.h>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	28 #endif /* HAVE_ERR_H */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	29 #include <errno.h>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	30 #include <setjmp.h>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	31 #include <signal.h>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	32 #include <stdarg.h>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	33 #include <stdio.h>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	34 #include <stdio.h>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	35 #include <string.h>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	36 #include <unistd.h>
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	37
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	38 #include "io.h"
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	39 #include "util.h"
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	40 #include "pwm.h"
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	41
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	42 static sigjmp_buf signal_env;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	43
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	44 static void
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	45 signal_handler(int signal_no)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	46 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	47 siglongjmp(signal_env, signal_no);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	48 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	49
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	50 int
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	51 io_gl_complete_nothing(WordCompletion cpl, void data, const char *line,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	52 int word_end)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	53 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	54 return (0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	55 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	56
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	57 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	58 io_get_char(const char prompt, int cp)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	59 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	60 enum io_status retval = IO_OK;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	61 GetLine *gl = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	62
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	63 gl = new_GetLine(16, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	64 if (gl== NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	65 err(1, "new_GetLine");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	66 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	67 gl_catch_blocked(gl);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	68
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	69 /* prompt with echo off */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	70 gl_echo_mode(gl, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	71 if ((*cp = gl_query_char(gl, prompt, '\0')) == EOF) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	72 switch (gl_return_status(gl)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	73 case GLR_SIGNAL:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	74 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	75 break;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	76 case GLR_ERROR:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	77 errx(1, "gl_get_line: %s",
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	78 gl_error_message(gl, NULL, 0));
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	79 default:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	80 errx(1, "unknown error in gl_get_line");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	81 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	82 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	83
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	84 /* erase prompt */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	85 if (io_printf("\r%*s\r", (int)strlen(prompt), "") == IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	86 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	87 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	88
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	89 del_GetLine(gl);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	90
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	91 return (retval);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	92 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	93
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	94 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	95 io_get_line(GetLine gl, const char prompt, int with_history,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	96 const char start_line, int start_pos, size_t buf_size, char buf)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	97 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	98 enum io_status retval = IO_OK;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	99 GetLine *gl_private = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	100 GlHistoryState state;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	101 char *line;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	102
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	103 if (gl == NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	104 gl = gl_private = new_GetLine(buf_size - 1, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	105 if (gl_private == NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	106 err(1, "new_GetLine");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	107 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	108 gl_catch_blocked(gl_private);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	109 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	110
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	111 gl_state_of_history(gl, &state);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	112 gl_toggle_history(gl, with_history);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	113
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	114 line = gl_get_line(gl, prompt, start_line, start_pos);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	115 if (line == NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	116 switch (gl_return_status(gl)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	117 case GLR_BLOCKED:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	118 break;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	119 case GLR_SIGNAL:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	120 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	121 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	122 case GLR_EOF:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	123 retval = IO_EOF;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	124 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	125 case GLR_ERROR:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	126 errx(1, "gl_get_line: %s",
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	127 gl_error_message(gl, NULL, 0));
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	128 default:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	129 errx(1, "unknown error in gl_get_line");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	130 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	131 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	132
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	133 if (snprintf(buf, buf_size, "%s", line) >= (int)buf_size) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	134 retval = IO_TRUNCATED;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	135 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	136
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	137 out:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	138 if (gl != NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	139 gl_toggle_history(gl, state.enabled);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	140 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	141 del_GetLine(gl_private);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	142
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	143 return (retval);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	144 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	145
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	146 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	147 io_get_password(const char prompt, const char confirm_prompt,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	148 size_t buf_size, char *buf)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	149 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	150 enum io_status retval = IO_OK;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	151 GetLine *gl = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	152 size_t len;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	153 char *password_buf = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	154 char *confirm_buf = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	155
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	156 gl = new_GetLine(buf_size - 1, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	157 if (gl == NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	158 err(1, "new_GetLine");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	159 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	160 /* disable default filename completion */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	161 gl_customize_completion(gl, NULL, io_gl_complete_nothing);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	162 gl_echo_mode(gl, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	163
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	164 password_buf = xmalloc(buf_size);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	165
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	166 if (io_get_line(gl, prompt, 0, NULL, 0, buf_size, password_buf) ==
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	167 IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	168 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	169 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	170 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	171 len = strlen(password_buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	172 /* strip trailing newline */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	173 if ((len > 0) && (password_buf[len - 1] == '\n')) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	174 password_buf[--len] = '\0';
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	175 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	176 if (len == 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	177 retval = IO_PASSWORD_EMPTY;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	178 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	179 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	180
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	181 if (confirm_prompt != NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	182 if (io_printf("\n") == IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	183 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	184 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	185 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	186
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	187 /* confirm new password */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	188 confirm_buf = xmalloc(buf_size);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	189 if (io_get_line(gl, confirm_prompt, 0, NULL, 0,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	190 buf_size, confirm_buf) == IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	191 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	192 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	193 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	194 len = strlen(confirm_buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	195 /* strip trailing newline */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	196 if ((len > 0) && (confirm_buf[len - 1] == '\n')) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	197 confirm_buf[--len] = '\0';
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	198 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	199 if (strcmp(password_buf, confirm_buf) != 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	200 retval = IO_PASSWORD_MISMATCH;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	201 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	202 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	203 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	204
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	205 strcpy(buf, password_buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	206
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	207 out:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	208 if (io_printf("\n") == IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	209 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	210 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	211 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	212 free(password_buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	213 free(confirm_buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	214 del_GetLine(gl);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	215
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	216 return (retval);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	217 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	218
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	219 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	220 io_dputs(int fd, const char *s)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	221 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	222 struct sigaction action;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	223 struct sigaction oaction;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	224 int signal_no = 0;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	225 const char *p = s;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	226 size_t remaining;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	227 ssize_t n;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	228
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	229 /* install signal handlers */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	230 action.sa_handler = signal_handler;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	231 action.sa_flags = 0;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	232 sigemptyset(&action.sa_mask);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	233 sigaddset(&action.sa_mask, SIGINT);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	234 sigaddset(&action.sa_mask, SIGTERM);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	235 sigaddset(&action.sa_mask, SIGHUP);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	236 sigaddset(&action.sa_mask, SIGQUIT);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	237 if ((sigaction(SIGINT, &action, &oaction) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	238 (sigaction(SIGTERM, &action, &oaction) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	239 (sigaction(SIGHUP, &action, &oaction) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	240 (sigaction(SIGQUIT, &action, &oaction) != 0)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	241 err(1, "sigaction");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	242 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	243
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	244 if ((signal_no = sigsetjmp(signal_env, 1)) != 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	245 /* signal received, signal mask has been restored */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	246 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	247 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	248
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	249 remaining = strlen(s);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	250 while (remaining > 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	251 pwm_unblock_signals();
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	252 n = write(fd, p, remaining);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	253 if ((n < (int)remaining) && (errno != EINTR)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	254 err(1, "write");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	255 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	256 pwm_block_signals();
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	257 remaining -= MAX(n, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	258 p += MAX(n, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	259 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	260
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	261 out:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	262 /* restore signal handlers */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	263 if ((sigaction(SIGINT, &oaction, NULL) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	264 (sigaction(SIGTERM, &oaction, NULL) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	265 (sigaction(SIGHUP, &oaction, NULL) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	266 (sigaction(SIGQUIT, &oaction, NULL) != 0)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	267 err(1, "sigaction");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	268 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	269
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	270 return ((signal_no == 0) ? IO_OK : IO_SIGNAL);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	271 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	272
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	273 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	274 io_vdprintf(int fd, const char *fmt, va_list args)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	275 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	276 enum io_status retval;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	277 char *buf;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	278
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	279 xvasprintf(&buf, fmt, args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	280 retval = io_dputs(fd, buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	281 free(buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	282
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	283 return (retval);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	284 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	285
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	286 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	287 io_dprintf(int fd, const char *fmt, ...)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	288 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	289 enum io_status retval;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	290 va_list args;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	291
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	292 va_start(args, fmt);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	293 retval = io_vdprintf(fd, fmt, args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	294 va_end(args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	295
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	296 return (retval);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	297 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	298
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	299 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	300 io_printf(const char *fmt, ...)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	301 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	302 enum io_status retval;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	303 va_list args;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	304
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	305 va_start(args, fmt);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	306 retval = io_vdprintf(STDOUT_FILENO, fmt, args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	307 va_end(args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	308
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	309 return (retval);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	310 }

Mercurial > projects > pwm

annotate io.c @ 21:ee4d36c85287