Shell Script – Funções

FuncoesShellScriptLinuxcleitonbueno.wordpress.com

Assim como em qualquer outra linguagem, no Shell Script também é possível utilizar funções, mas com suas particularidades e veremos como criar uma função, chamar uma função, passar parâmetros para ela e pegar o retorno da mesma.

Primeiro vamos criar nosso primeiro script, que será chamado de myFunc1.sh e criar nossa primeira função.

bueno@vm010:~/ShellScript$ touch myFunc1.sh
bueno@vm010:~/ShellScript$ chmod +x myFunc1.sh

Código:

#! /bin/bash

function VersaoLinux () {
    /usr/bin/lsb_release -ds
}

echo "Iniciando script, aguarde..."

VersaoLinux

Executando:

bueno@vm010:~/ShellScript$ ./myFunc1.sh
Iniciando script, aguarde...
Linux Mint 15 Olivia

Tranquilo né? Quando for criar uma função, deve-se colocar function e em seguida o nome da função, como na linha 3 e dentro o código que deseja executar, para chamar a função é só digitar o nome dela em qualquer parte do script, como é feito com VersaoLinux na linha 9.

Vamos agregar nosso script, só que agora ele irá verificar se o usuário que esta executando é root ou esta executando com permissão de super-usuário, caso sim, continua sem problemas, do contrario encerra o script.

#! /bin/bash

function VersaoLinux () {
   /usr/bin/lsb_release -ds
}

CheckUser () {
   USER_ID=$(/usr/bin/id -u)
   return $USER_ID
}

echo "Iniciando script, aguarde..."

CheckUser
if [ $? -ne "0" ]; then
   echo -e "nVoce não é root, execute como super-usuario!n"
   exit 1
fi

# Macete para imprimir saida de funcao dentro do echo
echo -e "SO:t$(VersaoLinux)"

Executando o script:

bueno@vm010:~/ShellScript$ ./myFunc1.sh
Iniciando script, aguarde...

Voce não é root, execute como super-usuario!

bueno@vm010:~/ShellScript$ sudo ./myFunc1.sh
Iniciando script, aguarde...
SO: Linux Mint 15 Olivia
bueno@vm010:~/ShellScript$

Criei uma função CheckUser(), como pode ver também é possível criar sem o nome function antes ;), onde usando o comando id, obtenho o numero do ID do usuário e no final da função uso return para devolve este resultado para meu script, para pegar o valor de retorno, logo após chamar o script CheckUser() na linha 14 eu coloco um if com $? que obtenho o valor do ultimo comando executado, então se este valor for -ne (diferente) de zero, o usuário não é root.

Se esta parte de if, -ne e $? é novidade pra você, visite este Shell Script – Estrutura Condicional e este outro Shell Script – Variáveis que foram dois post que escrevi que aborda toda esta parte.

Na linha 21 eu dou um echo com a função VersaoLinux(), onde para anexar junto ao echo() tem o macete de usar uma subshell como eu fiz.
E como vimos na saída do script com usuário comum ele saiu e como root ou permissão de super-usuário ele prosseguiu, e o mais legal, pegamos o retorno da função CheckUser().

Dando continuidade, agora vamos passar alguns parâmetros para nossa função e ver como realizar e tratar esse cara, e construir um script maior e mais completo.

#! /bin/bash

function VersaoLinux () {
   /usr/bin/lsb_release -ds
}

function CheckUser () {
   USER_ID=$(/usr/bin/id -u)
   return $USER_ID
}

function CheckGCC () {
   /usr/bin/gcc -v
}

function CheckPython () {
   /usr/bin/python --version
}

function CheckPerl () {
   /usr/bin/perl -v
}

function CheckSystem () {

   # Macete para imprimir saida de funcao dentro do echo
   echo -e "nn###\033[0;31m Distrubuição:\033[0mt$(VersaoLinux)"
   #echo -e "### Distribuição:t$(VersaoLinux)"

   echo -e "nn###\033[0;31m Processador\033[0m"
   cat /proc/cpuinfo | grep "model name"

   echo -e "nn###\033[0;31m Memoria\033[0m"
   /usr/bin/free -m

   echo -e "nn###\033[0;31m Disco\033[0m"
   /bin/df -h
}

function CheckOptions () {
   if [ $1 = "python" ]; then
      CheckPython
   elif [ $1 = "perl" ]; then
      CheckPerl
   elif [ $1 = "gcc" ]; then
      CheckGCC
   elif [ $1 = "system" ]; then
      CheckSystem
   elif [ $1 = "all" ]; then
      CheckSystem
      CheckGCC
      CheckPerl
      CheckPython
   else
      echo -e "$1 é uma opção invalida!"
      exit 1
   fi

}

echo "Iniciando script, aguarde..."

CheckUser
if [ $? -ne "0" ]; then
   echo -e "nVoce não é root, execute como super-usuario!n"
   exit 1
fi

while true;
do
   echo -e "### Voce deve digitar um dos parametros abaixo para obter informacoes: "
   echo -e "tpython"
   echo -e "tperl"
   echo -e "tgcc"
   echo -e "tsystem"
   echo -e "talln"
   read -p "### Opcao: " op

   CheckOptions $op

   read -p "### Continuar? (S/N) " cont

   if [ $cont = "N" ] || [ $cont = "n" ]; then
      echo -e "Saindo..."
      exit 1
   fi

   /usr/bin/clear

done

Executando o script:

bueno@vm010:~/ShellScript$ sudo ./myFunc1.sh
Iniciando script, aguarde...
### Voce deve digitar um dos parâmetros abaixo para obter informações:
python
perl
gcc
system
all

### Opcao: python
Python 2.7.1+
### Continuar? (S/N) s
### Você deve digitar um dos parâmetros abaixo para obter informações:
python
perl
gcc
system
all

### Opcao: gcc
Using built-in specs.
COLLECT_GCC=/usr/bin/gcc
COLLECT_LTO_WRAPPER=/usr/lib/i386-linux-gnu/gcc/i686-linux-gnu/4.5.2/lto-wrapper
Target: i686-linux-gnu
Configured with: ../src/configure -v --with-pkgversion='Ubuntu/Linaro 4.5.2-8ubuntu4' --with-bugurl=file:///usr/share/doc/gcc-4.5/README.Bugs --enable-languages=c,c++,fortran,objc,obj-c++ --prefix=/usr --program-suffix=-4.5 --enable-shared --enable-multiarch --with-multiarch-defaults=i386-linux-gnu --enable-linker-build-id --with-system-zlib --libexecdir=/usr/lib/i386-linux-gnu --without-included-gettext --enable-threads=posix --with-gxx-include-dir=/usr/include/c++/4.5 --libdir=/usr/lib/i386-linux-gnu --enable-nls --with-sysroot=/ --enable-clocale=gnu --enable-libstdcxx-debug --enable-libstdcxx-time=yes --enable-plugin --enable-gold --enable-ld=default --with-plugin-ld=ld.gold --enable-objc-gc --enable-targets=all --disable-werror --with-arch-32=i686 --with-tune=generic --enable-checking=release --build=i686-linux-gnu --host=i686-linux-gnu --target=i686-linux-gnu
Thread model: posix
gcc version 4.5.2 (Ubuntu/Linaro 4.5.2-8ubuntu4)
### Continuar? (S/N) n
Saindo...
bueno@vm010:~/ShellScript$

Usando todo o código dos scripts anteriores eu criei algumas funções com tarefas especificas que desprezam comentários, o foco é a função CheckOptions() que quando passamos algum parâmetro para alguma função, carregamos com $1 para o primeiro parâmetro $2 para o segundo e assim por diante, o parâmetro eu passei na linha 84 CheckOptions $op, que é obtido na linha 82 com o read.

Tudo o que foi implementado neste ultimo script pode ser encontrado no blog e nas referências no final.

Agora você pode treinar um pouco, baseado no que escrevi até agora, poderia nas funções CheckPerl(), CheckPython(), CheckGCC() verificar se tal binário existe, ou se o caminho esta correto e deixar mais completo e funcional.

Espero que tenham gostado e até a próxima!

 

Referências

https://cleitonbueno.wordpress.com/2014/03/10/shell-script-estrutura-de-repeticao-e-arrays/
https://cleitonbueno.wordpress.com/2014/03/31/shell-script-read/
https://cleitonbueno.wordpress.com/2014/02/17/shell-script-estrutura-condicional/
https://cleitonbueno.wordpress.com/2014/02/03/shell-script-variaveis/

Share Button

CC BY-NC-SA 4.0 Shell Script – Funções by Cleiton Bueno is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.