components/osol/ddu/srcs/usr/ddu/scripts/pkg_relate.sh

   1 #!/bin/sh
   2 #
   3 # CDDL HEADER START
   4 #
   5 # The contents of this file are subject to the terms of the
   6 # Common Development and Distribution License (the "License").
   7 # You may not use this file except in compliance with the License.
   8 #
   9 # You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
  10 # or http://www.opensolaris.org/os/licensing.
  11 # See the License for the specific language governing permissions
  12 # and limitations under the License.
  13 #
  14 # When distributing Covered Code, include this CDDL HEADER in each
  15 # file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  16 # If applicable, add the following below this CDDL HEADER, with the
  17 # fields enclosed by brackets "[]" replaced with your own identifying
  18 # information: Portions Copyright [yyyy] [name of copyright owner]
  19 #
  20 # CDDL HEADER END
  21 #
  22 #
  23 # Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
  24 #
  25 # Description: Perform pkg related function:
  26 #              list repo, add repo, search repo and modify
  27 #              repo etc.
  28 #
  29 #              Usage:
  30 #              pkg_relate.sh $1 $2
  31 #              $1 can be "add","list","search" and "modify"
  32 #              $2 is the sub-argument for $1
  33 #              $3 is the sub-argument for $1
  34 #
  35
  36 builtin chmod
  37 builtin cp
  38 builtin mkdir
  39 builtin mv
  40 builtin rm
  41 builtin cat
  42 builtin printf
  43
  44 #
  45 # Get all publishers info on the system
  46 #
  47 function fetch_repo
  48 {
  49     pfexec pkg publisher -Ha > /tmp/repo_temp_store 2>>$err_log
  50     if [ $? -ne 0 ]; then
  51         print -u2 "$0: Error found when getting repo list"
  52         exit 1
  53     fi
  54     repo_list=$(cat /tmp/repo_temp_store | cut -d " " -f 1 2>/dev/null)
  55     repo_url=$(cat /tmp/repo_temp_store | nawk -F"http://" '{ print $2 }' \
  56         2>/dev/null)
  57     repo_count=$(cat /tmp/repo_temp_store | wc -l 2>/dev/null)
  58 }
  59
  60 #
  61 # Output the publisher info in the following format:
  62 # opensolaris.org http://ipkg.sfbay.sun.com/dev/
  63 #
  64
  65 function print_repo
  66 {
  67     index=1
  68     while [ $index -le ${repo_count} ]; do
  69         repo_listi=$(echo ${repo_list} | nawk '{ print $'${index}' }')
  70         repo_urli=$(echo ${repo_url} | nawk '{ print $'${index}' }')
  71         printf "${repo_listi}\thttp://${repo_urli}\n"
  72         index=$(($index + 1))
  73     done
  74 }
  75
  76 #
  77 # Output the publisher info with following format based on the input
  78 # parameter:
  79 # opensolaris.org http://ipkg.sfbay.sun.com/dev/
  80 #
  81 # Input paramenters can be:
  82 # all:            list all the publisher info.
  83 # default:        list preferred publisher info.
  84 # publisher name: list specified pulibsher info.
  85 #
  86
  87 function list
  88 {
  89     if [ ${min_argu_1} == "all" ]; then #list all repo.
  90         fetch_repo
  91         print_repo
  92     elif [ ${min_argu_1} == "default" ]; then #list default repo
  93         pfexec pkg publisher -HP > /tmp/repo_default_store 2>>$err_log
  94         repo_list=$(cat /tmp/repo_default_store | cut -d " " -f 1 2>/dev/null)
  95         repo_url=$(cat /tmp/repo_default_store | nawk -F"http://" '{ print $2 }' \
  96             2>/dev/null)
  97         repo_count=$(cat /tmp/repo_default_store | wc -l 2>/dev/null)
  98         print_repo
  99     elif [ ! -z ${min_argu_1} ]; then #list a specific repo
 100         repo_name=${min_argu_1}
 101         pfexec pkg publisher ${repo_name} | grep "URI" \
 102             > /tmp/repo_spe_store 2>>$err_log
 103         if [  $? != 0 ]; then
 104             print -u2 "$0: ${repo_name} is not a valid repo name!"
 105             exit 1
 106         fi
 107         repo_url=$(cat /tmp/repo_spe_store | nawk -F"http://" '{ print $2 }' \
 108             2>/dev/null)
 109         for repo_urli in ${repo_url}
 110         do
 111             repo_url="http://"${repo_urli}
 112             printf "${repo_name}\t${repo_url}\n"
 113         done
 114     fi
 115 }
 116
 117 #
 118 # Search for driver package via pulisher based on the device compatible string
 119 #
 120 function search
 121 {
 122     if [ -z ${min_argu_2} ]; then
 123         pakcage_found=0
 124         fetch_repo
 125
 126         index=1
 127         while [ $index -le ${repo_count} ]; do
 128             repo_listi=$(echo ${repo_list} | nawk '{ print $'${index}' }')
 129             repo_urli=$(echo ${repo_url} | nawk '{ print $'${index}' }')
 130
 131             repo_urli="http://"${repo_urli}
 132
 133             pfexec pkg search -p -s "${repo_urli}" "${min_argu_1}" \
 134                2>>/dev/null | nawk -F"@" '{ print $1 }' | \
 135                nawk -F"pkg:/" '{ print $2 }' | uniq > /tmp/pkg_candi
 136
 137             pkgname=$(grep driver /tmp/pkg_candi | tail -1 2>/dev/null)
 138             if [ -z ${pkgname} ]; then
 139                 pkgname=$(tail -1 /tmp/pkg_candi 2>/dev/null)
 140             fi
 141
 142             if [[ $? != 0  ||  -z ${pkgname} ]]; then
 143                 index=$(($index + 1))
 144                 continue
 145             fi
 146
 147             depend_str="depend:incorporate:${pkgname}"
 148             pkg_ver=$(pfexec pkg search -l ${depend_str} | tail -1 2>>/dev/null)
 149
 150             if [ $? -eq 0 ]; then
 151                 pkgvalue=$(echo ${pkg_ver} | nawk '{ print $3 }')
 152                 if [ -z "`echo ${pkgvalue} | grep "pkg:/"`" ]; then
 153                     pkgcondidate=${pkgvalue}
 154                 else
 155                     pkgcondidate=$(echo ${pkgvalue} | nawk -F"pkg:/" '{ print $2 }')
 156                 fi
 157             fi
 158
 159             if [ -z ${pkgcondidate} ]; then
 160                 pkgcondidate=${pkgname}
 161             fi
 162             pfexec pkg info -r pkg://${repo_listi}/${pkgcondidate} \
 163                 >/dev/null 2>>$err_log
 164             if [ $? -eq 0 ]; then
 165                 /usr/bin/printf "${repo_listi}\t${pkgcondidate}\n"
 166                 pakcage_found=1
 167             fi
 168             index=$(($index + 1))
 169         done
 170     else
 171         repo_listi=${min_argu_2}
 172         pakcage_found=0
 173         repo_urli=$(pfexec pkg publisher ${repo_listi}  | grep "URI" \
 174             | nawk -F"http://" '{ print $2 }')
 175
 176         for repo_urli_i in $repo_urli; do
 177             repo_urli_i="http://"${repo_urli_i}
 178
 179             pfexec pkg search -p -s "${repo_urli_i}" "${min_argu_1}" \
 180               2>>/dev/null | nawk -F"@" '{ print $1 }' | \
 181               nawk -F"pkg:/" '{ print $2 }' | uniq > /tmp/pkg_candi
 182
 183             pkgname=$(grep driver /tmp/pkg_candi | tail -1 2>/dev/null)
 184             if [ -z ${pkgname} ]; then
 185                 pkgname=$(tail -1 /tmp/pkg_candi 2>/dev/null)
 186             fi
 187             if [ ! -z ${pkgname} ]; then
 188                 break
 189             fi
 190         done
 191
 192         if [[ $? != 0  ||  -z ${pkgname} ]]; then
 193             exit 1
 194         fi
 195
 196         depend_str="depend:incorporate:${pkgname}"
 197         pkg_ver=$(pfexec pkg search -l ${depend_str} | tail -1 2>>/dev/null)
 198
 199         if [ $? -eq 0 ]; then
 200             pkgvalue=$(echo ${pkg_ver} | nawk '{ print $3 }')
 201             if [ -z "`echo ${pkgvalue} | grep "pkg:/"`" ]; then
 202                 pkgcondidate=${pkgvalue}
 203             else
 204                 pkgcondidate=$(echo ${pkgvalue} | nawk -F"pkg:/" '{ print $2 }')
 205             fi
 206         fi
 207
 208         if [ -z ${pkgcondidate} ]; then
 209             pkgcondidate=${pkgname}
 210         fi
 211         pfexec pkg info -r pkg://${repo_listi}/${pkgcondidate} \
 212             >/dev/null 2>>$err_log
 213         if [ $? -eq 0 ]; then
 214             printf "${repo_listi}\t${pkgcondidate}\n"
 215             pakcage_found=1
 216         fi
 217     fi
 218     if [ ${pakcage_found} -eq 0 ]; then
 219         exit 1
 220     fi
 221 }
 222
 223 #
 224 # Add publisher on the system
 225 #
 226
 227 function add
 228 {
 229     #
 230     #add a repo and set it to default one if set.
 231     #
 232     if [ -z ${min_argu_2} ]; then
 233         print -u2 "\n$0: Third argument can not be null..."
 234         exit 1
 235     fi
 236     publisher_name=${min_argu_1}
 237     publisher_url=${min_argu_2}
 238     pfexec pkg set-publisher -O ${publisher_url} ${publisher_name} \
 239         2>>$err_log 1>&2
 240     if [ $? -ne 0 ]; then
 241         print -u2 "$0: Error found when adding repo: "
 242         print -u2 "${publisher_name} ${publisher_url}"
 243         exit 1
 244     fi
 245 }
 246
 247 #
 248 # Modify the existing publisher on the system via
 249 # pkg set-publisher and pkg unset-publisher.
 250 #
 251 function modify
 252 {
 253     #
 254     #modify a repo and set it to default one if set.
 255     #
 256     if [ -z ${min_argu_2} ]; then
 257         print -u2 "\n$0: Third argument can not be null..."
 258         exit 1
 259     fi
 260     publisher_name=${min_argu_1}
 261     publisher_url=${min_argu_2}
 262
 263      LC_ALL="C" pfexec pkg publisher ${publisher_name} > \
 264         /tmp/spe_repo_temp_store 2>>$err_log
 265
 266
 267     origin_uri=`cat /tmp/spe_repo_temp_store | grep "Origin URI" | \
 268                 nawk -F ": " '{ print $2 }' 2>>$err_log`
 269
 270     mirror_uris=`cat /tmp/spe_repo_temp_store | grep "Mirror URI" | \
 271                 nawk -F ": " '{ print $2 }' 2>>$err_log`
 272
 273     store_repo=" -O "${origin_uri}
 274     extra_repo=" "
 275     unset_extra_repo=" "
 276
 277     for mirror_uri in ${mirror_uris}
 278     do
 279         extra_repo=${extra_repo}" -m "${mirror_uri}
 280         unset_extra_repo=${unset_extra_repo}" -M "${mirror_uri}
 281     done
 282
 283     extra_repo=${extra_repo}" "${publisher_name}
 284
 285     pfexec pkg unset-publisher ${publisher_name} 2>>/dev/null 1>&2
 286     if [ $? -ne 0 ]; then
 287         print -u2 "Warning: Can not remove repo ${publisher_name}"
 288
 289         pfexec pkg set-publisher -O ${publisher_url} \
 290                                     ${unset_extra_repo} \
 291                                     ${publisher_name} \
 292                                     2>>$err_log 1>&2
 293         if [ $? -ne 0 ]; then
 294             print -u2 "$0: Error found when adding repo:"
 295             print -u2 "${publisher_name} ${publisher_url}"
 296             exit 1
 297         else
 298             echo ${store_repo} > /tmp/spe_repo_store
 299             echo ${extra_repo} >> /tmp/spe_repo_store
 300         fi
 301     else
 302         echo ${store_repo} > /tmp/spe_repo_store
 303         echo ${extra_repo} >> /tmp/spe_repo_store
 304     fi
 305 }
 306
 307 #
 308 # Restore the original publisher on the system
 309 #
 310 function restore
 311 {
 312     pfexec pkg set-publisher ${min_argu_1} ${min_argu_2} \
 313         2>>$err_log 1>&2
 314     if [ $? -ne 0 ]; then
 315         print -u2 "$0: Error found when restoring repo!"
 316         exit 1
 317     fi
 318 }
 319
 320 function clean_up
 321 {
 322     {
 323         rm -f  /tmp/repo_temp_store /tmp/repo_default_store \
 324                /tmp/repo_spe_store \
 325                /tmp/spe_repo_temp_store /tmp/pkg_candi
 326     } >/dev/null 2>&1
 327 }
 328
 329
 330 #Main()
 331
 332 PATH=/usr/bin:/usr/sbin:$PATH; export PATH
 333 LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/ddu/lib; export LD_LIBRARY_PATH
 334
 335 trap 'clean_up;exit 10' KILL INT
 336 trap 'clean_up' EXIT
 337
 338
 339 #
 340 # Confirm the correct number of commnad line argument
 341 #
 342 if (( $# < 2 )) || (( $# > 3 )); then
 343     print -u2 "\n$0: Wrong number of arguments provided..."
 344     exit 1
 345 fi
 346
 347 if [ -z $1 ]; then
 348     print -u2 "\n$0: First argument can not be null..."
 349     exit 1
 350 fi
 351
 352 if [ -z $2 ]; then
 353     print -u2 "\n$0: Second argument can not be null..."
 354     exit 1
 355 fi
 356
 357
 358 typeset main_argu=$1
 359 typeset min_argu_1=$2
 360 typeset min_argu_2=$3
 361 typeset repo_list repo_url repo_count
 362 typeset repo_listi repo_urli repo_name pkgname
 363 typeset depend_str pkg_ver pkgcondidate pakcage_found
 364 typeset index
 365 typeset  err_log=/tmp/ddu_err.log
 366
 367 #
 368 # Copy /var/pkg to /tmp if booted off a ramdisk
 369 # Needed by pkg utility so it can expand its bootkeeping files in swap space.
 370 #
 371 if [ -x /usr/lib/install/live_img_pkg5_prep ] ; then
 372     pfexec /usr/lib/install/live_img_pkg5_prep 2>>$err_log
 373     if [ $? != 0 ]; then
 374         print -u2 "Error executing: /usr/lib/install/live_img_pkg5_prep."
 375         exit 1
 376     fi
 377
 378 fi
 379
 380 case ${main_argu} in
 381 list)   list;;
 382 search) search;;
 383 add)    add;;
 384 modify) modify;;
 385 restore) restore;;
 386 *)      print -u2 "\n$0: ${main_argu} is not valid"
 387         exit 1;;
 388 esac