Shell编程声名显赫
—— Larry Wall
本章目录
一个最简单的脚本其实就是将一连串系统命令存储在一个文件中。最起码,它能帮你省下重复输入这一连串命令的功夫。
样例 2-1. cleanup:清理/var/log
目录下的日志文件
# Cleanup
# 请使用root权限执行
cd /var/log
cat /dev/null > messages
cat /dev/null > wtmp
echo "Log files cleaned up."
这支脚本仅仅是一些可以很容易从终端或控制台输入的命令的集合罢了,没什么特殊的地方。将命令放在脚本中的好处是,你不用再一遍遍重复输入这些命令啦。脚本成了一支程序、一款工具,它可以很容易的被修改或为特殊需求定制。
样例 2-2. cleanup:改进的清理脚本
#!/bin/bash
# Bash脚本标准起始行。
# Cleanup, version 2
# 请使用root权限执行。
# 这里可以插入代码来打印错误信息,并在未使用root权限时退出。
LOG_DIR=/var/log
# 使用变量比硬编码(hard-coded)更合适
cd $LOG_DIR
cat /dev/null > messages
cat /dev/null > wtmp
echo "Logs cleaned up."
exit # 正确终止脚本的方式。
# 不带参数的exit返回上一条指令的运行结果。
现在我们看到了一个真正意义上的脚本! 让我们继续前进...
样例 2-3. cleanup:改良、通用版
#!/bin/bash
# Cleanup, version 3
# 注意:
# --------
# 此脚本涉及到许多后边才会解释的特性。
# 当你阅读完整本书的一半以后,理解它们就没有任何困难了。
LOG_DIR=/var/log
ROOT_UID=0 # UID为0的用户才拥有root权限。
LINES=50 # 默认保存messages日志文件行数。
E_XCD=86 # 无法切换工作目录的错误码。
E_NOTROOT=87 # 非root权限用户执行的错误码。
# 请使用root权限运行。
if [ "$UID" -ne "$ROOT_UID" ]
then
echo "Must be root to run this script."
exit $E_NOTROOT
fi
if [ -n "$1" ]
# 测试命令行参数(保存行数)是否为空
then
lines=$1
else
lines=$LINES # 如果为空则使用默认设置
fi
# Stephane Chazelas 建议使用如下方法检查命令行参数,
# 但是这已经超出了此阶段教程的范围。
#
# E_WRONGARGS=85 # Non-numerical argument (bad argument format).
# case "$1" in
# "" ) lines=50;;
# *[!0-9]*) echo "Usage: `basename $0` lines-to-cleanup";
# exit $E_WRONGARGS;;
# * ) lines=$1;;
# esac
#
#* 在第十一章“循环与分支”中会对此作详细的阐述。
cd $LOG_DIR
if [ `pwd` != "$LOG_DIR" ] # 也可以这样写 if [ "$PWD" != "$LOG_DIR" ]
# 检查工作目录是否为 /var/log ?
then
echo "Can't change to $LOG_DIR"
exit $E_XCD
fi # 在清理日志前,二次确认是否在正确的工作目录下。
# 更高效的写法:
#
# cd /var/log || {
# echo "Cannot change to necessary directory." >&2
# exit $E_XCD;
# }
tail -n $lines messages > mesg.temp # 保存messages日志文件最后一部分
mv mesg.temp messages # 替换系统日志文件以达到清理目的
# cat /dev/null > messages
#* 我们不需要使用这个方法了,上面的方法更安全
cat /dev/null > wtmp # ': > wtmp' 与 '> wtmp' 有同样的效果
echo "Log files cleaned up."
# 注意在/var/log目录下的其他日志文件不会被这个脚本清除
exit 0
# 返回0表示脚本运行成功
也许你并不希望清空全部的系统日志,这个脚本保留了messages日志的最后一部分。随着学习的深入,你将明白更多提高脚本运行效率的方法。
脚本起始行sha-bang(#!)告诉系统这个脚本文件需要使用指定的命令解释器来执行。#!实际上是一个占两字节的幻数(magic number),幻数可以用来标识特殊的文件类型,在这里则是标记可执行shell脚本(你可以在终端中输入man magic
了解更多信息)。紧随#!的是一个路径名。此路径指向用来解释此脚本的程序,它可以是shell,可以是程序设计语言,也可以是实用程序。这个解释器从头(#!的下一行)开始执行整个脚本的命令,同时忽略注释。
#!/bin/sh
#!/bin/bash
#!/usr/bin/perl
#!/usr/bin/tcl
#!/bin/sed -f
#!/bin/awk -f
上面每一条脚本起始行都调用了不同的解释器,比如/bin/sh
调用了系统默认shell(Linux系统中默认是bash)。大部分UNIX商业发行版中默认的是Bourne shell,即#!/bin/sh
。你可以以牺牲Bash特性为代价,在非Linux的机器上运行sh脚本。当然,脚本得遵循POSIX sh标准。
需要注意的是#!
后的路径必须正确,否则当你运行脚本时只会得到一条错误信息,通常是"Command not found."
当脚本仅包含一些通用的系统命令而不使用shell内部指令时,可以省略#!
。第三个例子需要#!
是因为当对变量赋值时,例如lines=50
,使用了与shell特性相关的结构。再重复一次,#!/bin/sh
调用的是系统默认shell解释器,在Linux系统中默认为/bin/bash
。
这个例子鼓励读者使用模块化的方式编写脚本,并在平时记录和收集一些在以后可能会用到的代码模板。最终你将拥有一个相当丰富易用的代码库。以下的代码可以用来测试脚本被调用时的参数数量是否正确。
E_WRONG_ARGS=85
script_parameters="-a -h -m -z"
# -a = all, -h = help 等等
if [ $# -ne $Number_of_expected_args ]
then
echo "Usage: `basename $0` $script_parameters"
# `basename $0` 是脚本的文件名
exit $E_WRONG_ARGS
fi
大多数情况下,你会针对特定的任务编写脚本。本章的第一个脚本就是这样。然后你也许会泛化(generalize)脚本使其能够适应更多相似的任务,比如用变量代替硬编码,用函数代替重复代码。