diff options
| -rwxr-xr-x | bin/envstore | 210 |
1 files changed, 94 insertions, 116 deletions
diff --git a/bin/envstore b/bin/envstore index fdedb20..289777b 100755 --- a/bin/envstore +++ b/bin/envstore @@ -1,116 +1,94 @@ -#!/usr/bin/env python -# envstore - save and restore environment variables -# -# Copyright (C) 2008 Maximilian Gass <mxey@cloudconnected.org> -# Permission to use, copy, modify, and/or distribute this software for any -# purpose with or without fee is hereby granted, provided that the above -# copyright notice and this permission notice appear in all copies. -# -# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES -# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF -# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR -# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES -# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN -# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF -# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. -# -# Minor modifications by Daniel Friesel <derf@derf.homelinux.org> - -import cPickle as pickle -import getopt -import os -import stat -import sys - -def usage(): - print >> sys.stderr, """envstore - save and restore environment variables - -Usage: - envstore COMMAND [ARG] - -Commands: - show List saved variables in human-readable format - eval Generate shell code to set the variables - save Save a variable - rm Delete a variable - clear Clear out the stored variable""" - sys.exit(1) - -def save_store(): - store_file = open(store_filename, 'w') - pickle.dump(store, store_file, protocol=-1) - store_file.close() - - raw_file = open(raw_filename, 'w') - for item, value in store.iteritems(): - print >> raw_file, "export " + item + "=" + value - raw_file.close() - -def get_key(): - try: - return sys.argv[2] - except IndexError: - print >> sys.stderr, "Usage: envstore " + command + " VARIABLE" - sys.exit(2) - -try: - command = sys.argv[1] -except IndexError: - usage() - -store_filename = "/tmp/envstore-" + str(os.getuid()) -raw_filename = os.environ["HOME"] + "/var/tmp/envstore-raw-" + str(os.getuid()) - -try: - if os.stat(store_filename).st_uid != os.getuid(): - print >> sys.stderr, "envstore: Store file does not belong to current user" - sys.exit(6) - store_file = open(store_filename) - store = pickle.load(store_file) - store_file.close() -except (IOError, OSError): - store = {} - -if command == "show": - for k, v in store.iteritems(): - print k + "\t\t" + v - print "%i entries" % len(store) - -elif command == "eval": - for k, v in store.iteritems(): - print "export " + k + "=" + v - -elif command == "save": - arg = get_key() - - if "=" in arg: - key, value = arg.split('=', 2) - - else: - key = arg - try: - value = os.environ[key] - except KeyError: - print >> sys.stderr, key + " does not exist in environment" - sys.exit(3) - - store[key] = value - save_store() - -elif command == "rm": - key = get_key() - - try: - del store[key] - except KeyError: - print >> sys.stderr, key + " does not exist in store" - sys.exit(4) - - save_store() - -elif command == "clear": - os.remove(store_filename) - os.remove(raw_filename) - -else: - usage() +#!/usr/bin/env perl +## envstore - save and load environment variables +## Copyright © 2009 by Daniel Friesel <derf@derf.homelinux.org> +## License: WTFPL <http://sam.zoy.org/wtfpl> +use strict; +use warnings; + +my $store_file = "/tmp/.envstore-$>"; +my %store; +my $action = shift; +my $arg = shift; +my ($key, $value); + +sub check_store { + my ($mode, $uid); + unless (-e $store_file) { + return(0); + } + ($mode, $uid) = (stat($store_file))[2,4]; + $mode &= 0x00077; + if ($uid != $<) { + print STDERR "envstore: store file not owned by us\n"; + exit(1); + } + if ($mode > 0) { + print STDERR "envstore: store file writable by group/others\n"; + printf("%o", $mode); + exit(1); + } + return(1); +} + +sub load_store { + my ($key, $value); + return unless check_store; + open(my $handle, '<', $store_file) or die("Cannot read $store_file: $!"); + while (<$handle>) { + chomp; + /^(\S+)\s+(.*)$/ or next; + ($key, $value) = ($1, $2); + $store{$key} = $value; + } + close($handle); +} + +sub save_store { + my $key; + umask(0077); + open(my $handle, '>', $store_file) or die("Cannot open $store_file: $!"); + foreach $key (keys(%store)) { + print $handle "$key\t$store{$key}\n"; + } + close($handle); +} + +sub get_keyvalue { + my $arg = shift; + my ($key, $value); + $arg =~ /^(\w+)(?:=(.*))?$/; + ($key, $value) = ($1, $2); + unless (defined($2)) { + if (exists($ENV{$key})) { + $value = $ENV{$key}; + } else { + print STDERR "No such parameter: $key"; + exit(1); + } + } + return($key, $value); +} + +load_store; +if ($action eq 'save') { + ($key, $value) = get_keyvalue($arg); + $store{$key} = $value; + save_store; +} elsif ($action eq 'eval') { + while (($key, $value) = each(%store)) { + $value =~ s/'/'"'"'/g; + print "export $key='$value'\n"; + } +} elsif ($action eq 'show') { + while (($key, $value) = each(%store)) { + printf("%-15s = %s\n", $key, $value); + } +} elsif ($action eq 'rm') { + delete($store{$arg}); + save_store; +} elsif ($action eq 'clear') { + unlink($store_file); +} else { + print STDERR "Usage: envstore <save|eval|show|rm|clear> [args]\n"; + exit(1); +} |