Added 16_keystore
This commit is contained in:
parent
123b2942e9
commit
85c9c3bea3
230
bash/bashrc/sources/16_keystore
Normal file
230
bash/bashrc/sources/16_keystore
Normal file
@ -0,0 +1,230 @@
|
||||
#!/bin/bash
|
||||
|
||||
_SQLITE=$( which sqlite3 2>&1 )
|
||||
if [ $? -ne 0 ]; then
|
||||
echo "sqlite3 is missing" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
echo "test" | openssl enc -e -aes-256-cbc -pbkdf2 -k test 2>/dev/null > /dev/null
|
||||
if [ $? -ne 0 ]; then
|
||||
echo "openssl does not exists nor support pbkdf2" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
if [ -z `which xxd` ]; then
|
||||
echo "xxd is required" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
_AUTH_DB=$RBASH_HOME/keystore.db
|
||||
_AUTH_SECRET=
|
||||
_KSTORE_DEF_PROP=${_KSTORE_DEF_PROP:-val}
|
||||
|
||||
if [ -f "$RBASH_HOME/keystore.secret" ]; then
|
||||
_AUTH_SECRET=$( cat "$RBASH_HOME/keystore.secret" )
|
||||
fi
|
||||
|
||||
function kstore {
|
||||
case $1 in
|
||||
add) shift; kstore-add "$@" ;;
|
||||
del) shift; kstore-del "$@" ;;
|
||||
get) shift; kstore-get "$@" ;;
|
||||
list) shift; kstore-list "$@" ;;
|
||||
query) shift; kstore-query "$@" ;;
|
||||
update) shift; kstore-update "$@" ;;
|
||||
search) shift; kstore-search $@ ;;
|
||||
secret) shift; kstore-secret $@ ;;
|
||||
*)
|
||||
__func_head "add key value [prop, default: $_KSTORE_DEF_PROP]"
|
||||
__func_help "update key value [prop, default: $_KSTORE_DEF_PROP]"
|
||||
__func_help "get key [prop, default: $_KSTORE_DEF_PROP]"
|
||||
__func_help "list"
|
||||
__func_help "del key"
|
||||
__func_help "search key"
|
||||
__func_help "query SQL"
|
||||
return 1
|
||||
;;
|
||||
esac
|
||||
return $?
|
||||
}
|
||||
|
||||
function kstore-secret {
|
||||
case "$1" in
|
||||
clear) shift; kstore-secret-clear "$@" ;;
|
||||
config) shift; kstore-secret-config "$@" ;;
|
||||
*)
|
||||
__func_head "clear"
|
||||
__func_help "config"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
function kstore-init {
|
||||
cat <<___SQL___ | $_SQLITE "$_AUTH_DB"
|
||||
CREATE TABLE IF NOT EXISTS store (
|
||||
key TEXT UNIQUE NOT NULL
|
||||
, prop TEXT NOT NULL
|
||||
, data BLOB
|
||||
, PRIMARY KEY( key ASC, prop ASC )
|
||||
);
|
||||
___SQL___
|
||||
kstore secret config
|
||||
}
|
||||
|
||||
function kstore-quote {
|
||||
echo -n "$1" | sed -e "s/'/''/g"
|
||||
}
|
||||
|
||||
function kstore-enc {
|
||||
if [ -z "$_AUTH_SECRET" ]; then
|
||||
echo "Secret key is not set yet" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
openssl enc -e -aes-256-cbc -pbkdf2 -k "$_AUTH_SECRET" | xxd -p | tr -d "\n"
|
||||
}
|
||||
|
||||
function kstore-dec {
|
||||
if [ -z "$_AUTH_SECRET" ]; then
|
||||
echo "Secret key is not set yet" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
openssl enc -d -aes-256-cbc -pbkdf2 -k "$_AUTH_SECRET"
|
||||
}
|
||||
|
||||
function kstore-secret-macos {
|
||||
local _A
|
||||
case $1 in
|
||||
get)
|
||||
_A=`security find-generic-password -a default -gs rbash-kstore 2>&1 | grep ^password | cut -c 11-`
|
||||
if [ $? -ne 0 ]; then
|
||||
return 1
|
||||
fi
|
||||
_AUTH_SECRET="${_A:1:-1}"
|
||||
;;
|
||||
set)
|
||||
security add-generic-password -a default -s rbash-kstore -w "$_AUTH_SECRET"
|
||||
;;
|
||||
*)
|
||||
echo "Unknown action: $1" >&2
|
||||
return 1
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
function kstore-secret-windows {
|
||||
# Not impelemented yet
|
||||
if [ "$1" == "get" ]; then
|
||||
return 1
|
||||
fi
|
||||
return 1
|
||||
}
|
||||
|
||||
function kstore-secret-config {
|
||||
local _CONFIRM
|
||||
if [ -z "$_AUTH_SECRET" ]; then
|
||||
|
||||
kstore-secret-windows get || kstore-secret-macos get
|
||||
if [ -n "$_AUTH_SECRET" ]; then
|
||||
return 0
|
||||
fi
|
||||
|
||||
read -sp "Enter passphrase: " _AUTH_SECRET
|
||||
echo
|
||||
if [ -z "$_AUTH_SECRET" ]; then
|
||||
echo "Passphrase cannot be empty" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
read -p "Save this password to OS's keystore? (y/n): " _CONFIRM
|
||||
if [ "$_CONFIRM" != "y" ]; then
|
||||
return 0
|
||||
fi
|
||||
|
||||
case $OSTYPE in
|
||||
darwin*) kstore-secret-macos "set" ;;
|
||||
esac
|
||||
fi
|
||||
}
|
||||
|
||||
function kstore-secret-clear {
|
||||
_AUTH_SECRET=
|
||||
case $OSTYPE in
|
||||
darwin*)
|
||||
security delete-generic-password -a default -s rbash-kstore 2>&1 > /dev/null
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
function kstore-update {
|
||||
kstore-init || return 1
|
||||
local _key _val _prop
|
||||
_key=`kstore-quote "$1"`
|
||||
_val=`echo -n "$2" | kstore-enc`
|
||||
_val=`kstore-quote "$_val"`
|
||||
_prop=`kstore-quote "${3:-$_KSTORE_DEF_PROP}"`
|
||||
_cond="key = '$_key' AND prop = '$_prop'"
|
||||
$_SQLITE "$_AUTH_DB" "UPDATE store SET data = '$_val' WHERE $_cond;"
|
||||
}
|
||||
|
||||
function kstore-add {
|
||||
kstore-init || return 1
|
||||
local _key _val _prop
|
||||
_key=`kstore-quote "$1"`
|
||||
_val=`echo -n "$2" | kstore-enc`
|
||||
_val=`kstore-quote "$_val"`
|
||||
_prop=`kstore-quote "${3:-$_KSTORE_DEF_PROP}"`
|
||||
sqlite3 "$_AUTH_DB" \
|
||||
"INSERT INTO store ( key, prop, data )
|
||||
VALUES( '$_key', '$_prop', X'$_val' );"
|
||||
}
|
||||
|
||||
function kstore-get {
|
||||
kstore-init || return 1
|
||||
local _key _prop _cond
|
||||
_key=`kstore-quote "$1"`
|
||||
_prop=`kstore-quote "${2:-$_KSTORE_DEF_PROP}"`
|
||||
_cond="key = '$_key' AND prop = '$_prop'"
|
||||
|
||||
echo -n "`$_SQLITE -list "$_AUTH_DB" "SELECT ( data ) FROM store WHERE $_cond;"`" | kstore-dec
|
||||
}
|
||||
|
||||
function kstore-del {
|
||||
kstore-init || return 1
|
||||
|
||||
local _CONFIRM _key _prop _cond
|
||||
_key=`kstore-quote "$1"`
|
||||
_prop=`kstore-quote "${2:-$_KSTORE_DEF_PROP}"`
|
||||
_cond="key = '$_key' AND prop = '$_prop'"
|
||||
|
||||
sqlite3 "$_AUTH_DB" "SELECT 1111 FROM store WHERE $_cond;" | grep -q 1111
|
||||
if [ $? -eq 0 ]; then
|
||||
sqlite3 "$_AUTH_DB" "SELECT * FROM store WHERE $_cond;"
|
||||
echo
|
||||
read -p "Delete this entry (yes/no)? " _CONFIRM
|
||||
if [ "$_CONFIRM" == "yes" ]; then
|
||||
sqlite3 "$_AUTH_DB" "DELETE FROM store WHERE $_cond;"
|
||||
if [ $? -eq 0 ]; then
|
||||
echo "deleted"
|
||||
fi
|
||||
else
|
||||
echo "action canceled"
|
||||
fi
|
||||
else
|
||||
echo "$1 not found"
|
||||
fi
|
||||
}
|
||||
|
||||
function kstore-search {
|
||||
echo "Not implemented yet"
|
||||
}
|
||||
|
||||
function kstore-list {
|
||||
$_SQLITE -header -column "$_AUTH_DB" "SELECT key, prop, length( data ) FROM store;"
|
||||
}
|
||||
|
||||
function kstore-query {
|
||||
sqlite3 "$_AUTH_DB" "$@"
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user