src/functions/functions.device

   1 #!/bin/bash
   2 ###############################################################################
   3 #                                                                             #
   4 # IPFire.org - A linux based firewall                                         #
   5 # Copyright (C) 2010  Michael Tremer & Christian Schmidt                      #
   6 #                                                                             #
   7 # This program is free software: you can redistribute it and/or modify        #
   8 # it under the terms of the GNU General Public License as published by        #
   9 # the Free Software Foundation, either version 3 of the License, or           #
  10 # (at your option) 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 General Public License for more details.                                #
  16 #                                                                             #
  17 # You should have received a copy of the GNU General Public License           #
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
  19 #                                                                             #
  20 ###############################################################################
  21
  22 device_list() {
  23         local devices
  24
  25         # Add all interfaces
  26         list_append devices $(devices_get_all)
  27
  28         # Add all PHYs
  29         list_append devices $(phy_list)
  30
  31         # Add all serial devices
  32         list_append devices $(serial_list)
  33
  34         # Return a sorted result
  35         list_sort ${devices}
  36 }
  37
  38 # Check if the device exists
  39 device_exists() {
  40         local device=${1}
  41
  42         # If device name was not found, exit.
  43         [ -n "${device}" ] || return ${EXIT_ERROR}
  44
  45         # Check for a normal network device.
  46         [ -d "${SYS_CLASS_NET}/${device}" ] && return ${EXIT_OK}
  47
  48         # If the check above did not find a result,
  49         # we check for PHYs.
  50         phy_exists "${device}" && return ${EXIT_OK}
  51
  52         # If the check above did not find a result,
  53         # we check for serial devices.
  54         serial_exists ${device}
  55 }
  56
  57 device_matches_pattern() {
  58         local device="${1}"
  59         assert isset device
  60
  61         local pattern="${2}"
  62         assert isset pattern
  63
  64         pattern="^${pattern//N/[[:digit:]]+}$"
  65
  66         [[ ${device} =~ ${pattern} ]] \
  67                 && return ${EXIT_TRUE} || return ${EXIT_FALSE}
  68 }
  69
  70 device_delete() {
  71         local device=${1}
  72         assert isset device
  73
  74         # Nothing to do, it device does not exist.
  75         device_exists ${device} || return ${EXIT_OK}
  76
  77         # Delete the device.
  78         cmd_quiet ip link delete ${device}
  79         local ret=$?
  80
  81         if [ ${ret} -ne ${EXIT_OK} ]; then
  82                 log ERROR "device: Could not delete device '${device}': ${ret}"
  83                 return ${EXIT_ERROR}
  84         fi
  85
  86         return ${ret}
  87 }
  88
  89 device_has_flag() {
  90         local device=${1}
  91         local flag=${2}
  92
  93         local flags=$(__device_get_file ${device} flags)
  94
  95         if [[ "$(( ${flags} & ${flag} ))" -eq 0 ]]; then
  96                 return ${EXIT_FALSE}
  97         else
  98                 return ${EXIT_TRUE}
  99         fi
 100 }
 101
 102 # Check if the device is up
 103 device_is_up() {
 104         local device=${1}
 105
 106         device_exists ${device} || return ${EXIT_ERROR}
 107
 108         device_has_flag ${device} 0x1
 109 }
 110
 111 device_ifindex_to_name() {
 112         local idx=${1}
 113         assert isset idx
 114
 115         local device device_idx
 116         for device in ${SYS_CLASS_NET}/*; do
 117                 device=$(basename ${device})
 118                 device_exists ${device} || continue
 119
 120                 device_idx=$(device_get_ifindex ${device})
 121
 122                 if [ "${device_idx}" = "${idx}" ]; then
 123                         print "${device}"
 124                         return ${EXIT_OK}
 125                 fi
 126         done
 127
 128         return ${EXIT_ERROR}
 129 }
 130
 131 device_get_ifindex() {
 132         local device=${1}
 133         assert isset device
 134
 135         local path="${SYS_CLASS_NET}/${1}/ifindex"
 136
 137         # Check if file can be read.
 138         [ -r "${path}" ] || return ${EXIT_ERROR}
 139
 140         print "$(<${path})"
 141 }
 142
 143 # Check if the device is a batman-adv bridge
 144 device_is_batman_adv() {
 145         [ -d "${SYS_CLASS_NET}/${1}/mesh" ]
 146 }
 147
 148 # Check if the device is a batman-adv slave port
 149 device_is_batman_adv_slave() {
 150         local device="${1}"
 151
 152         if [ -d "${SYS_CLASS_NET}/${device}/batman_adv" ]; then
 153                 local status="$(<${SYS_CLASS_NET}/${device}/batman_adv/iface_status)"
 154
 155                 case "${status}" in
 156                         "active")
 157                                 return ${EXIT_TRUE}
 158                                 ;;
 159                         *)
 160                                 return ${EXIT_FALSE}
 161                                 ;;
 162                 esac
 163         fi
 164
 165         return ${EXIT_FALSE}
 166 }
 167
 168 # Check if the device is a bonding device
 169 device_is_bonding() {
 170         [ -d "/sys/class/net/${1}/bonding" ]
 171 }
 172
 173 # Check if the device bonded in a bonding device
 174 device_is_bonded() {
 175         local device=${1}
 176
 177         [ -d "${SYS_CLASS_NET}/${device}/bonding_slave" ]
 178 }
 179
 180 # Check if the device is a bridge
 181 device_is_bridge() {
 182         [ -d "/sys/class/net/${1}/bridge" ]
 183 }
 184
 185 device_is_bridge_attached() {
 186         local device=${1}
 187         [ -d "${SYS_CLASS_NET}/${device}/brport" ]
 188 }
 189
 190 device_is_wireless_monitor() {
 191         local device="${1}"
 192         assert isset device
 193
 194         device_is_wireless "${device}" && \
 195                 device_matches_pattern "${device}" "${PORT_PATTERN_WIRELESS_MONITOR}"
 196 }
 197
 198 device_is_wireless_adhoc() {
 199         local device="${1}"
 200         assert isset device
 201
 202         device_is_wireless "${device}" && \
 203                 device_matches_pattern "${device}" "${PORT_PATTERN_WIRELESS_ADHOC}"
 204 }
 205
 206 device_get_bridge() {
 207         local device=${1}
 208         assert isset device
 209
 210         # Check if device is attached to a bridge.
 211         device_is_bridge_attached ${device} || return ${EXIT_ERROR}
 212
 213         local ifindex_path="${SYS_CLASS_NET}/${device}/brport/bridge/ifindex"
 214         [ -r "${ifindex_path}" ] || return ${EXIT_ERROR}
 215
 216         local ifindex=$(<${ifindex_path})
 217         assert isset ifindex
 218
 219         device_ifindex_to_name ${ifindex}
 220 }
 221
 222 # Check if the device is a vlan device
 223 device_is_vlan() {
 224         local device=${1}
 225         assert isset device
 226
 227         [ -e "${PROC_NET_VLAN}/${device}" ]
 228 }
 229
 230 # Check if the device has vlan devices
 231 device_has_vlans() {
 232         local device=${1}
 233         assert isset device
 234
 235         if device_is_vlan ${device}; then
 236                 return ${EXIT_FALSE}
 237         fi
 238
 239         local vlans=$(device_get_vlans ${device})
 240         [ -n "${vlans}" ] && return ${EXIT_OK} || return ${EXIT_ERROR}
 241 }
 242
 243 device_get_vlans() {
 244         local device=${1}
 245         assert isset device
 246
 247         # If no 8021q module has been loaded into the kernel,
 248         # we cannot do anything.
 249         [ -r "${PROC_NET_VLAN_CONFIG}" ] || return ${EXIT_OK}
 250
 251         local dev spacer1 id spacer2 parent
 252         while read dev spacer1 id spacer2 parent; do
 253                 [ "${parent}" = "${device}" ] || continue
 254
 255                 print "${dev}"
 256         done < ${PROC_NET_VLAN_CONFIG}
 257 }
 258
 259 # Check if the device is a ppp device
 260 device_is_ppp() {
 261         local device=${1}
 262
 263         local type=$(__device_get_file ${device} type)
 264
 265         [ "${type}" = "512" ] && return ${EXIT_OK} || return ${EXIT_ERROR}
 266 }
 267
 268 # Check if the device is a pointopoint device.
 269 device_is_ptp() {
 270         local device=${1}
 271
 272         device_has_flag ${device} 0x10
 273 }
 274
 275 # Check if the device is a loopback device
 276 device_is_loopback() {
 277         local device=${1}
 278
 279         [ "${device}" = "lo" ]
 280 }
 281
 282 # Check if the device is a dummy device
 283 #  This is the worst possible check, but all I could come up with
 284 device_is_dummy() {
 285         local device="${1}"
 286
 287         [[ ${device} =~ ^dummy[0-9]+$ ]]
 288 }
 289
 290 # Check if the device is a wireless device
 291 device_is_wireless() {
 292         local device=${1}
 293
 294         [ -d "${SYS_CLASS_NET}/${device}/phy80211" ]
 295 }
 296
 297 device_get_phy() {
 298         local device="${1}"
 299
 300         if device_is_wireless "${device}"; then
 301                 print "$(<${SYS_CLASS_NET}/${device}/phy80211/name)"
 302                 return ${EXIT_OK}
 303         fi
 304
 305         return ${EXIT_ERROR}
 306 }
 307
 308 device_is_phy() {
 309         phy_exists $@
 310 }
 311
 312 device_is_serial() {
 313         serial_exists $@
 314 }
 315
 316 # Returns true if a device is a tun device
 317 device_is_tun() {
 318         local device="${1}"
 319
 320         [ -e "${SYS_CLASS_NET}/${device}/tun_flags" ]
 321 }
 322
 323 # Check if the device is a physical network interface
 324 device_is_ethernet() {
 325         local device=${1}
 326
 327         device_is_ethernet_compatible "${device}" || \
 328                 return ${EXIT_ERROR}
 329
 330         device_is_loopback ${device} && \
 331                 return ${EXIT_ERROR}
 332
 333         device_is_bonding ${device} &&  \
 334                 return ${EXIT_ERROR}
 335
 336         device_is_bridge ${device} && \
 337                 return ${EXIT_ERROR}
 338
 339         device_is_ppp ${device} && \
 340                 return ${EXIT_ERROR}
 341
 342         device_is_vlan ${device} && \
 343                 return ${EXIT_ERROR}
 344
 345         device_is_dummy ${device} && \
 346                 return ${EXIT_ERROR}
 347
 348         device_is_tun ${device} && \
 349                 return ${EXIT_ERROR}
 350
 351         return ${EXIT_OK}
 352 }
 353
 354 # Get the device type
 355 device_get_type() {
 356         local device=${1}
 357
 358         # If the device does not exist (happens on udev remove events),
 359         # we do not bother to run all checks.
 360         if ! device_exists "${device}"; then
 361                 echo "unknown"
 362
 363         elif device_is_vlan ${device}; then
 364                 echo "vlan"
 365
 366         elif device_is_bonding ${device}; then
 367                 echo "bonding"
 368
 369         elif device_is_bridge ${device}; then
 370                 echo "bridge"
 371
 372         elif device_is_ppp ${device}; then
 373                 echo "ppp"
 374
 375         elif device_is_batman_adv ${device}; then
 376                 echo "batman-adv"
 377
 378         elif device_is_loopback ${device}; then
 379                 echo "loopback"
 380
 381         elif device_is_wireless_adhoc ${device}; then
 382                 echo "wireless-adhoc"
 383
 384         elif device_is_wireless ${device}; then
 385                 echo "wireless"
 386
 387         elif device_is_dummy ${device}; then
 388                 echo "dummy"
 389
 390         elif device_is_tun ${device}; then
 391                 echo "tun"
 392
 393         elif device_is_ethernet ${device}; then
 394                 echo "ethernet"
 395
 396         elif device_is_serial ${device}; then
 397                 echo "serial"
 398
 399         elif device_is_phy ${device}; then
 400                 echo "phy"
 401
 402         else
 403                 echo "unknown"
 404         fi
 405 }
 406
 407 device_is_ethernet_compatible() {
 408         local device="${1}"
 409
 410         # /sys/class/net/*/type must equal 1 for ethernet compatible devices
 411         local type="$(__device_get_file "${device}" "type")"
 412         [[ "${type}" = "1" ]]
 413 }
 414
 415 device_get_status() {
 416         local device=${1}
 417         assert isset device
 418
 419         local status=${STATUS_DOWN}
 420
 421         if device_is_up ${device}; then
 422                 status=${STATUS_UP}
 423
 424                 if ! device_has_carrier ${device}; then
 425                         status=${STATUS_NOCARRIER}
 426                 fi
 427         fi
 428
 429         echo "${status}"
 430 }
 431
 432 device_get_address() {
 433         local device=${1}
 434
 435         cat ${SYS_CLASS_NET}/${device}/address 2>/dev/null
 436 }
 437
 438 device_set_address() {
 439         assert [ $# -eq 2 ]
 440
 441         local device="${1}"
 442         local addr="${2}"
 443
 444         if ! device_exists "${device}"; then
 445                 error "Device '${device}' does not exist."
 446                 return ${EXIT_ERROR}
 447         fi
 448
 449         # Do nothing if the address has not changed
 450         local old_addr="$(device_get_address "${device}")"
 451         if [ -n "${old_addr}" -a "${addr}" = "${old_addr}" ]; then
 452                 return ${EXIT_OK}
 453         fi
 454
 455         log DEBUG "Setting address of '${device}' from '${old_addr}' to '${addr}'"
 456
 457         local up
 458         if device_is_up "${device}"; then
 459                 device_set_down "${device}"
 460                 up=1
 461         fi
 462
 463         ip link set "${device}" address "${addr}"
 464         local ret=$?
 465
 466         if [ "${up}" = "1" ]; then
 467                 device_set_up "${device}"
 468         fi
 469
 470         if [ "${ret}" != "0" ]; then
 471                 error_log "Could not set address '${addr}' on device '${device}'"
 472         fi
 473
 474         return ${ret}
 475 }
 476
 477 device_get() {
 478         local device
 479         local devices
 480
 481         for device in ${SYS_CLASS_NET}/*; do
 482                 device=$(basename ${device})
 483
 484                 # bonding_masters is no device
 485                 [ "${device}" = "bonding_masters" ] && continue
 486
 487                 devices="${devices} ${device}"
 488         done
 489
 490         echo ${devices}
 491         return ${EXIT_OK}
 492 }
 493
 494 devices_get_all() {
 495         device_get
 496 }
 497
 498 # Check if a device has a cable plugged in
 499 device_has_carrier() {
 500         local device=${1}
 501         assert isset device
 502
 503         local carrier=$(__device_get_file ${device} carrier)
 504         [ "${carrier}" = "1" ]
 505 }
 506
 507 device_is_promisc() {
 508         local device=${1}
 509
 510         device_has_flag ${device} 0x200
 511 }
 512
 513 device_set_promisc() {
 514         local device=${1}
 515         local state=${2}
 516
 517         assert device_exists ${device}
 518         assert isset state
 519         assert isoneof state on off
 520
 521         ip link set ${device} promisc ${state}
 522 }
 523
 524 # Check if the device is free
 525 device_is_free() {
 526         ! device_is_used $@
 527 }
 528
 529 # Check if the device is used
 530 device_is_used() {
 531         local device=${1}
 532
 533         device_has_vlans ${device} && \
 534                 return ${EXIT_OK}
 535         device_is_bonded ${device} && \
 536                 return ${EXIT_OK}
 537         device_is_bridge_attached ${device} && \
 538                 return ${EXIT_OK}
 539
 540         return ${EXIT_ERROR}
 541 }
 542
 543 # Give the device a new name
 544 device_set_name() {
 545         local source=$1
 546         local destination=${2}
 547
 548         # Check if devices exists
 549         if ! device_exists ${source} || device_exists ${destination}; then
 550                 return 4
 551         fi
 552
 553         local up
 554         if device_is_up ${source}; then
 555                 ip link set ${source} down
 556                 up=1
 557         fi
 558
 559         ip link set ${source} name ${destination}
 560
 561         if [ "${up}" = "1" ]; then
 562                 ip link set ${destination} up
 563         fi
 564 }
 565
 566 # Set device up
 567 device_set_up() {
 568         assert [ $# -eq 1 ]
 569
 570         local device=${1}
 571
 572         # Do nothing if device is already up
 573         device_is_up ${device} && return ${EXIT_OK}
 574
 575         log INFO "Bringing up ${device}"
 576
 577         device_set_parent_up ${device}
 578         if ! cmd ip link set ${device} up; then
 579                 return ${EXIT_ERROR}
 580         fi
 581
 582         # Set SMP affinity
 583         if interrupt_use_smp_affinity; then
 584                 device_auto_configure_smp_affinity ${device}
 585         fi
 586
 587         return ${EXIT_OK}
 588 }
 589
 590 device_set_parent_up() {
 591         local device=${1}
 592         local parent
 593
 594         if device_is_vlan ${device}; then
 595                 parent=$(vlan_get_parent ${device})
 596
 597                 device_is_up ${parent} && return ${EXIT_OK}
 598
 599                 log DEBUG "Setting up parent device '${parent}' of '${device}'"
 600
 601                 device_set_up ${parent}
 602                 return $?
 603         fi
 604
 605         return ${EXIT_OK}
 606 }
 607
 608 # Set device down
 609 device_set_down() {
 610         assert [ $# -eq 1 ]
 611
 612         local device=${1}
 613         local ret=${EXIT_OK}
 614
 615         if device_is_up ${device}; then
 616                 log INFO "Bringing down ${device}"
 617
 618                 cmd ip link set ${device} down
 619                 ret=$?
 620         fi
 621
 622         device_set_parent_down ${device}
 623
 624         return ${ret}
 625 }
 626
 627 device_set_parent_down() {
 628         local device=${1}
 629         local parent
 630
 631         if device_is_vlan ${device}; then
 632                 parent=$(vlan_get_parent ${device})
 633
 634                 device_is_up ${parent} || return ${EXIT_OK}
 635
 636                 if device_is_free ${parent}; then
 637                         log DEBUG "Tearing down parent device '${parent}' of '${device}'"
 638
 639                         device_set_down ${parent}
 640                 fi
 641         fi
 642
 643         return ${EXIT_OK}
 644 }
 645
 646 device_get_mtu() {
 647         local device=${1}
 648
 649         # Return an error if the device does not exist
 650         device_exists ${device} || return ${EXIT_ERROR}
 651
 652         echo $(<${SYS_CLASS_NET}/${device}/mtu)
 653 }
 654
 655 # Set mtu to a device
 656 device_set_mtu() {
 657         local device=${1}
 658         local mtu=${2}
 659
 660         assert device_exists ${device}
 661
 662         # Handle bridges differently
 663         if device_is_bridge ${device}; then
 664                 local port
 665                 for port in $(bridge_get_members ${device}); do
 666                         device_set_mtu ${port} ${mtu}
 667                 done
 668         fi
 669
 670         log INFO "Setting MTU of ${device} to ${mtu}"
 671
 672         local up
 673         if device_is_up ${device}; then
 674                 device_set_down ${device}
 675                 up=1
 676         fi
 677
 678         local ret=${EXIT_OK}
 679         if ! cmd ip link set ${device} mtu ${mtu}; then
 680                 ret=${EXIT_ERROR}
 681
 682                 log ERROR "Could not set MTU ${mtu} on ${device}"
 683         fi
 684
 685         if [ "${up}" = "1" ]; then
 686                 device_set_up ${device}
 687         fi
 688
 689         return ${ret}
 690 }
 691
 692 device_adjust_mtu() {
 693         assert [ $# -eq 2 ]
 694
 695         local device="${1}"
 696         local other_device="${2}"
 697
 698         local mtu="$(device_get_mtu "${other_device}")"
 699         device_set_mtu "${device}" "${mtu}"
 700 }
 701
 702 device_discover() {
 703         local device=${1}
 704
 705         log INFO "Running discovery process on device '${device}'."
 706
 707         local hook
 708         for hook in $(hook_zone_get_all); do
 709                 hook_zone_exec ${hook} discover ${device}
 710         done
 711 }
 712
 713 device_identify() {
 714         assert [ $# -ge 1 ]
 715
 716         local device="${1}"
 717
 718         # Flash for ten seconds by default
 719         local seconds="10"
 720
 721         # Run in background?
 722         local background="false"
 723
 724         local arg
 725         while read arg; do
 726                 case "${arg}" in
 727                         --background)
 728                                 background="true"
 729                                 ;;
 730                         --seconds=*)
 731                                 seconds="$(cli_get_val "${arg}")"
 732                                 ;;
 733                 esac
 734         done <<< "$(args $@)"
 735
 736         assert isinteger seconds
 737
 738         if ! device_exists "${device}"; then
 739                 log ERROR "Cannot identify device ${device}: Does not exist"
 740                 return ${EXIT_ERROR}
 741         fi
 742
 743         if ! device_is_ethernet "${device}"; then
 744                 log DEBUG "Cannot identify device ${device}: Not an ethernet device"
 745                 return ${EXIT_NOT_SUPPORTED}
 746         fi
 747
 748         log DEBUG "Identifying device ${device}"
 749
 750         local command="ethtool --identify ${device} ${seconds}"
 751         local ret=0
 752
 753         if enabled background; then
 754                 cmd_background "${command}"
 755         else
 756                 cmd_quiet "${command}"
 757                 ret=$?
 758         fi
 759
 760         return ${ret}
 761 }
 762
 763 device_has_ip() {
 764         local device=${1}
 765         local addr=${2}
 766
 767         assert isset addr
 768         assert device_exists ${device}
 769
 770         # IPv6 addresses must be fully imploded
 771         local protocol=$(ip_detect_protocol ${addr})
 772         case "${protocol}" in
 773                 ipv6)
 774                         addr=$(ipv6_format "${addr}")
 775                         ;;
 776         esac
 777
 778         listmatch ${addr} $(device_get_addresses ${device})
 779 }
 780
 781 device_get_addresses() {
 782         local device=${1}
 783
 784         assert device_exists ${device}
 785
 786         local prot
 787         local addr
 788         local line
 789         ip addr show ${device} | \
 790                 while read prot addr line; do
 791                         [ "${prot:0:4}" = "inet" ] && echo "${addr}"
 792                 done
 793 }
 794
 795 __device_get_file() {
 796         local device=${1}
 797         local file=${2}
 798
 799         assert isset device
 800         assert isset file
 801
 802         local path="${SYS_CLASS_NET}/${device}/${file}"
 803         [ -r "${path}" ] || return ${EXIT_ERROR}
 804
 805         echo "$(<${path})"
 806 }
 807
 808 __device_set_file() {
 809         assert [ $# -eq 3 ]
 810
 811         local device="${1}"
 812         local file="${2}"
 813         local value="${3}"
 814
 815         local path="${SYS_CLASS_NET}/${device}/${file}"
 816         if [ ! -w "${path}" ]; then
 817                 log DEBUG "Cannot write to file '${file}' (${value})"
 818                 return ${EXIT_ERROR}
 819         fi
 820
 821         echo "${value}" > "${path}"
 822 }
 823
 824 device_get_rx_bytes() {
 825         local device=${1}
 826
 827         __device_get_file ${device} statistics/rx_bytes
 828 }
 829
 830 device_get_tx_bytes() {
 831         local device=${1}
 832
 833         __device_get_file ${device} statistics/tx_bytes
 834 }
 835
 836 device_get_rx_packets() {
 837         local device=${1}
 838
 839         __device_get_file ${device} statistics/rx_packets
 840 }
 841
 842 device_get_tx_packets() {
 843         local device=${1}
 844
 845         __device_get_file ${device} statistics/tx_packets
 846 }
 847
 848 device_get_rx_errors() {
 849         local device=${1}
 850
 851         __device_get_file ${device} statistics/rx_errors
 852 }
 853
 854 device_get_tx_errors() {
 855         local device=${1}
 856
 857         __device_get_file ${device} statistics/tx_errors
 858 }
 859
 860 device_get_speed() {
 861         local device=${1}
 862
 863         local speed=$(__device_get_file ${device} speed)
 864
 865         # Exit for no output (i.e. no link detected)
 866         isset speed || return ${EXIT_ERROR}
 867
 868         # Don't return anything for negative values
 869         [ ${speed} -lt 0 ] && return ${EXIT_ERROR}
 870
 871         print "${speed}"
 872 }
 873
 874 device_get_duplex() {
 875         local device=${1}
 876
 877         local duplex=$(__device_get_file ${device} duplex)
 878
 879         case "${duplex}" in
 880                 unknown)
 881                         return ${EXIT_ERROR}
 882                         ;;
 883                 *)
 884                         print "${duplex}"
 885                         ;;
 886         esac
 887 }
 888
 889 device_get_link_string() {
 890         local device="${1}"
 891         assert isset device
 892
 893         local s
 894
 895         local speed="$(device_get_speed "${device}")"
 896         if isset speed; then
 897                 list_append s "${speed} MBit/s"
 898         fi
 899
 900         local duplex="$(device_get_duplex "${device}")"
 901         if isset duplex; then
 902                 list_append s "${duplex} duplex"
 903         fi
 904
 905         print "${s}"
 906 }
 907
 908 device_auto_configure_smp_affinity() {
 909         assert [ $# -eq 1 ]
 910
 911         local device=${1}
 912
 913         if lock_acquire "smp-affinity" 60; then
 914                 device_set_smp_affinity ${device} auto
 915
 916                 lock_release "smp-affinity"
 917         fi
 918 }
 919
 920 device_set_smp_affinity() {
 921         assert [ $# -eq 2 ]
 922
 923         local device=${1}
 924         local mode=${2}
 925
 926         # mode can be auto which will automatically try to find
 927         # the least busy processor, or an integer for the desired
 928         # processor that should handle this device
 929
 930         local num_processors=$(system_get_processors)
 931
 932         if [ "${mode}" = "auto" ]; then
 933                 local processor=$(interrupt_choose_least_busy_processor)
 934         else
 935                 assert isinteger mode
 936                 local processor=${mode}
 937
 938                 if [ ${processor} -gt ${num_processors} ]; then
 939                         log ERROR "Processor ${processor} does not exist"
 940                         return ${EXIT_ERROR}
 941                 fi
 942         fi
 943
 944         local interrupts=$(interrupts_for_device ${device})
 945         if ! isset interrupts; then
 946                 log DEBUG "${device} has no interrupts. Not changing SMP affinity"
 947                 return ${EXIT_OK}
 948         fi
 949
 950         # Set SMP affinity
 951         local interrupt
 952         for interrupt in ${interrupts}; do
 953                 interrupt_set_smp_affinity ${interrupt} ${processor}
 954         done
 955
 956         # Find all queues and assign them to the next processor
 957         local queue
 958         for queue in $(device_get_queues ${device}); do
 959                 case "${queue}" in
 960                         # Only handle receive queues
 961                         rx-*)
 962                                 for interrupt in $(interrupts_for_device_queue ${device} ${queue}); do
 963                                         interrupt_set_smp_affinity ${interrupt} ${processor}
 964                                 done
 965
 966                                 device_queue_set_smp_affinity ${device} ${queue} ${processor}
 967                                 ;;
 968
 969                         # Ignore the rest
 970                         *)
 971                                 continue
 972                                 ;;
 973                 esac
 974
 975                 # Get the next available processor if in auto mode
 976                 [ "${mode}" = "auto" ] && processor=$(system_get_next_processor ${processor})
 977         done
 978
 979         return ${EXIT_OK}
 980 }
 981
 982 device_get_queues() {
 983         assert [ $# -eq 1 ]
 984
 985         local device=${1}
 986
 987         local queue
 988         for queue in ${SYS_CLASS_NET}/${device}/queues/*; do
 989                 [ -d "${queue}" ] || continue
 990
 991                 basename "${queue}"
 992         done
 993 }
 994
 995 device_supports_multiqueue() {
 996         local device=${1}
 997
 998         local num_queues=$(device_num_queues ${device})
 999
1000         if isset num_queues && [ ${num_queues} -gt 2 ]; then
1001                 return ${EXIT_TRUE}
1002         fi
1003
1004         return ${EXIT_FALSE}
1005 }
1006
1007 device_num_queues() {
1008         local device=${1}
1009         local type=${2}
1010
1011         isset type && assert isoneof type rx tx
1012
1013         local i=0
1014
1015         local q
1016         for q in $(device_get_queues ${device}); do
1017                 case "${type},${q}" in
1018                         rx,rx-*)
1019                                 (( i++ ))
1020                                 ;;
1021                         tx,tx-*)
1022                                 (( i++ ))
1023                                 ;;
1024                         *,*)
1025                                 (( i++ ))
1026                                 ;;
1027                 esac
1028         done
1029
1030         print ${i}
1031 }
1032
1033 device_queue_get_smp_affinity() {
1034         assert [ $# -eq 2 ]
1035
1036         local device=${1}
1037         local queue=${2}
1038
1039         local path="${SYS_CLASS_NET}/${device}/queues/${queue}"
1040
1041         case "${queue}" in
1042                 rx-*)
1043                         path="${path}/rps_cpus"
1044                         ;;
1045                 tx-*)
1046                         path="${path}/xps_cpus"
1047                         ;;
1048         esac
1049         assert [ -r "${path}" ]
1050
1051         __bitmap_to_processor_ids $(<${path})
1052 }
1053
1054 device_queue_set_smp_affinity() {
1055         assert [ $# -eq 3 ]
1056
1057         local device=${1}
1058         local queue=${2}
1059         local processor=${3}
1060
1061         local path="${SYS_CLASS_NET}/${device}/queues/${queue}/rps_cpus"
1062         assert [ -w "${path}" ]
1063
1064         log DEBUG "Setting SMP affinity of ${device} (${queue}) to processor ${processor}"
1065
1066         __processor_id_to_bitmap ${processor} > ${path}
1067 }