projects/pwm: io.c annotate

annotate io.c @ 25:616385fa1fd9

Build common functions as a library that can be reused

author	Guido Berhoerster <guido+pwm@berhoerster.name>
date	Wed, 20 Sep 2017 23:57:51 +0200
parents	ec01c579024a
children	131c35592054

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
22 ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	50 void
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	51 io_trim_nl(char *s)
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	52 {
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	53 size_t len;
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	54
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	55 len = strlen(s);
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	56 if ((len > 0) && (s[len - 1] == '\n')) {
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	57 s[len - 1] = '\0';
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	58 }
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	59 }
ec01c579024a Add fully interactive mode Guido Berhoerster <guido+pwm@berhoerster.name> parents: 19 diff changeset	60
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	61 int
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	62 io_gl_complete_nothing(WordCompletion cpl, void data, const char *line,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	63 int word_end)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	64 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	65 return (0);
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
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	68 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	69 io_get_char(const char prompt, int cp)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	70 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	71 enum io_status retval = IO_OK;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	72 GetLine *gl = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	73
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	74 gl = new_GetLine(16, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	75 if (gl== NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	76 err(1, "new_GetLine");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	77 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	78 gl_catch_blocked(gl);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	79
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	80 /* prompt with echo off */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	81 gl_echo_mode(gl, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	82 if ((*cp = gl_query_char(gl, prompt, '\0')) == EOF) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	83 switch (gl_return_status(gl)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	84 case GLR_SIGNAL:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	85 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	86 break;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	87 case GLR_ERROR:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	88 errx(1, "gl_get_line: %s",
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	89 gl_error_message(gl, NULL, 0));
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	90 default:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	91 errx(1, "unknown error in gl_get_line");
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
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	95 /* erase prompt */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	96 if (io_printf("\r%*s\r", (int)strlen(prompt), "") == IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	97 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	98 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	99
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	100 del_GetLine(gl);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	101
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	102 return (retval);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	103 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	104
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	105 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	106 io_get_line(GetLine gl, const char prompt, int with_history,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	107 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	108 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	109 enum io_status retval = IO_OK;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	110 GetLine *gl_private = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	111 GlHistoryState state;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	112 char *line;
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 if (gl == NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	115 gl = gl_private = new_GetLine(buf_size - 1, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	116 if (gl_private == NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	117 err(1, "new_GetLine");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	118 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	119 gl_catch_blocked(gl_private);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	120 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	121
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	122 gl_state_of_history(gl, &state);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	123 gl_toggle_history(gl, with_history);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	124
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	125 line = gl_get_line(gl, prompt, start_line, start_pos);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	126 if (line == NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	127 switch (gl_return_status(gl)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	128 case GLR_BLOCKED:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	129 break;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	130 case GLR_SIGNAL:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	131 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	132 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	133 case GLR_EOF:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	134 retval = IO_EOF;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	135 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	136 case GLR_ERROR:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	137 errx(1, "gl_get_line: %s",
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	138 gl_error_message(gl, NULL, 0));
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	139 default:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	140 errx(1, "unknown error in gl_get_line");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	141 }
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
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	144 if (snprintf(buf, buf_size, "%s", line) >= (int)buf_size) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	145 retval = IO_TRUNCATED;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	146 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	147
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	148 out:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	149 if (gl != NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	150 gl_toggle_history(gl, state.enabled);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	151 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	152 del_GetLine(gl_private);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	153
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	154 return (retval);
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
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	157 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	158 io_get_password(const char prompt, const char confirm_prompt,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	159 size_t buf_size, char *buf)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	160 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	161 enum io_status retval = IO_OK;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	162 GetLine *gl = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	163 size_t len;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	164 char *password_buf = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	165 char *confirm_buf = NULL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	166
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	167 gl = new_GetLine(buf_size - 1, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	168 if (gl == NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	169 err(1, "new_GetLine");
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 /* disable default filename completion */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	172 gl_customize_completion(gl, NULL, io_gl_complete_nothing);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	173 gl_echo_mode(gl, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	174
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	175 password_buf = xmalloc(buf_size);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	176
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	177 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	178 IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	179 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	180 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	181 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	182 len = strlen(password_buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	183 /* strip trailing newline */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	184 if ((len > 0) && (password_buf[len - 1] == '\n')) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	185 password_buf[--len] = '\0';
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 if (len == 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	188 retval = IO_PASSWORD_EMPTY;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	189 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	190 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	191
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	192 if (confirm_prompt != NULL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	193 if (io_printf("\n") == IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	194 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	195 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	196 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	197
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	198 /* confirm new password */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	199 confirm_buf = xmalloc(buf_size);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	200 if (io_get_line(gl, confirm_prompt, 0, NULL, 0,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	201 buf_size, confirm_buf) == IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	202 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	203 goto out;
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 len = strlen(confirm_buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	206 /* strip trailing newline */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	207 if ((len > 0) && (confirm_buf[len - 1] == '\n')) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	208 confirm_buf[--len] = '\0';
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	209 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	210 if (strcmp(password_buf, confirm_buf) != 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	211 retval = IO_PASSWORD_MISMATCH;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	212 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	213 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	214 }
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 strcpy(buf, password_buf);
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 out:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	219 if (io_printf("\n") == IO_SIGNAL) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	220 retval = IO_SIGNAL;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	221 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	222 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	223 free(password_buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	224 free(confirm_buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	225 del_GetLine(gl);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	226
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	227 return (retval);
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
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	230 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	231 io_dputs(int fd, const char *s)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	232 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	233 struct sigaction action;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	234 struct sigaction oaction;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	235 int signal_no = 0;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	236 const char *p = s;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	237 size_t remaining;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	238 ssize_t n;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	239
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	240 /* install signal handlers */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	241 action.sa_handler = signal_handler;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	242 action.sa_flags = 0;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	243 sigemptyset(&action.sa_mask);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	244 sigaddset(&action.sa_mask, SIGINT);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	245 sigaddset(&action.sa_mask, SIGTERM);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	246 sigaddset(&action.sa_mask, SIGHUP);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	247 sigaddset(&action.sa_mask, SIGQUIT);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	248 if ((sigaction(SIGINT, &action, &oaction) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	249 (sigaction(SIGTERM, &action, &oaction) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	250 (sigaction(SIGHUP, &action, &oaction) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	251 (sigaction(SIGQUIT, &action, &oaction) != 0)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	252 err(1, "sigaction");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	253 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	254
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	255 if ((signal_no = sigsetjmp(signal_env, 1)) != 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	256 /* signal received, signal mask has been restored */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	257 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	258 }
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 remaining = strlen(s);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	261 while (remaining > 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	262 pwm_unblock_signals();
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	263 n = write(fd, p, remaining);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	264 if ((n < (int)remaining) && (errno != EINTR)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	265 err(1, "write");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	266 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	267 pwm_block_signals();
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	268 remaining -= MAX(n, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	269 p += MAX(n, 0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	270 }
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 out:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	273 /* restore signal handlers */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	274 if ((sigaction(SIGINT, &oaction, NULL) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	275 (sigaction(SIGTERM, &oaction, NULL) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	276 (sigaction(SIGHUP, &oaction, NULL) != 0) \|\|
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	277 (sigaction(SIGQUIT, &oaction, NULL) != 0)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	278 err(1, "sigaction");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	279 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	280
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	281 return ((signal_no == 0) ? IO_OK : IO_SIGNAL);
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
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	284 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	285 io_vdprintf(int fd, const char *fmt, va_list args)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	286 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	287 enum io_status retval;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	288 char *buf;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	289
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	290 xvasprintf(&buf, fmt, args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	291 retval = io_dputs(fd, buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	292 free(buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	293
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	294 return (retval);
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
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	297 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	298 io_dprintf(int fd, const char *fmt, ...)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	299 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	300 enum io_status retval;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	301 va_list args;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	302
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	303 va_start(args, fmt);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	304 retval = io_vdprintf(fd, fmt, args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	305 va_end(args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	306
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	307 return (retval);
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
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	310 enum io_status
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	311 io_printf(const char *fmt, ...)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	312 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	313 enum io_status retval;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	314 va_list args;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	315
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	316 va_start(args, fmt);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	317 retval = io_vdprintf(STDOUT_FILENO, fmt, args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	318 va_end(args);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	319
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	320 return (retval);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	321 }

Mercurial > projects > pwm

annotate io.c @ 25:616385fa1fd9