主頁 > 後端開發 > 如何寫出安全的、基本功能完善的Bash腳本

如何寫出安全的、基本功能完善的Bash腳本

2020-12-28 06:13:31 後端開發

每個人或多或少總會碰到要使用并且自己完成撰寫一個最基礎的Bash腳本的情況,真實情況是,沒有人會說“哇哦,我喜歡寫這些腳本”,所以這也是為什么很少有人在寫的時候專注在這些腳本上,

我本身也不是一個Bash腳本專家,但是我會在本文中跟你展示一個最基礎最簡單的安全腳本模板,會讓你寫的Bash腳本更加安全實用,你掌握了之后肯定會受益匪淺,

為什么要寫Bash腳本

其實關于Bash腳本最好的解釋如下:

The opposite of "it's like riding a bike" is "it's like programming in bash".

A phrase which means that no matter how many times you do something, you will have to re-learn it every single time.

— Jake Wharton (@JakeWharton)

December 2, 2020

意思就是,跟騎自行車相反,無論做了多少次,每次都感覺像重新學一樣,

但是Bash腳本語言和其他一些廣受歡迎的語言,例如JavaScript一樣,他們不會輕易突然消失,雖然Bash腳本語言不太可能成為業界的主流語言,但實際他就在我們周圍,無處不在,

Bash就像繼承了shell的衣缽一樣,在每臺linux上都可以看到他的身影,這可是大多數后端程式運行的環境,因此當你需要撰寫服務器的應用程式啟動、CI/CD步驟或集成測驗用的腳本,Bash就在那里等著你,

將幾個命令粘在一起,將輸出從一個傳遞到另一個,然后只啟動一些可執行檔案,Bash是眾多方案中最簡單的一個,雖然用其他語言撰寫更大、更復雜的腳本更有效果,但你不能指望Python、Ruby、fish或其他任何你認為最好的程式,可以在任何地方編譯使用,所以在將其添加到某個prod server、Docker image或CI環境之前,往往會讓人三思而后行,

當然啦,Bash還遠遠不夠完美兩個字,他的語法對初學者就像一個噩夢,錯誤處理也很困難,到處都是我們必須處理掉的陷阱,

Bash script template(Bash腳本模板)

廢話不多說,獻上我的模板


#!/usr/bin/env bash

set -Eeuo pipefail
trap cleanup SIGINT SIGTERM ERR EXIT

script_dir=$(cd "$(dirname "${BASH_SOURCE[0]}")" &>/dev/null && pwd -P)

usage() {
  cat <<EOF
Usage: $(basename "${BASH_SOURCE[0]}") [-h] [-v] [-f] -p param_value arg1 [arg2...]

Script description here.

Available options:

-h, --help      Print this help and exit
-v, --verbose   Print script debug info
-f, --flag      Some flag description
-p, --param     Some param description
EOF
  exit
}

cleanup() {
  trap - SIGINT SIGTERM ERR EXIT
  # script cleanup here
}

setup_colors() {
  if [[ -t 2 ]] && [[ -z "${NO_COLOR-}" ]] && [[ "${TERM-}" != "dumb" ]]; then
    NOFORMAT='\033[0m' RED='\033[0;31m' GREEN='\033[0;32m' ORANGE='\033[0;33m' BLUE='\033[0;34m' PURPLE='\033[0;35m' CYAN='\033[0;36m' YELLOW='\033[1;33m'
  else
    NOFORMAT='' RED='' GREEN='' ORANGE='' BLUE='' PURPLE='' CYAN='' YELLOW=''
  fi
}

msg() {
  echo >&2 -e "${1-}"
}

die() {
  local msg=$1
  local code=${2-1} # default exit status 1
  msg "$msg"
  exit "$code"
}

parse_params() {
  # default values of variables set from params
  flag=0
  param=''

  while :; do
    case "${1-}" in
    -h | --help) usage ;;
    -v | --verbose) set -x ;;
    --no-color) NO_COLOR=1 ;;
    -f | --flag) flag=1 ;; # example flag
    -p | --param) # example named parameter
      param="${2-}"
      shift
      ;;
    -?*) die "Unknown option: $1" ;;
    *) break ;;
    esac
    shift
  done

  args=("$@")

  # check required params and arguments
  [[ -z "${param-}" ]] && die "Missing required parameter: param"
  [[ ${#args[@]} -eq 0 ]] && die "Missing script arguments"

  return 0
}

parse_params "$@"
setup_colors

# script logic here

msg "${RED}Read parameters:${NOFORMAT}"
msg "- flag: ${flag}"
msg "- param: ${param}"
msg "- arguments: ${args[*]-}"

Choose Bash

#!/usr/bin/env bash

腳本為了獲得最佳兼容性,它參考/usr/bin/env,而不是直接參考/bin/bash,

Fail fast

set -Eeuo pipefail

set命令可以更改腳本執行選項,例如,通常Bash不關心某個命令是否失敗,回傳非零退出狀態代碼,它只是快速地跳到下一個,現在考慮一下這個小腳本:

#!/usr/bin/env bash
cp important_file ./backups/
rm important_file

如果備份目錄不存在,會發生什么情況?確切地說,你將在控制臺中收到一條錯誤訊息,但是在你能夠做出反應之前,該檔案已經被第二個命令洗掉,

Get the location

script_dir=$(cd "$(dirname "${BASH_SOURCE[0]}")" &>/dev/null && pwd -P)

這行代碼盡其所能定義腳本的位置目錄,然后我們對其進行cd配置,為什么?

通常,我們的腳本在相對于腳本位置的路徑上運行,復制檔案并執行命令,假設腳本目錄也是一個作業目錄,是的,只要我們從它的目錄執行腳本,

但是,假設我們的CI配置執行腳本如下所示呢:

/opt/ci/project/script.sh

那么我們的腳本不是在專案目錄中操作的,而是在CI工具的一些完全不同的作業目錄中操作的,我們可以通過在執行腳本之前轉到目錄來修復它:

cd /opt/ci/project && ./script.sh

但從腳本的角度解決這個問題要好得多,因此,如果腳本從同一目錄中讀取某個檔案或執行另一個程式,請按如下方式呼叫:

cat "$script_dir/my_file"

同時,腳本不會更改作業目錄的位置,如果腳本是從其他目錄執行的,并且用戶提供了指向某個檔案的相對路徑,我們仍然可以讀取它,

Try to clean up

trap cleanup SIGINT SIGTERM ERR EXIT

cleanup() {
  trap - SIGINT SIGTERM ERR EXIT
  # script cleanup here
}

在腳本結束時,將執行cleanup()函式,你可以在這里嘗試洗掉腳本創建的所有臨時檔案,

請記住,cleanup()不僅可以在最后呼叫,在任何時候都可以,

Display helpful help

usage() {
  cat <<EOF
Usage: $(basename "${BASH_SOURCE[0]}") [-h] [-v] [-f] -p param_value arg1 [arg2...]

Script description here.

...
EOF
  exit
}

盡量讓usage()函式相對靠近腳本的頂部,有兩種作用:

  • 要為不知道所有選項并且不想查看整個腳本來發現這些選項的人顯示幫助,

  • 當有人修改腳本時,保存一個最小的檔案(因為兩周后,你甚至不記得當初是怎么寫的),

我不主張在這里記錄每個函式,但是一個簡短、漂亮的腳本使用這些訊息是必需的,

setup_colors() {
  if [[ -t 2 ]] && [[ -z "${NO_COLOR-}" ]] && [[ "${TERM-}" != "dumb" ]]; then
    NOFORMAT='\033[0m' RED='\033[0;31m' GREEN='\033[0;32m' ORANGE='\033[0;33m' BLUE='\033[0;34m' PURPLE='\033[0;35m' CYAN='\033[0;36m' YELLOW='\033[1;33m'
  else
    NOFORMAT='' RED='' GREEN='' ORANGE='' BLUE='' PURPLE='' CYAN='' YELLOW=''
  fi
}

msg() {
  echo >&2 -e "${1-}"
}

首先,如果你還不想在文本中使用顏色,那么先洗掉setup_colors()函式,我保留它是因為我知道如果我不必每次都用谷歌編碼的話,我會更頻繁地使用顏色,

其次,這些顏色只用于msg()函式,而不是echo命令,

msg()函式用于列印不是腳本輸出的所有內容,這包括所有日志和訊息,而不僅僅是錯誤,參考
12 Factor CLI Apps的文章說法:

In short: stdout is for output, stderr is for messaging.

— Jeff Dickey, who knows a little about building CLI apps

stdout用于輸出,stderr用于訊息傳遞,

這就是為什么在大多數情況下你不應該為stdout使用顏色,

用msg()列印的訊息被發送到stderr流并支持特殊的序列,比如顏色,如果stderr輸出不是互動式終端,或者傳遞了一個標準引數,那么顏色將被禁用,
用法如下:

msg "This is a ${RED}very important${NOFORMAT} message, but not a script output value!"

要檢查stderr是不是互動式終端時的行為,請在腳本中添加類似于上面的一行,然后執行它,將stderr重定向到stdout并通過管道將其發送到cat,管道操作使輸出不再直接發送到終端,而是發送到下一個命令,因此顏色會被禁用,

$ ./test.sh 2>&1 | cat
This is a very important message, but not a script output value!

Parse any parameters

parse_params() {
  # default values of variables set from params
  flag=0
  param=''

  while :; do
    case "${1-}" in
    -h | --help) usage ;;
    -v | --verbose) set -x ;;
    --no-color) NO_COLOR=1 ;;
    -f | --flag) flag=1 ;; # example flag
    -p | --param) # example named parameter
      param="${2-}"
      shift
      ;;
    -?*) die "Unknown option: $1" ;;
    *) break ;;
    esac
    shift
  done

  args=("$@")

  # check required params and arguments
  [[ -z "${param-}" ]] && die "Missing required parameter: param"
  [[ ${#args[@]} -eq 0 ]] && die "Missing script arguments"

  return 0
}

如果在腳本中引數化有意義的話,我就通常就會去做,即使整個腳本只在一個地方使用,它使復制和重用它變得更容易,而這通常是早晚發生的,而且,即使某些東西需要硬編碼,通常在比Bash腳本更高的級別上有更好的位置,

CLI引數有三種主要型別:標志、命名引數和位置引數,parse_params()函式支持所有這些引數,

這里沒有處理的唯一一個公共引數模式是連接多個單字母標志,為了能夠傳遞兩個標志作為-ab,而不是-a-b,需要一些額外的代碼,

while回圈是一種手動決議引數的方法,在其他語言中,您應該使用一個內置的決議器或可用的庫,但是,好吧,這是Bash,

模板中有一個示例標志(-f)和命名引數(-p),只需更改或復制它們以添加其他引數,之后不要忘記更新usage(),

這里最重要的一點是,當您使用第一個google結果進行Bash引數決議時,通常會丟失一個未知選項的錯誤,腳本收到未知選項的事實意味著用戶希望它執行腳本無法完成的操作,所以用戶的期望和腳本行為可能會有很大的不同,最好是在壞事發生之前完全阻止處決,

在Bash中決議引數有兩種選擇,是一個接一個的,有人贊成和反對使用它們,我發現這些工具不是最好的,因為默認情況下,macOS上的getopt行為完全不同,getopts不支持長引數(比如--help),

Using the template

復制粘貼它,就像你在網上找到的大多數代碼一樣,

復制后,只需更改4件事:

  • 包含腳本說明的usage()文本
  • cleanup()內容
  • parse_params()中的引數–保留--help和--no color,但替換示例:-f和-p
  • 實際的腳本邏輯

Portability

我在MacOS上測驗了這個模板(使用默認的bash3.2)和幾個Docker映像:Debian、Ubuntu、CentOS、amazonlinux、Fedora,它的確起作用了,

顯然,它不能在缺少Bash的環境中作業,比如alpinellinux,

Further reading

在用Bash或其他更好的語言創建CLI腳本時,有一些通用規則,這些資源將指導您如何使小型腳本和大型CLI應用程式可靠,參考如下:

  • Command Line Interface Guidelines(https://clig.dev/)
  • 12 Factor CLI Apps(https://medium.com/@jdxcode/12-factor-cli-apps-dd3c227a0e46)
  • Command line arguments anatomy explained with examples(https://betterdev.blog/command-line-arguments-anatomy-explained/)

Closing notes

我不會是第一個也不是最后一個創建Bash腳本模板的人,這個專案是一個很好的選擇,雖然對我的日常需求來說有點太大了,畢竟,我盡量使Bash腳本盡可能小(而且很少使用),

撰寫Bash腳本時,請使用支持ShellCheck linter的IDE,如JetBrains IDEs,它會阻止你做一堆適得其反的事情,

本文首發:http://blog.didispace.com/minimal-safe-bash-script-template/

歡迎關注我的公眾號:程式猿DD,獲得獨家整理的免費學習資源助力你的Java學習之路!另每周贈書不停哦~

轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/241205.html

標籤:Java

上一篇:idea啟動build過慢

下一篇:螞蟻開源的 SOFABoot,和 Spring Boot 有啥關系?

標籤雲
其他(157675) Python(38076) JavaScript(25376) Java(17977) C(15215) 區塊鏈(8255) C#(7972) AI(7469) 爪哇(7425) MySQL(7132) html(6777) 基礎類(6313) sql(6102) 熊猫(6058) PHP(5869) 数组(5741) R(5409) Linux(5327) 反应(5209) 腳本語言(PerlPython)(5129) 非技術區(4971) Android(4554) 数据框(4311) css(4259) 节点.js(4032) C語言(3288) json(3245) 列表(3129) 扑(3119) C++語言(3117) 安卓(2998) 打字稿(2995) VBA(2789) Java相關(2746) 疑難問題(2699) 细绳(2522) 單片機工控(2479) iOS(2429) ASP.NET(2402) MongoDB(2323) 麻木的(2285) 正则表达式(2254) 字典(2211) 循环(2198) 迅速(2185) 擅长(2169) 镖(2155) 功能(1967) .NET技术(1958) Web開發(1951) python-3.x(1918) HtmlCss(1915) 弹簧靴(1913) C++(1909) xml(1889) PostgreSQL(1872) .NETCore(1853) 谷歌表格(1846) Unity3D(1843) for循环(1842)

熱門瀏覽
  • 【C++】Microsoft C++、C 和匯編程式檔案

    ......

    uj5u.com 2020-09-10 00:57:23 more
  • 例外宣告

    相比于斷言適用于排除邏輯上不可能存在的狀態,例外通常是用于邏輯上可能發生的錯誤。 例外宣告 Item 1:當函式不可能拋出例外或不能接受拋出例外時,使用noexcept 理由 如果不打算拋出例外的話,程式就會認為無法處理這種錯誤,并且應當盡早終止,如此可以有效地阻止例外的傳播與擴散。 示例 //不可 ......

    uj5u.com 2020-09-10 00:57:27 more
  • Codeforces 1400E Clear the Multiset(貪心 + 分治)

    鏈接:https://codeforces.com/problemset/problem/1400/E 來源:Codeforces 思路:給你一個陣列,現在你可以進行兩種操作,操作1:將一段沒有 0 的區間進行減一的操作,操作2:將 i 位置上的元素歸零。最終問:將這個陣列的全部元素歸零后操作的最少 ......

    uj5u.com 2020-09-10 00:57:30 more
  • UVA11610 【Reverse Prime】

    本人看到此題沒有翻譯,就附帶了一個自己的翻譯版本 思考 這一題,它的第一個要求是找出所有 $7$ 位反向質數及其質因數的個數。 我們應該需要質數篩篩選1~$10^{7}$的所有數,這里就不慢慢介紹了。但是,重讀題,我們突然發現反向質數都是 $7$ 位,而將它反過來后的數字卻是 $6$ 位數,這就說明 ......

    uj5u.com 2020-09-10 00:57:36 more
  • 統計區間素數數量

    1 #pragma GCC optimize(2) 2 #include <bits/stdc++.h> 3 using namespace std; 4 bool isprime[1000000010]; 5 vector<int> prime; 6 inline int getlist(int ......

    uj5u.com 2020-09-10 00:57:47 more
  • C/C++編程筆記:C++中的 const 變數詳解,教你正確認識const用法

    1、C中的const 1、區域const變數存放在堆疊區中,會分配記憶體(也就是說可以通過地址間接修改變數的值)。測驗代碼如下: 運行結果: 2、全域const變數存放在只讀資料段(不能通過地址修改,會發生寫入錯誤), 默認為外部聯編,可以給其他源檔案使用(需要用extern關鍵字修飾) 運行結果: ......

    uj5u.com 2020-09-10 00:58:04 more
  • 【C++犯錯記錄】VS2019 MFC添加資源不懂如何修改資源宏ID

    1. 首先在資源視圖中,添加資源 2. 點擊新添加的資源,復制自動生成的ID 3. 在解決方案資源管理器中找到Resource.h檔案,編輯,使用整個專案搜索和替換的方式快速替換 宏宣告 4. Ctrl+Shift+F 全域搜索,點擊查找全部,然后逐個替換 5. 為什么使用搜索替換而不使用屬性視窗直 ......

    uj5u.com 2020-09-10 00:59:11 more
  • 【C++犯錯記錄】VS2019 MFC不懂的批量添加資源

    1. 打開資源頭檔案Resource.h,在其中預先定義好宏 ID(不清楚其實ID值應該設定多少,可以先新建一個相同的資源項,再在這個資源的ID值的基礎上遞增即可) 2. 在資源視圖中選中專案資源,按F7編輯資源檔案,按 ID 型別 相對路徑的形式添加 資源。(別忘了先把檔案拷貝到專案中的res檔案 ......

    uj5u.com 2020-09-10 01:00:19 more
  • C/C++編程筆記:關于C++的參考型別,專供新手入門使用

    今天要講的是C++中我最喜歡的一個用法——參考,也叫別名。 參考就是給一個變數名取一個變數名,方便我們間接地使用這個變數。我們可以給一個變數創建N個參考,這N + 1個變數共享了同一塊記憶體區域。(參考型別的變數會占用記憶體空間,占用的記憶體空間的大小和指標型別的大小是相同的。雖然參考是一個物件的別名,但 ......

    uj5u.com 2020-09-10 01:00:22 more
  • 【C/C++編程筆記】從頭開始學習C ++:初學者完整指南

    眾所周知,C ++的學習曲線陡峭,但是花時間學習這種語言將為您的職業帶來奇跡,并使您與其他開發人員區分開。您會更輕松地學習新語言,形成真正的解決問題的技能,并在編程的基礎上打下堅實的基礎。 C ++將幫助您養成良好的編程習慣(即清晰一致的編碼風格,在撰寫代碼時注釋代碼,并限制類內部的可見性),并且由 ......

    uj5u.com 2020-09-10 01:00:41 more
最新发布
  • Rust中的智能指標:Box<T> Rc<T> Arc<T> Cell<T> RefCell<T> Weak

    Rust中的智能指標是什么 智能指標(smart pointers)是一類資料結構,是擁有資料所有權和額外功能的指標。是指標的進一步發展 指標(pointer)是一個包含記憶體地址的變數的通用概念。這個地址參考,或 ” 指向”(points at)一些其 他資料 。參考以 & 符號為標志并借用了他們所 ......

    uj5u.com 2023-04-20 07:24:10 more
  • Java的值傳遞和參考傳遞

    值傳遞不會改變本身,參考傳遞(如果傳遞的值需要實體化到堆里)如果發生修改了會改變本身。 1.基本資料型別都是值傳遞 package com.example.basic; public class Test { public static void main(String[] args) { int ......

    uj5u.com 2023-04-20 07:24:04 more
  • [2]SpinalHDL教程——Scala簡單入門

    第一個 Scala 程式 shell里面輸入 $ scala scala> 1 + 1 res0: Int = 2 scala> println("Hello World!") Hello World! 檔案形式 object HelloWorld { /* 這是我的第一個 Scala 程式 * 以 ......

    uj5u.com 2023-04-20 07:23:58 more
  • 理解函式指標和回呼函式

    理解 函式指標 指向函式的指標。比如: 理解函式指標的偽代碼 void (*p)(int type, char *data); // 定義一個函式指標p void func(int type, char *data); // 宣告一個函式func p = func; // 將指標p指向函式func ......

    uj5u.com 2023-04-20 07:23:52 more
  • Django筆記二十五之資料庫函式之日期函式

    本文首發于公眾號:Hunter后端 原文鏈接:Django筆記二十五之資料庫函式之日期函式 日期函式主要介紹兩個大類,Extract() 和 Trunc() Extract() 函式作用是提取日期,比如我們可以提取一個日期欄位的年份,月份,日等資料 Trunc() 的作用則是截取,比如 2022-0 ......

    uj5u.com 2023-04-20 07:23:45 more
  • 一天吃透JVM面試八股文

    什么是JVM? JVM,全稱Java Virtual Machine(Java虛擬機),是通過在實際的計算機上仿真模擬各種計算機功能來實作的。由一套位元組碼指令集、一組暫存器、一個堆疊、一個垃圾回收堆和一個存盤方法域等組成。JVM屏蔽了與作業系統平臺相關的資訊,使得Java程式只需要生成在Java虛擬機 ......

    uj5u.com 2023-04-20 07:23:31 more
  • 使用Java接入小程式訂閱訊息!

    更新完微信服務號的模板訊息之后,我又趕緊把微信小程式的訂閱訊息給實作了!之前我一直以為微信小程式也是要企業才能申請,沒想到小程式個人就能申請。 訊息推送平臺🔥推送下發【郵件】【短信】【微信服務號】【微信小程式】【企業微信】【釘釘】等訊息型別。 https://gitee.com/zhongfuch ......

    uj5u.com 2023-04-20 07:22:59 more
  • java -- 緩沖流、轉換流、序列化流

    緩沖流 緩沖流, 也叫高效流, 按照資料型別分類: 位元組緩沖流:BufferedInputStream,BufferedOutputStream 字符緩沖流:BufferedReader,BufferedWriter 緩沖流的基本原理,是在創建流物件時,會創建一個內置的默認大小的緩沖區陣列,通過緩沖 ......

    uj5u.com 2023-04-20 07:22:49 more
  • Java-SpringBoot-Range請求頭設定實作視頻分段傳輸

    老實說,人太懶了,現在基本都不喜歡寫筆記了,但是網上有關Range請求頭的文章都太水了 下面是抄的一段StackOverflow的代碼...自己大修改過的,寫的注釋挺全的,應該直接看得懂,就不解釋了 寫的不好...只是希望能給視頻網站開發的新手一點點幫助吧. 業務場景:視頻分段傳輸、視頻多段傳輸(理 ......

    uj5u.com 2023-04-20 07:22:42 more
  • Windows 10開發教程_編程入門自學教程_菜鳥教程-免費教程分享

    教程簡介 Windows 10開發入門教程 - 從簡單的步驟了解Windows 10開發,從基本到高級概念,包括簡介,UWP,第一個應用程式,商店,XAML控制元件,資料系結,XAML性能,自適應設計,自適應UI,自適應代碼,檔案管理,SQLite資料庫,應用程式到應用程式通信,應用程式本地化,應用程式 ......

    uj5u.com 2023-04-20 07:22:35 more