-
Notifications
You must be signed in to change notification settings - Fork 84
/
profile-sync-daemon.in
811 lines (733 loc) · 26.8 KB
/
profile-sync-daemon.in
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
#!/bin/bash
#
# Profile-sync-daemon by graysky <graysky AT archlinux DOT us>
# Inspired by some code originally written by Colin Verot
#
# shellcheck disable=1090,2154
# needed for debian >=8.x
PATH=$PATH:/sbin
BLD="\e[01m"
RED="\e[01;31m"
GRN="\e[01;32m"
BLU="\e[01;34m"
NRM="\e[00m"
VERS="@VERSION@"
user=$(id -un)
HOME="$(getent passwd "$user" | cut -d: -f6)"
XDG_CONFIG_HOME="${XDG_CONFIG_HOME:-$HOME/.config}"
XDG_DATA_HOME="${XDG_DATA_HOME:-$HOME/.local/share}"
PSDCONFDIR="$XDG_CONFIG_HOME/psd"
PSDCONF="$PSDCONFDIR/psd.conf"
SHAREDIR="/usr/share/psd"
VOLATILE="$XDG_RUNTIME_DIR"
PID_FILE="$VOLATILE/psd.pid"
if [[ ! -d "$SHAREDIR" ]]; then
echo -e " ${RED}ERROR:${NRM}${BLD} Missing ${BLU}$SHAREDIR${NRM}${BLD} - reinstall the package to use profile-sync-daemon.${NRM}"
exit 1
fi
if [[ ! -d "$SHAREDIR/browsers" ]]; then
echo -e " ${RED}ERROR:${NRM}${BLD} Missing ${BLU}$SHAREDIR/browsers${NRM}${BLD} - reinstall the package to use profile-sync-daemon.${NRM}"
exit 1
fi
if [[ $EUID -eq 0 ]]; then
echo -e " ${RED}WARNING:${NRM}${BLD} Do not call ${BLU}$0${NRM}${BLD} as root.${NRM}"
exit 1
fi
if [[ ! -d "$XDG_RUNTIME_DIR" ]]; then
echo -e " ${RED}ERROR:${NRM}${BLD} Cannot find XDG_RUNTIME_DIR which should be set by systemd.${NRM}"
exit 1
fi
# Setup check for config file
if [[ -f "$PSDCONF" ]]; then
if [[ ! -f "$PID_FILE" ]]; then
# do nothing if psd is currently running, otherwise
# make sure only comments and variables/arrays are defined to prevent
# problems like issue #166
if grep -Eqv '^$|^#|^[^ ]*=[^;]*' "$PSDCONF"; then
# something that isn't a blank line, comment, or variable present
# so exit
echo -e " ${RED}ERROR:${NRM}${BLD} Syntax error(s) detected in ${BLU}$PSDCONF${NRM}${BLD} - edit and try again.${NRM}"
echo -e "${NRM}${BLD}Line number: offending comment${NRM}"
grep -Evn '^$|^#|^[^ ]*=[^;]*' "$PSDCONF"
exit 1
fi
fi
. "$PSDCONF"
elif [[ -d "$HOME/.psd" ]]; then
# first check to see if a legacy ~/.psd is present and then move it to the
# new location
mkdir -p "$PSDCONFDIR"
rsync -aX "$HOME/.psd/" "$PSDCONFDIR/"
rm -rf "$HOME/.psd"
echo " The use of $HOME/.psd is deprecated. Existing config has been moved to $PSDCONFDIR"
# source it again retaining any user options
[[ -f "$PSDCONF" ]] && . "$PSDCONF"
else
mkdir -p "$PSDCONFDIR"
if [[ -f "$SHAREDIR/psd.conf" ]]; then
cp "$SHAREDIR/psd.conf" "$PSDCONFDIR"
echo -e " First time running psd so please edit ${BLU}$PSDCONF${NRM}${BLD} to your liking and run again.${NRM}"
exit 0
fi
fi
# if psd is active, source the snapshot of psd.conf preferentially
# version 6.03 renames this file so if older version is running then
# rotate the old name to the new one
[[ -f "$PSDCONFDIR/.$user@$(uname -n).pid.conf" ]] &&
mv "$PSDCONFDIR/.$user@$(uname -n).pid.conf" "$PSDCONFDIR/.psd.conf"
if [[ -f "$PID_FILE" ]]; then
if [[ -f "$PSDCONFDIR/.psd.conf" ]]; then
PSDCONF="$PSDCONFDIR/.psd.conf"
unset USE_OVERLAYFS USE_SUSPSYNC BROWSERS USE_BACKUPS BACKUP_LIMIT
. "$PSDCONF"
# defining VOLATILE in the config is deprecated since v6.16
VOLATILE="$XDG_RUNTIME_DIR"
fi
fi
# define default number of crash-recovery snapshots to save if the user did not
# and check that it is an integer if user did define it
if [[ -z "$BACKUP_LIMIT" ]]; then
BACKUP_LIMIT=5
else
if [[ "$BACKUP_LIMIT" =~ ^[0-9]+$ ]]; then
# correctly setup
true
else
echo -e " ${RED}ERROR:${NRM}${BLD} Bad value for BACKUP_LIMIT detected!${NRM}"
exit 1
fi
fi
# scope to sync defined in BROWSERS array or from list of supported browsers
if [[ -z "$BROWSERS" ]]; then
mapfile -t BROWSERS < <(find "$SHAREDIR/browsers" -type f -printf "%f\n")
else
if ! declare -p BROWSERS | grep -q 'declare \-a'; then
# did not setup as array so redefine it here
IFS=' ' read -r -a BROWSERS <<< "${BROWSERS[@]}"
fi
fi
# remove duplicate elements if user accidentally polluted array with same entries
# https://stackoverflow.com/questions/13648410/how-can-i-get-unique-values-from-an-array-in-bash
readarray -t BROWSERS < <(printf '%s\n' "${BROWSERS[@]}" | awk '!x[$0]++')
# simple function to determine user intent rather than using a null value
case "${USE_OVERLAYFS,,}" in
y|yes|true|t|on|1|enabled|enable|use)
OLFS=1
;;
*)
OLFS=0
;;
esac
# simple function to determine user intent rather than using a null value
case "${USE_SUSPSYNC,,}" in
y|yes|true|t|on|1|enabled|enable|use)
SUSPSYNC=1
;;
*)
SUSPSYNC=0
;;
esac
# since the default for this one is a yes, need to force a null value to yes
[[ -z "${USE_BACKUPS,,}" ]] && USE_BACKUPS="yes"
case "${USE_BACKUPS,,}" in
y|yes|true|t|on|1|enabled|enable|use)
CRRE=1
;;
*)
CRRE=0
;;
esac
# determine if we are using overlayfs (v22 and below) or overlay (v23 and above)
# since mount should call modprobe on invocation, check to see if either
# module is in the tree using modinfo
if [[ $OLFS -eq 1 ]]; then
# first check to see if either is hardcoded into the kernel
# and of course prefer version 23
[[ $(grep -ciE "overlayfs$" /proc/filesystems) -eq 1 ]] && OLFSVER=22
[[ $(grep -ciE "overlay$" /proc/filesystems) -eq 1 ]] && OLFSVER=23
fi
if [[ -z $OLFSVER ]]; then
# if neither is hardcoded, see if either module is available
if modinfo overlayfs &>/dev/null; then
OLFSVER=22
fi
if modinfo overlay &>/dev/null; then
OLFSVER=23
fi
fi
header() {
echo -e "${BLD}Profile-sync-daemon v$VERS${NRM}"
echo
}
dep_check() {
# checks for dependencies and that psd.conf is setup correctly
if ! command -v rsync >/dev/null 2>&1; then
echo -e " ${BLD}I require rsync but it's not installed. ${RED}Aborting!${NRM}"
exit 1
fi
if ! command -v modinfo >/dev/null 2>&1; then
echo -e " ${BLD}I require modinfo but it's not installed. ${RED}Aborting!${NRM}" >&2
exit 1
fi
if ! command -v awk >/dev/null 2>&1; then
echo -e " ${BLD}I require awk but it's not installed. ${RED}Aborting!${NRM}" >&2
exit 1
fi
if ! command -v gdbus >/dev/null 2>&1; then
if [[ $SUSPSYNC -eq 1 ]]; then
echo -e " ${BLD}I require gdbus but it's not installed. ${RED}Aborting!${NRM}"
exit 1
fi
fi
if [[ $OLFS -eq 1 ]]; then
[[ $OLFSVER -ge 22 ]] || {
echo -e " ${BLD}Your kernel requires either the ${BLU}overlay${NRM}${BLD} or ${BLU}overlayfs${NRM}${BLD} module to use${NRM}"
echo -e " ${BLD}to use psd's in overlay mode. Cannot find either in your kernel so compile it in and${NRM}"
echo -e " ${BLD}try again or remove the option from ${BLU}$PSDCONF${NRM}${BLD}. ${RED}Aborting!${NRM}" >&2
exit 1
}
fi
if [[ -f /etc/psd.conf ]]; then
echo -e "${BLD} This version of psd does not support /etc/psd.conf since it runs as your user.${NRM}"
echo -e "${BLD} Recommend that you remove /etc/psd.conf and run it in systemd's user mode:${NRM}"
echo -e "${BLD} systemd --user <action> psd.service${NRM}"
exit 1
fi
}
config_check() {
if [[ $OLFS -eq 1 ]]; then
# user must have sudo rights to call /usr/bin/mount
# and /usr/bin/umount to use overlay mode
if ! sudo -kn psd-overlay-helper &>/dev/null; then
FAILCODE=1
fi
if [[ $FAILCODE -ne 0 ]]; then
echo -e "${BLD}${RED} ERROR!${NRM}${BLD} To use overlayfs mode, $user needs sudo access to ${BLU}/usr/bin/psd-overlay-helper${NRM}${BLD}${NRM}"
echo
echo -e " ${BLD}Add the following line to the end of ${BLU}/etc/sudoers${NRM}${BLD} to enable this functionality:${NRM}"
echo -e " ${BLD}$user ALL=(ALL) NOPASSWD: /usr/bin/psd-overlay-helper${NRM}"
exit 1
fi
fi
for browser in "${BROWSERS[@]}"; do
if [[ ! -f "$SHAREDIR/browsers/$browser" ]]; then
# user defined an invalid browser
echo -e " ${BLD}${RED}$browser${NRM}${BLD} is not a supported browser. Check config file for typos: ${NRM}${BLU}$PSDCONF${NRM}"
exit 1
fi
done
}
ungraceful_state_check() {
# if the machine was ungracefully shutdown then the backup will be
# on the filesystem and the link to tmpfs will be on the filesystem
# but the contents will be empty we need to simply remove the link
# and rotate the backup into place
local browser
for browser in "${BROWSERS[@]}"; do
load_env_for "$browser"
for item in "${DIRArr[@]}"; do
DIR="$item"
BACKUP="$item-backup"
BACK_OVFS="$item-back-ovfs"
# all is well so continue
[[ -e "$DIR/.flagged" ]] && continue
NOW=$(date +%Y%m%d_%H%M%S)
if [[ -h "$DIR" ]]; then
# symlinked browser profiles are not supported so bail if one is detected
if [[ -d $(readlink "$DIR") ]]; then
echo -e " ${RED}Warning!${NRM}"
echo -e " ${BLD}${BLU}$DIR${NRM}${BLD} appears to be a symlink but these are not supported.${NRM}"
echo -e " ${BLD}Please make the browser profile a live directory and try again. Exiting.${NRM}"
exit 1
else
echo "Ungraceful state detected for $DIR so fixing"
unlink "$DIR"
# refuse to start browser while recovery
ln -s /dev/null "$DIR"
fi
fi
if [[ -d "$BACKUP" ]]; then
if [[ -d $DIR ]]; then
echo "Unexpected state detected: we have $BACKUP, but $DIR already exists. Trying move $DIR to $DIR" "$DIR-old-profile-$NOW"
mv --no-target-directory "$DIR" "$DIR-old-profile-$NOW"
fi
if [[ -d "$BACK_OVFS" ]]; then
# always snapshot the most recent of these two dirs...
# if using overlayfs $BACK_OVFS and $BACKUP should be compared
# against each other to see which is newer and then that should
# be what psd snapshots since BACKUP (the lowerdir) is readonly
# at the time the user started psd could be many resync cycles
# in the past
BACKUP_TIME=$(stat "$BACKUP" | grep Change | awk '{ print $2,$3 }')
BACK_OVFS_TIME=$(stat "$BACK_OVFS" | grep Change | awk '{ print $2,$3 }')
[[ $(date -d "$BACK_OVFS_TIME" "+%s") -ge $(date -d "$BACKUP_TIME" "+%s") ]] &&
TARGETTOKEEP="$BACK_OVFS" ||
TARGETTOKEEP="$BACKUP"
if [[ $CRRE -eq 1 ]]; then
cp -a --reflink=auto "$TARGETTOKEEP" "$BACKUP-crashrecovery-$NOW"
fi
unlink "$DIR"
mv --no-target-directory "$TARGETTOKEEP" "$DIR"
rm -rf "$BACKUP"
else
# we only find the BACKUP and no BACKOVFS then either the initial resync
# never occurred before the crash using overlayfs or we aren't using overlayfs
# at all which can be treated the same way
if [[ $CRRE -eq 1 ]]; then
cp -a --reflink=auto "$BACKUP" "$BACKUP-crashrecovery-$NOW"
unlink "$DIR"
mv --no-target-directory "$BACKUP" "$DIR"
fi
fi
fi
# if overlayfs was active but is no longer, remove $BACK_OVFS
[[ $OLFS -eq 1 ]] || rm -rf "$BACK_OVFS"
done
done
}
cleanup() {
local browser
for browser in "${BROWSERS[@]}"; do
load_env_for "$browser"
for item in "${DIRArr[@]}"; do
DIR="$item"
local CRASHArr=()
while IFS= read -d '' -r backup; do
CRASHArr=("${CRASHArr[@]}" "$backup")
done < <(find "${DIR%/*}" -type d -name '*crashrecovery*' -print0 | sort -r -z)
if [[ ${#CRASHArr[@]} -gt 0 ]]; then
echo -e "${BLD}Deleting ${#CRASHArr[@]} crashrecovery dir(s) for profile ${BLU}$DIR${NRM}"
for backup in "${CRASHArr[@]}"; do
echo -e "${BLD}${RED} $backup${NRM}"
rm -rf "$backup"
done
else
echo -e "${BLD}Found no crashrecovery dirs for: ${BLU}$DIR${NRM}${BLD}${NRM}"
fi
echo
done
done
}
load_env_for() {
browser="$1"
homedir=$HOME
group=$(id -g "$user")
### Arrays
# profileArr is transient used to store profile paths parsed from
# firefox and aurora
unset profileArr
# DIRArr is a full path corrected for both relative and absolute paths
# reset global variables and arrays
unset DIRArr
unset PSNAME
. "$SHAREDIR/browsers/$browser"
}
running_check() {
# check for browsers running and refuse to start if so
# without this cannot guarantee profile integrity
local browser
for browser in "${BROWSERS[@]}"; do
load_env_for "$browser"
[[ -z "$PSNAME" ]] && continue
if pgrep -x -u "$user" "$PSNAME" &>/dev/null; then
echo "Refusing to start; $browser is running by $user!"
exit 1
fi
done
}
suffix_needed() {
browser=$1
unset check_suffix
. "$SHAREDIR/browsers/$browser"
[[ -n "$check_suffix" ]]
}
dup_check() {
# only for firefox, icecat, seamonkey, and palemoon
# the LAST directory in the profile MUST be unique
# make sure there are no duplicates in ~/.mozilla/<browser>/profiles.ini
local browser
for browser in "${BROWSERS[@]}"; do
load_env_for "$browser"
if suffix_needed "$browser"; then
# nothing to check
[[ -z "${DIRArr[*]}" ]] && continue
# browser is on system so check profiles
#
# check that the LAST DIRECTORY in the full path is unique
unique_count=$(printf "%s\n" "${DIRArr[@]##*/}" | sort -u | wc -l)
# no problems so do nothing
[[ ${#DIRArr[@]} -eq $unique_count ]] && continue
echo -e " ${RED}Error: ${NRM}${BLD}dup profile for ${GRN}$browser${NRM}${BLD} detected. See psd manpage, correct, and try again.${NRM}"
# clip of the 'heftig-' to give correct path
[[ "$browser" = "heftig-aurora" ]] && browser="${browser##*-}"
profile_ini="$homedir/.mozilla/$browser/profiles.ini"
if [[ "$browser" = "palemoon" ]]; then
profile_ini="$homedir/.moonchild productions/pale moon/profiles.ini"
echo -e " ${BLD}Must have unique last directories in ${BLU}${profile_ini}${NRM}${BLD} to use psd.${NRM}"
exit 1
fi
fi
done
}
kill_browsers() {
# check for browsers running and kill them to safely sync/unsync
# without this cannot guarantee profile integrity
local browser
for browser in "${BROWSERS[@]}"; do
load_env_for "$browser"
local x=1
while [[ $x -le 60 ]]; do
[[ -n "$PSNAME" ]] || break
pgrep -x -u "$user" "$PSNAME" &>/dev/null || break
if [[ $x -le 5 ]]; then
pkill -x -SIGTERM -u "$user" "$PSNAME"
else
pkill -x -SIGKILL -u "$user" "$PSNAME"
fi
x=$(( x + 1 ))
sleep .05
done
done
}
do_sync_for() {
browser=$1
load_env_for "$browser"
for item in "${DIRArr[@]}"; do
DIR="$item"
BACKUP="$item-backup"
BACK_OVFS="$item-back-ovfs"
suffix=
if suffix_needed "$browser"; then
suffix="-${item##*/}"
fi
TMP="$VOLATILE/$user-$browser$suffix"
UPPER="$VOLATILE/$user-$browser${suffix}-rw"
WORK="$VOLATILE/.$user-$browser${suffix}"
local REPORT
# make tmpfs container
if [[ -d "$DIR" ]]; then
# retain permissions on sync target
PREFIXP=$(stat -c %a "$DIR")
[[ -r "$TMP" ]] || install -dm"$PREFIXP" --owner="$user" --group="$group" "$TMP"
if [[ $OLFS -eq 1 ]]; then
if [[ $OLFSVER -eq 23 ]]; then
[[ -r "$UPPER" ]] || install -dm"$PREFIXP" --owner="$user" --group="$group" "$UPPER"
[[ -r "$WORK" ]] || install -dm"$PREFIXP" --owner="$user" --group="$group" "$WORK"
elif [[ $OLFSVER -eq 22 ]]; then
[[ -r "$UPPER" ]] || install -dm"$PREFIXP" --owner="$user" --group="$group" "$UPPER"
fi
fi
# backup target and link to tmpfs container
if [[ $(readlink "$DIR") != "$TMP" ]]; then
mv --no-target-directory "$DIR" "$BACKUP"
# refuse to start browser while initial sync
ln -s /dev/null "$DIR"
fi
# sync the tmpfs targets to the disc
if [[ -e "$DIR"/.flagged ]]; then
REPORT="resync"
if [[ $OLFS -eq 1 ]]; then
rsync -aX --delete-after --inplace --no-whole-file --exclude .flagged "$DIR/" "$BACK_OVFS/"
else
rsync -aX --delete-after --inplace --no-whole-file --exclude .flagged "$DIR/" "$BACKUP/"
fi
else
# initial sync
REPORT="sync"
if [[ $OLFS -eq 1 ]]; then
if ! sudo psd-overlay-helper -v "$OLFSVER" -l "$BACKUP" -u "$UPPER" -w "$WORK" -d "$TMP" mountup; then
echo -e "Error in trying to mount $TMP - this should not happen!"
exit 1
fi
else
# keep user from launching browser while rsync is active
rsync -aX --inplace --no-whole-file "$BACKUP/" "$TMP"
fi
# now browser can start
[[ $(readlink "$DIR") = "/dev/null" ]] && unlink "$DIR"
ln -s "$TMP" "$DIR"
chown -h "$user":"$group" "$DIR"
touch "$DIR/.flagged"
fi
echo -e "${BLD}$browser $REPORT successful${NRM}"
else
if [[ ! -d "$homedir" ]] ; then
echo -e "${RED}$DIR does not exist! Is /home unmounted?${NRM}" >&2
exit 1
elif [[ -d "$BACKUP" ]] ; then
echo -e "${RED}$DIR does not exist or is a broken symlink! Is $VOLATILE unmounted?${NRM}" >&2
exit 1
fi
fi
done
}
do_sync() {
touch "$PID_FILE"
# make a snapshot of psd.conf and redefine its location to this snapshot
# while psd is running to keep any edits made to the "live" psd.conf from
# potentially orphaning the snapshot copies thus preserving the data
if [[ ! -f "$PSDCONFDIR/.psd.conf" ]]; then
{
echo "# Automatically generated file; DO NOT EDIT!"
echo "# The purpose is to snapshot the settings used when psd was activated."
echo "# Any edits to the live config: $PSDCONFDIR/psd.conf"
echo "# will be applied the _next_ time psd is activated."
echo "#"
echo "USE_OVERLAYFS=\"$USE_OVERLAYFS\""
echo "USE_SUSPSYNC=\"$USE_SUSPSYNC\""
echo "BROWSERS=\"${BROWSERS[*]}\""
echo "USE_BACKUPS=\"$USE_BACKUPS\""
} >> "$PSDCONFDIR/.psd.conf"
chmod 400 "$PSDCONFDIR/.psd.conf"
fi
local browser
for browser in "${BROWSERS[@]}"; do
do_sync_for "$browser"
done
}
enforce() {
local browser
for browser in "${BROWSERS[@]}"; do
local CRASHArr=()
while IFS= read -d '' -r backup; do
CRASHArr=("${CRASHArr[@]}" "$backup")
done < <(find "${DIR%/*}" -type d -name '*crashrecovery*' -print0 | sort -r -z)
if [[ ${#CRASHArr[@]} -gt $BACKUP_LIMIT ]]; then
for remove in "${CRASHArr[@]:$BACKUP_LIMIT}"; do
rm -rf "$remove"
done
fi
done
}
do_unsync() {
rm -f "$PID_FILE" "$PSDCONFDIR/.psd.conf"
local browser
for browser in "${BROWSERS[@]}"; do
load_env_for "$browser"
for item in "${DIRArr[@]}"; do
DIR="$item"
BACKUP="$item-backup"
BACK_OVFS="$item-back-ovfs"
suffix=
if suffix_needed "$browser"; then
suffix="-${item##*/}"
fi
TMP="$VOLATILE/$user-$browser$suffix"
UPPER="$VOLATILE/$user-$browser${suffix}-rw"
WORK="$VOLATILE/.$user-$browser${suffix}"
# check if user has browser profile
if [[ -h "$DIR" ]]; then
unlink "$DIR"
# this assumes that the backup is always updated so
# be sure to invoke a sync before an unsync
#
# restore original dirtree
[[ -d "$BACKUP" ]] && mv --no-target-directory "$BACKUP" "$DIR"
if [[ $OLFS -eq 1 ]] && mountpoint -q "$TMP"; then
rsync -aX --delete-after --inplace --no-whole-file --exclude .flagged "$BACK_OVFS/" "$DIR/"
sudo psd-overlay-helper -d "$TMP" -w "$WORK" mountdown && rm -rf "$TMP" "$UPPER"
fi
[[ -d "$TMP" ]] && rm -rf "$TMP"
echo -e "${BLD}$browser unsync successful${NRM}"
else
if [[ ! -d "$homedir" ]] ; then
echo -e "${RED}$DIR does not exist! Is /home unmounted?${NRM}" >&2
exit 1
fi
fi
done
done
}
parse() {
psd_state=$(systemctl --user is-active psd)
resync_state=$(systemctl --user is-active psd-resync.timer)
psd_suspend_sync_state=$(pgrep -cf "psd-suspend-sync")
if [[ "$psd_suspend_sync_state" != 0 ]]; then
psss_state="enabled"
else
psss_state="disabled"
fi
if [[ $OLFS -eq 1 ]]; then
olfs_color="${GRN}"
ofls_state="enabled"
else
olfs_color="${RED}"
ofls_state="disabled"
fi
[[ "$psd_state" = "active" ]] && psd_color="${GRN}" || psd_color="${RED}"
[[ "$resync_state" = "active" ]] && resync_color="${GRN}" || resync_color="${RED}"
[[ "$psd_suspend_sync_state" != 0 ]] && psss_color="${GRN}" || psss_color="${RED}"
echo -en " ${BLD}systemd service:"
echo -e "$(tput cr)$(tput cuf 17) ${psd_color}$psd_state${NRM}${BLD}${NRM}"
echo -en " ${BLD}resync-timer:"
echo -e "$(tput cr)$(tput cuf 17) ${resync_color}$resync_state${NRM}${BLD}${NRM}"
echo -en " ${BLD}sync on sleep:"
echo -e "$(tput cr)$(tput cuf 17) ${psss_color}$psss_state${NRM}${BLD}${NRM}"
echo -en " ${BLD}use overlayfs:"
echo -e "$(tput cr)$(tput cuf 17) ${olfs_color}$ofls_state${NRM}${BLD}${NRM}"
echo
echo -e "${BLD}Psd will manage the following per ${BLU}${PSDCONF}${NRM}${BLD}:${NRM}"
echo
local browser
for browser in "${BROWSERS[@]}"; do
load_env_for "$browser"
for item in "${DIRArr[@]}"; do
DIR="$item"
BACKUP="$item-backup"
suffix=
if suffix_needed "$browser"; then
suffix="-${item##*/}"
fi
UPPER="$VOLATILE/$user-$browser${suffix}-rw"
if [[ -d "$DIR" ]]; then
local CRASHArr=()
while IFS= read -d '' -r backup; do
CRASHArr=("${CRASHArr[@]}" "$backup")
done < <(find "${DIR%/*}" -type d -name '*crashrecovery*' -print0 | sort -r -z)
# get permissions on profile dir and be smart about it since symlinks are all 777
[[ -f $PID_FILE ]] && TRUEP=$(stat -c %a "$BACKUP") || TRUEP=$(stat -c %a "$DIR")
# since $XDG_RUNTIME_DIR is 700 by default so pass on by
if [[ "$VOLATILE" = "$XDG_RUNTIME_DIR" ]]; then
warn=
else
# using something other than $XDG_RUNTIME_DIR so check for privacy
[[ $TRUEP -ne 700 ]] && warn=1
fi
# profile dir size
psize=$(du -Dh --max-depth=0 "$DIR" 2>/dev/null | awk '{ print $1 }')
echo -en " ${BLD}browser/psname:"
echo -e "$(tput cr)$(tput cuf 17) $browser/$PSNAME${NRM}"
echo -en " ${BLD}owner/group id:"
echo -e "$(tput cr)$(tput cuf 17) $user/$group${NRM}"
echo -en " ${BLD}sync target:"
echo -e "$(tput cr)$(tput cuf 17) ${BLU}$DIR${NRM}"
if [[ $warn -eq 1 ]]; then
echo -e "$(tput cr)$(tput cuf 17) ${RED} Permissions are $TRUEP on this profile.${NRM}"
echo -e "$(tput cr)$(tput cuf 17) ${RED} Recommend a setting of 700 for increased privacy!${NRM}"
warn=
fi
echo -en " ${BLD}tmpfs dir:"
echo -e "$(tput cr)$(tput cuf 17) ${GRN}$VOLATILE/$user-$browser$suffix${NRM}"
echo -en " ${BLD}profile size:"
echo -e "$(tput cr)$(tput cuf 17) $psize${NRM}"
if [[ $OLFS -eq 1 ]]; then
rwsize=$(du -Dh --max-depth=0 "$UPPER" 2>/dev/null | awk '{ print $1 }')
echo -en " ${BLD}overlayfs size:"
echo -e "$(tput cr)$(tput cuf 17) $rwsize${NRM}"
fi
if [[ $BACKUP_LIMIT -ne 5 ]]; then
# only print if not the hardcoded default
echo -en " ${BLD}backup limit:"
echo -e "$(tput cr)$(tput cuf 17) $BACKUP_LIMIT${NRM}"
fi
echo -en " ${BLD}recovery dirs:"
if [[ "${#CRASHArr[@]}" -eq 0 ]]; then
echo -e "$(tput cr)$(tput cuf 17) none${NRM}"
else
echo -e "$(tput cr)$(tput cuf 17) ${RED}${#CRASHArr[@]}${NRM}${BLD} <- delete with the c option${NRM}"
for backup in "${CRASHArr[@]}"; do
psize=$(du -Dh --max-depth=0 "$backup" 2>/dev/null | awk '{ print $1 }')
echo -en " ${BLD} dir path/size:"
echo -e "$(tput cr)$(tput cuf 17) ${BLU}$backup ${NRM}${BLD}($psize)${NRM}"
done
fi
echo
fi
done
done
}
take_inhibit_lock() {
# ensure we only take one lock at a time
release_inhibit_lock
# background this to avoid hanging on startup
/usr/bin/psd-suspend-sync &
}
release_inhibit_lock() {
[[ "$(pgrep -cf "psd-suspend-sync")" != 0 ]] && pkill -f psd-suspend-sync
}
case "$1" in
p|P|Parse|parse|Preview|preview|debug)
dep_check
config_check
dup_check
ungraceful_state_check
header
parse
;;
c|C|clean|Clean)
dep_check
config_check
dup_check
header
cleanup
;;
startup)
# target for psd.service's ExecStart= for consistent error handling
if [[ ! -f $PID_FILE ]]; then
dep_check
config_check
dup_check
running_check
ungraceful_state_check
[[ $SUSPSYNC -eq 1 ]] && take_inhibit_lock
echo -e "${BLD}psd startup check successful${NRM}"
fi
;;
suspend-sync)
if [[ -f $PID_FILE ]]; then
# system preparing for sleep mode, resync when psd is active
do_sync
fi
;;
recycle-inhibit-lock)
# take inhibit lock for next suspend cycle
take_inhibit_lock
;;
sync|resync)
if [[ -f $PID_FILE ]]; then
do_sync
else
dep_check
config_check
dup_check
running_check
ungraceful_state_check
do_sync
enforce
fi
;;
unsync)
if [[ -f $PID_FILE ]]; then
do_sync
kill_browsers
do_unsync
[[ $SUSPSYNC -eq 1 ]] && release_inhibit_lock
fi
;;
*)
header
echo -e " ${BLD}$0 ${NRM}${GRN}[option]${NRM}"
echo -e " ${BLD} ${NRM}${GRN}preview${NRM}${BLD} Parse config file (${NRM}${BLU}${PSDCONF}${NRM}${BLD}) to see which profiles will be managed.${NRM}"
echo -e " ${BLD} ${NRM}${GRN}clean${NRM}${BLD} Clean (delete without prompting) ALL crashrecovery dirs for all profiles.${NRM}"
echo
echo -e " ${BLD}It is ${RED}HIGHLY DISCOURAGED${NRM}${BLD} to directly call $0 to sync, resync, or to unsync.${NRM}"
echo
if [[ -f /usr/lib/systemd/user/psd.service ]]; then
echo -e " ${BLD}Instead, use systemd to start/stop profile-sync-daemon.${NRM}"
echo
echo -e " ${BLD}systemctl --user ${NRM}${GRN}[option]${NRM}${BLD} psd${NRM}"
echo -e " ${BLD} ${NRM}${GRN}start${NRM}${BLD} Turn on daemon; make symlinks and actively manage targets in tmpfs.${NRM}"
echo -e " ${BLD} ${NRM}${GRN}stop${NRM}${BLD} Turn off daemon; remove symlinks and rotate tmpfs data back to disc.${NRM}"
echo -e " ${BLD} ${NRM}${GRN}enable${NRM}${BLD} Autostart daemon when system comes up.${NRM}"
echo -e " ${BLD} ${NRM}${GRN}disable${NRM}${BLD} Remove daemon from the list of autostart daemons.${NRM}"
elif [[ -f /etc/init.d/psd ]]; then
echo -e " ${BLD}Instead, use the init system to start/stop profile-sync-daemon.${NRM}"
echo
echo -e " ${BLD}sudo service psd ${NRM}${GRN}[option]${NRM}${BLD} or /etc/init.d/psd ${NRM}${GRN}[option]${NRM}"
echo -e " ${BLD} ${NRM}${GRN}start${NRM}${BLD} Turn on daemon; make symlinks and actively manage targets in tmpfs.${NRM}"
echo -e " ${BLD} ${NRM}${GRN}stop${NRM}${BLD} Turn off daemon; remove symlinks and rotate tmpfs data back to disc.${NRM}"
fi
;;
esac
exit 0
# vim:set ts=2 sw=2 et: