2022-12-13 15:11:13 +01:00
|
|
|
#!/usr/bin/env bash
|
2016-04-18 12:05:19 +02:00
|
|
|
|
2017-05-02 15:13:05 +02:00
|
|
|
. /etc/log2ram.conf
|
2016-12-02 01:26:09 +01:00
|
|
|
|
2020-05-21 20:37:27 +02:00
|
|
|
if [ -z "$PATH_DISK" ]; then
|
2020-09-30 09:52:39 +02:00
|
|
|
PATH_DISK='/var/log'
|
2020-05-21 20:37:27 +02:00
|
|
|
fi
|
|
|
|
|
2020-09-30 09:52:39 +02:00
|
|
|
LOG_NAME='log2ram.log'
|
2022-02-03 12:13:37 +01:00
|
|
|
NO_RSYNC=${USE_RSYNC#true}
|
2016-12-01 01:27:06 +01:00
|
|
|
|
2022-12-01 13:58:54 +01:00
|
|
|
## @fn is_safe()
|
|
|
|
## @brief Check if hdd log exists
|
|
|
|
is_safe() {
|
2020-09-30 09:52:39 +02:00
|
|
|
[ -d "$HDD_LOG" ] || echo "ERROR: $HDD_LOG/ doesn't exist! Can't sync."
|
|
|
|
[ -d "$HDD_LOG" ] || exit 1
|
2016-12-02 01:26:09 +01:00
|
|
|
}
|
|
|
|
|
2022-12-13 15:11:13 +01:00
|
|
|
## @fn journald_logrotate()
|
|
|
|
## @brief Logrotate the journal if the current RAM_LOG path is part of the journald directory
|
|
|
|
journald_logrotate() {
|
|
|
|
if ! [ -x "$(command -v journalctl)" ] || ! [ "$JOURNALD_AWARE" = true ]; then
|
|
|
|
return 1
|
|
|
|
fi
|
|
|
|
|
|
|
|
if journalctl --header | grep "File path" | grep "$RAM_LOG" >/dev/null 2>&1; then
|
|
|
|
journalctl --rotate
|
|
|
|
return 0
|
|
|
|
else
|
|
|
|
return 1
|
|
|
|
fi
|
|
|
|
}
|
|
|
|
|
2022-12-01 13:58:54 +01:00
|
|
|
## @fn sync_to_disk()
|
|
|
|
## @brief Sync memory back to hard disk
|
|
|
|
sync_to_disk() {
|
|
|
|
is_safe
|
2020-05-14 23:20:20 +02:00
|
|
|
|
2022-12-13 15:11:13 +01:00
|
|
|
optional_params=()
|
|
|
|
|
|
|
|
if journald_logrotate; then
|
|
|
|
optional_params+=("--include=journal/*/*@*.journal")
|
|
|
|
optional_params+=("--exclude=journal/*/*")
|
|
|
|
fi
|
|
|
|
|
|
|
|
if [ -z "${NO_RSYNC}" ] && [ -x "$(command -v rsync)" ]; then
|
|
|
|
rsync -aXv --inplace --no-whole-file --delete-after "${optional_params[@]}" "$RAM_LOG"/ "$HDD_LOG"/ 2>&1 |
|
|
|
|
tee -a "$LOG2RAM_LOG"
|
2016-12-01 01:27:06 +01:00
|
|
|
else
|
2020-09-30 09:52:39 +02:00
|
|
|
cp -rfup "$RAM_LOG"/ -T "$HDD_LOG"/ 2>&1 | tee -a "$LOG2RAM_LOG"
|
2016-12-01 01:27:06 +01:00
|
|
|
fi
|
|
|
|
}
|
|
|
|
|
2022-12-01 13:58:54 +01:00
|
|
|
## @fn sync_from_disk()
|
|
|
|
## @brief Sync hard disk to memory
|
|
|
|
sync_from_disk() {
|
|
|
|
is_safe
|
2020-05-14 23:20:20 +02:00
|
|
|
|
2020-05-07 11:52:44 +02:00
|
|
|
TP_SIZE=$SIZE
|
|
|
|
if [ "$ZL2R" = true ]; then
|
|
|
|
TP_SIZE=$LOG_DISK_SIZE
|
2020-05-07 13:46:01 +02:00
|
|
|
fi
|
2020-09-30 09:52:39 +02:00
|
|
|
|
|
|
|
if [ -n "$(du -sh -t "$TP_SIZE" "$HDD_LOG"/ | cut -f1)" ]; then
|
|
|
|
echo "ERROR: RAM disk for \"$HDD_LOG/\" too small. Can't sync."
|
|
|
|
umount -l "$RAM_LOG"/
|
|
|
|
umount -l "$HDD_LOG"/
|
2017-12-07 17:42:40 +01:00
|
|
|
if [ "$MAIL" = true ]; then
|
2022-12-01 13:58:54 +01:00
|
|
|
echo "LOG2RAM : No place on RAM for \"$HDD_LOG/\" anymore, fallback on the disk" | mail -s 'Log2Ram Error' root
|
2017-12-07 17:42:40 +01:00
|
|
|
fi
|
2017-04-18 15:53:36 +02:00
|
|
|
exit 1
|
|
|
|
fi
|
|
|
|
|
2022-12-13 15:11:13 +01:00
|
|
|
if [ -z "${NO_RSYNC}" ] && [ -x "$(command -v rsync)" ]; then
|
2020-09-30 09:52:39 +02:00
|
|
|
rsync -aXv --inplace --no-whole-file --delete-after "$HDD_LOG"/ "$RAM_LOG"/ 2>&1 | tee -a "$LOG2RAM_LOG"
|
2016-12-01 01:27:06 +01:00
|
|
|
else
|
2020-09-30 09:52:39 +02:00
|
|
|
cp -rfup "$HDD_LOG"/ -T "$RAM_LOG"/ 2>&1 | tee -a "$LOG2RAM_LOG"
|
2016-12-01 01:27:06 +01:00
|
|
|
fi
|
2020-05-14 23:20:20 +02:00
|
|
|
|
2016-12-01 01:27:06 +01:00
|
|
|
}
|
|
|
|
|
2022-12-01 13:58:54 +01:00
|
|
|
## @fn wait_for()
|
|
|
|
## @brief Wait for directory and create test file to make sure the directory exists
|
|
|
|
## @param param1 path to the directory
|
|
|
|
wait_for() {
|
|
|
|
WAIT_PATH="$1"
|
|
|
|
while ! findmnt "$WAIT_PATH" >/dev/null; do
|
2022-01-05 22:25:07 +01:00
|
|
|
sleep 0.1
|
|
|
|
done
|
2016-04-18 12:05:19 +02:00
|
|
|
|
2022-12-01 13:58:54 +01:00
|
|
|
while [ ! -f "$WAIT_PATH/log2ram.test" ]; do
|
|
|
|
touch "$WAIT_PATH/log2ram.test"
|
|
|
|
sleep 0.1
|
|
|
|
done
|
|
|
|
rm "$WAIT_PATH/log2ram.test"
|
2019-03-08 05:58:06 +01:00
|
|
|
}
|
|
|
|
|
2022-12-01 13:58:54 +01:00
|
|
|
## @fn create_zram_log_drive()
|
|
|
|
## @brief Create zram log device
|
|
|
|
create_zram_log_drive() {
|
|
|
|
# Check Zram Class created
|
|
|
|
if [ ! -d "/sys/class/zram-control" ]; then
|
|
|
|
modprobe zram
|
|
|
|
RAM_DEV='0'
|
|
|
|
else
|
|
|
|
RAM_DEV=$(cat /sys/class/zram-control/hot_add)
|
|
|
|
fi
|
|
|
|
echo "$COMP_ALG" >"/sys/block/zram${RAM_DEV}/comp_algorithm"
|
|
|
|
echo "$LOG_DISK_SIZE" >"/sys/block/zram${RAM_DEV}/disksize"
|
|
|
|
echo "$SIZE" >"/sys/block/zram${RAM_DEV}/mem_limit"
|
|
|
|
mke2fs -t ext4 "/dev/zram${RAM_DEV}"
|
2020-05-14 23:20:20 +02:00
|
|
|
}
|
|
|
|
|
2016-04-18 12:05:19 +02:00
|
|
|
case "$1" in
|
2022-12-01 13:58:54 +01:00
|
|
|
start)
|
|
|
|
IFS=';'
|
|
|
|
for i in $PATH_DISK; do
|
|
|
|
# Skip the path if the folder doesn't exist
|
|
|
|
[ ! -d "$i" ] && continue
|
|
|
|
|
|
|
|
PATH_FIRST_PART="${i%/*}"
|
|
|
|
PATH_LAST_PART="${i##/*/}"
|
|
|
|
RAM_LOG="$i"
|
|
|
|
HDD_LOG="${PATH_FIRST_PART}/hdd.${PATH_LAST_PART}"
|
|
|
|
LOG2RAM_LOG="${RAM_LOG}/${LOG_NAME}"
|
|
|
|
|
|
|
|
[ -d "$HDD_LOG" ] || mkdir "$HDD_LOG"
|
|
|
|
|
|
|
|
mount --bind "$RAM_LOG"/ "$HDD_LOG"/
|
|
|
|
mount --make-private "$HDD_LOG"/
|
|
|
|
wait_for "$HDD_LOG"
|
|
|
|
|
|
|
|
if [ "$ZL2R" = true ]; then
|
|
|
|
create_zram_log_drive
|
|
|
|
mount -t ext4 -o nosuid,noexec,noatime,nodev,user=log2ram "/dev/zram${RAM_DEV}" "$RAM_LOG"/
|
|
|
|
else
|
|
|
|
mount -t tmpfs -o "nosuid,noexec,noatime,nodev,mode=0755,size=${SIZE}" log2ram "$RAM_LOG"/
|
|
|
|
fi
|
|
|
|
wait_for "$RAM_LOG"
|
|
|
|
sync_from_disk
|
|
|
|
done
|
|
|
|
exit 0
|
|
|
|
;;
|
|
|
|
|
|
|
|
stop)
|
|
|
|
IFS=';'
|
|
|
|
for i in $PATH_DISK; do
|
|
|
|
PATH_FIRST_PART="${i%/*}"
|
|
|
|
PATH_LAST_PART="${i##/*/}"
|
|
|
|
RAM_LOG="$i"
|
|
|
|
HDD_LOG="${PATH_FIRST_PART}/hdd.${PATH_LAST_PART}"
|
|
|
|
LOG2RAM_LOG="${RAM_LOG}/${LOG_NAME}"
|
|
|
|
|
|
|
|
sync_to_disk
|
|
|
|
#ZRAM_LOG=$(awk '$2 == "/var/log" {print $1}' /proc/mounts)
|
|
|
|
#ZRAM_LOG=$(echo ${ZRAM_LOG} | grep -o -E '[0-9]+')
|
|
|
|
umount -l "$RAM_LOG"/
|
|
|
|
umount -l "$HDD_LOG"/
|
|
|
|
# Unsure as even with Root permision denied
|
|
|
|
#echo ${ZRAM_LOG} > /sys/class/zram-control/hot_remove
|
|
|
|
done
|
|
|
|
exit 0
|
|
|
|
;;
|
|
|
|
|
|
|
|
write)
|
|
|
|
IFS=';'
|
|
|
|
for i in $PATH_DISK; do
|
|
|
|
PATH_FIRST_PART="${i%/*}"
|
|
|
|
PATH_LAST_PART="${i##/*/}"
|
|
|
|
RAM_LOG="$i"
|
|
|
|
HDD_LOG="${PATH_FIRST_PART}/hdd.${PATH_LAST_PART}"
|
|
|
|
LOG2RAM_LOG="${RAM_LOG}/${LOG_NAME}"
|
|
|
|
|
|
|
|
sync_to_disk
|
|
|
|
done
|
|
|
|
exit 0
|
|
|
|
;;
|
|
|
|
|
|
|
|
*)
|
|
|
|
echo 'Usage: log2ram {start|stop|write}' >&2
|
|
|
|
exit 1
|
|
|
|
;;
|
2016-04-18 12:05:19 +02:00
|
|
|
esac
|