ceph/qa/workunits/ceph-helpers-root.sh

   1 #!/usr/bin/env bash
   2 #
   3 # Copyright (C) 2015 Red Hat <contact@redhat.com>
   4 #
   5 # Author: Loic Dachary <loic@dachary.org>
   6 #
   7 # This program is free software; you can redistribute it and/or modify
   8 # it under the terms of the GNU Library Public License as published by
   9 # the Free Software Foundation; either version 2, or (at your option)
  10 # any later version.
  11 #
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU Library Public License for more details.
  16 #
  17
  18 #######################################################################
  19
  20 function distro_id() {
  21     source /etc/os-release
  22     echo $ID
  23 }
  24
  25 function distro_version() {
  26     source /etc/os-release
  27     echo $VERSION
  28 }
  29
  30 function install() {
  31     for package in "$@" ; do
  32         install_one $package
  33     done
  34 }
  35
  36 function install_one() {
  37     case $(distro_id) in
  38         ubuntu|debian|devuan|softiron)
  39             sudo env DEBIAN_FRONTEND=noninteractive apt-get install -y "$@"
  40             ;;
  41         centos|fedora|rhel)
  42             sudo yum install -y "$@"
  43             ;;
  44         opensuse*|suse|sles)
  45             sudo zypper --non-interactive install "$@"
  46             ;;
  47         *)
  48             echo "$(distro_id) is unknown, $@ will have to be installed manually."
  49             ;;
  50     esac
  51 }
  52
  53 function install_pkg_on_ubuntu {
  54     local project=$1
  55     shift
  56     local sha1=$1
  57     shift
  58     local codename=$1
  59     shift
  60     local force=$1
  61     shift
  62     local pkgs=$@
  63     local missing_pkgs
  64     if [ $force = "force" ]; then
  65         missing_pkgs="$@"
  66     else
  67         for pkg in $pkgs; do
  68             if ! dpkg -s $pkg &> /dev/null; then
  69                 missing_pkgs+=" $pkg"
  70             fi
  71         done
  72     fi
  73     if test -n "$missing_pkgs"; then
  74         local shaman_url="https://shaman.ceph.com/api/repos/${project}/master/${sha1}/ubuntu/${codename}/repo"
  75         sudo curl --silent --location $shaman_url --output /etc/apt/sources.list.d/$project.list
  76         sudo env DEBIAN_FRONTEND=noninteractive apt-get update -y -o Acquire::Languages=none -o Acquire::Translation=none || true
  77         sudo env DEBIAN_FRONTEND=noninteractive apt-get install --allow-unauthenticated -y $missing_pkgs
  78     fi
  79 }
  80
  81 #######################################################################
  82
  83 function control_osd() {
  84     local action=$1
  85     local id=$2
  86
  87     sudo systemctl $action ceph-osd@$id
  88
  89     return 0
  90 }
  91
  92 #######################################################################
  93
  94 function pool_read_write() {
  95     local size=${1:-1}
  96     local dir=/tmp
  97     local timeout=360
  98     local test_pool=test_pool
  99
 100     ceph osd pool delete $test_pool $test_pool --yes-i-really-really-mean-it || return 1
 101     ceph osd pool create $test_pool 4 || return 1
 102     ceph osd pool set $test_pool size $size --yes-i-really-mean-it || return 1
 103     ceph osd pool set $test_pool min_size $size || return 1
 104     ceph osd pool application enable $test_pool rados
 105
 106     echo FOO > $dir/BAR
 107     timeout $timeout rados --pool $test_pool put BAR $dir/BAR || return 1
 108     timeout $timeout rados --pool $test_pool get BAR $dir/BAR.copy || return 1
 109     diff $dir/BAR $dir/BAR.copy || return 1
 110     ceph osd pool delete $test_pool $test_pool --yes-i-really-really-mean-it || return 1
 111 }
 112
 113 #######################################################################
 114
 115 set -x
 116
 117 "$@"