diff options
| author | Timmy Keller <tjk@tjkeller.xyz> | 2024-09-02 08:44:38 -0500 | 
|---|---|---|
| committer | Timmy Keller <tjk@tjkeller.xyz> | 2024-09-02 08:44:38 -0500 | 
| commit | a4373a898e65604f58299c947882f50294dd813f (patch) | |
| tree | d07fd4eb4ccf9fef1fa8a73d1160b7174c7c317e /misc | |
| parent | 988cadef2c2e51adbfa64da83ec7d25a8de3d924 (diff) | |
| download | scripts-a4373a898e65604f58299c947882f50294dd813f.tar.xz scripts-a4373a898e65604f58299c947882f50294dd813f.zip  | |
various changes over time
Diffstat (limited to 'misc')
| -rwxr-xr-x | misc/diskhealth | 46 | ||||
| -rwxr-xr-x | misc/mkmount | 18 | ||||
| -rwxr-xr-x | misc/mountimg | 42 | ||||
| -rwxr-xr-x | misc/verifoo | 110 | 
4 files changed, 216 insertions, 0 deletions
diff --git a/misc/diskhealth b/misc/diskhealth new file mode 100755 index 0000000..5cce07b --- /dev/null +++ b/misc/diskhealth @@ -0,0 +1,46 @@ +#!/usr/bin/env python3 + +import subprocess +import json +from pprint import pprint + +smart_result = subprocess.run(["sudo", "smartctl", "-Aij", "/dev/sda"], capture_output=True, text=True) + +smart_result = json.loads(smart_result.stdout) + +sector_size = smart_result["logical_block_size"] + +from typing import NamedTuple +class RawValue(NamedTuple): +    value: int +    string: str + +class Flags(NamedTuple): +    value: int +    string: str +    prefailure: bool +    updated_online: bool +    performance: bool +    error_rate: bool +    event_count: bool +    auto_keep: bool + +class SmartAttribute(NamedTuple): +    id: int +    name: str +    value: int +    worst: int +    thresh: int +    when_failed: str +    flags: dict +    raw: dict + +attributes = {} + +for attr in smart_result["ata_smart_attributes"]["table"]: +    attr_l = SmartAttribute(**attr) +    attributes[attr_l.id] = attr_l + +if 241 in attributes: +    lbas_written = attributes[241].raw["value"] +    print((lbas_written * sector_size) / 10**12) diff --git a/misc/mkmount b/misc/mkmount new file mode 100755 index 0000000..b23d85b --- /dev/null +++ b/misc/mkmount @@ -0,0 +1,18 @@ +#!/bin/sh + +set -e + +mpdir="/run/media/$USER" +mkdir -p "$mpdir" + +mp="$1" + +if [ -z "$mp" ]; then +	mp="$(mktemp -d "$mpdir/XXXXXX")" +else +	mp="$mpdir/$mp" +	[ -f "$mp" ] && echo "Error: $mp already exists" && exit 1 +	mkdir -p "$mp" +fi + +echo "$mp" diff --git a/misc/mountimg b/misc/mountimg new file mode 100755 index 0000000..efe886c --- /dev/null +++ b/misc/mountimg @@ -0,0 +1,42 @@ +#!/bin/sh + +set -e + +die() { echo "$(basename "$0"): $1" && exit 1 ; } + +# file check +! [ "$1" ]         && die "No file was specified" +! [ -f "$1" ]      && die "$1 is not a file" +# mountpoint check +! [ "$2" ]         && die "No mountpoint was specified" +! [ -d "$2" ]      && die "Mountpoint $2 is not a directory" +[ "`ls -A "$2"`" ] && die "Mountpoint $2 is not empty" +# check disk image validity +! fdisk -lu "$1" 2>&1 | grep -m1 "\s*Start\s*End" >/dev/null && die "File $1 is not a disk image" + +# get sectorsize +#sectorsize=$(fdisk -lu "$1" | grep -m1 '^Units:' | cut -d' ' -f8)  # Sector size is 8th word +sectorsize=512 + +# get partition num, if there is more than one part, interactively if not available +# TODO check if there is no partitions +partition="$3" +if [ -z "$partition" ]; then +	fdisk -o Size,Type,Name -lu "$1" | awk '!f{print $0};f>0{print "  " f-1 ") " $0; f++};f==-1{print "Part " $0;f=1};/^$/{f=-1}' + +	printf "\nChoose partition number to mount: " +	read partition +fi + +# check if part num valid +[ -z "${partition##*[!0-9]*}" ] && die "Illegal number: '$partition'" + +# get offset from fdisk +fdisk_ignore_lines=9  # ignore first 9 lines of output +offset=$(fdisk -o Start -l "$1" | sed $(( fdisk_ignore_lines + partition ))'q;d' ) + +# check if that produced a valid offset +[ -z "$offset" ] && die "Partition $partition does not exist" + +# try to mount +mount -o loop,offset=$(( offset * sectorsize )) "$1" "$2" && echo "Successfully mounted" || die "Error mounting" diff --git a/misc/verifoo b/misc/verifoo new file mode 100755 index 0000000..34089cb --- /dev/null +++ b/misc/verifoo @@ -0,0 +1,110 @@ +#!/bin/sh + +help() { +cat << HELPDOC +usage: $(basename "$0") [options] [file]... + +OPTIONS: +  -h, --help                show this menu +  -c, --check               read and verify checksums from file +  -r, --recursive           produce a checksum for all files in a directory recursively +  --recursive-sum           produce a checksum for a directory. sensitive to filenames +  --recursive-sum-nopath    produce a checksum for a directory. not sensitive to filenames/placement +  -s, --silent              produce no output to stderr +  -v, --verbose             output a progress bar of hash progress if pv is avaliable (wip) + +MODES: +  -S, --sha                 use sha1sum for checksums +  -M, --md5                 use md5sum for checksums +  -X, --xxh                 use xxhsum for checksums (default if avaliable with sha1sum as fallback) +HELPDOC +} + +# set defaults +cat="pv -F'%r_[%b]_[%t]_[%e]_%p'" +! command -v pv >/dev/null && cat=cat + +hash=xxhsum +! command -v xxhsum >/dev/null && hash=sha1sum + +hashfiles="" +verbose=1 +recursive="" +outputfile="" + +while true; do +	case "$1" in +		-c|--check) [ -n "$2" ] && hashfiles="$hashfiles $2" && shift ;; +		-s|--silent)  verbose=0 ;; +		-v|--verbose) verbose=2 ;; +		--recursive|-r)         recursive=3 ;; +		--recursive-sum)        recursive=2 ;; +		--recursive-sum-nopath) recursive=1 ;; +		-S|--sha) hash=sha1sum ;; +		-M|--md5) hash=md5sum  ;; +		-X|--xxh) hash=xxhsum  ;; +		-h|--help) help; exit 0 ;; +		-*)        help; exit 1 ;; +		*) break ;; +	esac +	shift +done + +case $verbose in +	0) cat=cat ;; +	1) cat=cat ;; +esac + +# run hash +while [ -n "$1" ]; do +	if [ -d "$1" ]; then # is directory +		# check if using -r flag +		case "$recursive" in +			1) +				# get hash of all files, sort hashes, hash hashes +				# TODO make pv work in this mode + +				#catd="$cat" +				#if [ $verbose -ge 2 ] && [ "$catd" != "cat" ]; then +				#	total_files=$(du -a "$1" | wc -l) +				#	catd="pv -l -s$total_files -F'[%b/$total_files]_[%t]_[%e]_%p'" +				#fi +				#sum=`find "$1" -type f -print0 | xargs -r0 $hash | $catd | cut -d' ' -f1 | sort | $hash | cut -d' ' -f1` 2>/dev/null +				sum=`find "$1" -type f -print0 | xargs -r0 $hash | cut -d' ' -f1 | sort | $hash | cut -d' ' -f1` 2>/dev/null +				echo "r/$sum  $1" +			;; +			2) +				# hash sorted filenames +				# sort all filenames w/ full path and hash +				# hash those sums together +				# `tar -c -f - "$1" | $hash` is a good idea but it doesn't work since changes to the fs affect tar +				filehash=`find "$1" -type f -print0 | sort -z | xargs -r0 $cat | $hash | cut -d' ' -f1` +				pathhash=`find "$1" -printf "%P\0" | sort -z | $hash | cut -d' ' -f1` +				sum=`printf "$filehash$pathhash" | $hash | cut -d' ' -f1` +				echo "p/$sum  $1" +			;; +			3) +				# sort files and hash each one in sequence +				# TODO make pv work in this mode +				find "$1" -type f -print0 | sort -z | xargs -r0 $hash +			;; +			*) +				[ $verbose -ge 1 ] && echo "-r not specified for directory $1. Exiting" +				exit 1 +			;; +		esac +	else # is file +		if ! [ -f "$1" ]; then +			[ $verbose -gt 1 ] && echo "File $1 does not exist. Exiting" +			exit 1 +		fi +		files="$1" + +		while [ -f "$2" ]; do +			files="$files\0$2" +			shift +		done +		printf "$files" | xargs -r0 $hash +	fi +	shift +done  | 
