From 1211a036baeed9a652346fc3fea6896b277d344a Mon Sep 17 00:00:00 2001 From: Peter Elliott Date: Sat, 25 Jul 2020 17:38:55 -0600 Subject: [PATCH] LibCore: add get_password(). A serenity-style getpass that is thread-safe --- Libraries/LibCore/CMakeLists.txt | 1 + Libraries/LibCore/GetPassword.cpp | 63 +++++++++++++++++++++++++++++++ Libraries/LibCore/GetPassword.h | 36 ++++++++++++++++++ 3 files changed, 100 insertions(+) create mode 100644 Libraries/LibCore/GetPassword.cpp create mode 100644 Libraries/LibCore/GetPassword.h diff --git a/Libraries/LibCore/CMakeLists.txt b/Libraries/LibCore/CMakeLists.txt index 3b3662d4052..6d288e85627 100644 --- a/Libraries/LibCore/CMakeLists.txt +++ b/Libraries/LibCore/CMakeLists.txt @@ -7,6 +7,7 @@ set(SOURCES Event.cpp EventLoop.cpp File.cpp + GetPassword.cpp Gzip.cpp IODevice.cpp LocalServer.cpp diff --git a/Libraries/LibCore/GetPassword.cpp b/Libraries/LibCore/GetPassword.cpp new file mode 100644 index 00000000000..73d123bd0a9 --- /dev/null +++ b/Libraries/LibCore/GetPassword.cpp @@ -0,0 +1,63 @@ +/* + * Copyright (c) 2020, Peter Elliott + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER + * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include +#include +#include +#include + +namespace Core { + +Result get_password(const StringView& prompt) +{ + fwrite(prompt.characters_without_null_termination(), sizeof(char), prompt.length(), stdout); + fflush(stdout); + + struct termios original; + tcgetattr(STDIN_FILENO, &original); + + struct termios no_echo = original; + no_echo.c_lflag &= ~ECHO; + if (tcsetattr(STDIN_FILENO, TCSAFLUSH, &no_echo) < 0) { + return errno; + } + + char* password = nullptr; + size_t n = 0; + + int ret = getline(&password, &n, stdin); + tcsetattr(STDIN_FILENO, TCSAFLUSH, &original); + putchar('\n'); + if (ret < 0) { + return errno; + } + + String s(password); + free(password); + + return s; +} +} diff --git a/Libraries/LibCore/GetPassword.h b/Libraries/LibCore/GetPassword.h new file mode 100644 index 00000000000..b0993c56f3c --- /dev/null +++ b/Libraries/LibCore/GetPassword.h @@ -0,0 +1,36 @@ +/* + * Copyright (c) 2020, Peter Elliott + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER + * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#pragma once + +#include +#include + +namespace Core { + +Result get_password(const StringView& prompt = "Password: "); + +}