Welcome! In this lesson, we will explore the powerful trap command in Bash. The trap command executes a set of commands when a script exits, whether it completes successfully or encounters an error. This command is crucial for managing cleanup operations in your scripts. Proper cleanup ensures that temporary files and other resources do not clutter your system or cause unexpected behaviors.

Let's get started!

The trap command in Bash allows you to specify commands that should be executed when a certain signal is received by the script. Signals are notifications sent to a process to notify it of various events. For example, the EXIT signal is sent when the script exits, either normally or due to an error. The INT (interrupt) signal triggers when you press Ctrl+C to interrupt the script.

The general syntax for the trap command in Bash is as follows:

This line instructs the script to run commands when the specified signal is received. commands can be a string of commands, another script or a function.

The following script creates a file called temp.txt then exits. We use the trap command to remove this temporary file once the script exits.

solution.sh

In this script:

• The cleanup function removes temporary files and prints ls output before and after rm -f temp.txt.
• trap cleanup EXIT: Sets up a trap to call the cleanup function on script exit.
• touch temp.txt: Creates an empty file named temp.txt.
• Two echo statements simulate work with the temporary file.
• When exit 0 is executed, the trap command ensures the cleanup function is called to remove the temporary file.

The output of the script is:

The output of the script shows that when the exit 0 command is executed, the cleanup function is called, successfully removing the temp.txt. The trap command ensures that the cleanup function is executed regardless of how the script exits, providing a robust way to manage resources.

The trap command can handle various signals, not just EXIT. For instance, the INT signal is sent when you press Ctrl+C to interrupt a script. Let’s modify our script to handle the INT signal as well. This script sets up two cleanup functions: one for handling the EXIT signal and another for the INT signal. When the script exits normally, the cleanup_exit function is invoked. If the user interrupts the script by pressing Ctrl+C, the cleanup_int function prints a message indicating that the script was interrupted. However, it does not remove the temporary file.

In the script:

• cleanup_exit defines a function that prints a descriptive message and removes temp.txt
• cleanup_int defines a function that prints a descriptive message but does not remove temp.txt
• trap cleanup_exit EXIT instructs the script to call cleanup_exit if the script receives an EXIT signal
• trap cleanup_int INT instructs the script to call cleanup_int if the script receives an INT signal
• exit 0 exits the script, sending an EXIT signal

If the script runs without the user pressing Ctrl+C, the output is:

The temp.txt no longer exists in the current working directory.

If the user pressed Ctrl+C, the output is:

temp.txt will still exist in the current working directory.

We can also specify what commands to run based on the exit status. This script sets up two cleanup functions to handle success (cleanup_success) and failure (cleanup_failure), and a general cleanup function (cleanup) to dispatch the appropriate cleanup based on the exit code. It attempts to read a file file.txt. Depending on if cat file.txt succeeds or fails, the script exits with the appropriate status, triggering the relevant cleanup function.

• The cleanup_success function prints a success message then calls rm -f fail.txt to remove fail.txt

• The cleanup_failure function prints a failure message then calls rm -f success.txt to remove success.txt

• The cleanup function checks the script's exit status ($?) and calls either cleanup_success or cleanup_failure based on whether the script exited successfully or with an error.

• trap cleanup EXIT sets up a trap to call the cleanup function when the script exits.

• The cleanup function checks the script's exit status ($?) and calls either cleanup_success or cleanup_failure based on whether the script exited successfully or with an error.

• if [ $? -eq 0 ] checks whether the exit status of the most recent command ($?) equals 0.

• If successful, the script executes exit 0 and calls the cleanup function which calls cleanup_success

• If not successful (non-zero exit status), the script executes exit 1. This calls the cleanup function which calls the cleanup_failure function

Now let's look at our command that may or may not error.

• If touch file.txt is not commented out, cat file.txt will not error and the $? special variable will have a value of 0.
• If touch file.txt is commented out, cat file.txt will error $? will have a non-zero value.

If the script includes the touch file.txt command the output is:

If the script does not include touch file.txt, the output is:

Great job! In this lesson, you learned how to:

1. Define cleanup functions to manage resources.
2. Use the trap command to execute cleanup functions on script exit.
3. Handle various signals like EXIT and INT with specific cleanup commands.
4. Manage multiple exit statuses using the trap command to ensure proper cleanup for successful and failed script executions.

Now it's time to put your new knowledge into practice! Head over to the practice section to try these concepts on your own. By practicing, you'll gain confidence and become adept at writing scripts that handle resources efficiently. Happy scripting!