Nhắc người dùng đăng nhập bằng root khi chạy shell script

Vấn đề tôi nhận được là, khi tôi nhập lệnh,

su - root

ở đầu tập tin shell shell của nó, nó sẽ nhắc người dùng nhập mật khẩu và sau đó KHÔNG tiếp tục với phần còn lại của tập lệnh shell. Sau đó tôi phải tự xác định vị trí và chạy tập lệnh shell qua thiết bị đầu cuối. Tôi muốn tập lệnh đảm bảo rằng người dùng đăng nhập bằng root và sau đó tiếp tục với phần còn lại của tập lệnh shell.

Nói cách khác, tôi muốn chạy tập lệnh như bất kỳ người dùng nào nhưng ngay khi tập lệnh bắt đầu thực thi, người dùng phải thay đổi thành root và sau đó tiếp tục với phần còn lại của tập lệnh là root cho đến khi hoàn thành. Điều này có thể được thực hiện?

Điều này rất dễ thực hiện:

#!/bin/sh
[ "$(whoami)" != "root" ] && exec sudo -- "$0" "$@"

Khi người dùng hiện tại không root, hãy thực thi lại tập lệnh thông qua sudo.

Lưu ý rằng tôi đang sử dụng sudoở đây thay vì su. Điều này là do nó cho phép bạn bảo tồn các đối số. Nếu bạn sử dụng su, lệnh của bạn sẽ phải xử su -c "$0 $@"lý các đối số của bạn nếu chúng có khoảng trắng hoặc ký tự shell đặc biệt.

Nếu shell của bạn bị bash, bạn có thể tránh cuộc gọi bên ngoài tới whoami:

(( EUID != 0 )) && exec sudo -- "$0" "$@"

Bạn cũng có thể kiểm tra UID:

 if [ $(id -u) != 0 ]; then
     echo "You're not root"
     # elevate script privileges
 fi

Bạn có thể gọi chính kịch bản và kiểm tra:

#! /bin/bash

if [ "root" != "$USER" ]; then
  su -c "$0" root
  exit
fi

...

Tại điểm bạn muốn đăng nhập, chỉ cần thêm lệnh sau

sudo su

Điều này đang hoạt động hoàn hảo với mã của tôi