Move over here from https://reviews.freebsd.org/D3778HEADmaster

1 files changed, 257 insertions, 0 deletions

diff --git a/bluetooth-config b/bluetooth-config
new file mode 100644
index 0000000..daaf074
--- /dev/null
+++ b/bluetooth-config
@@ -0,0 +1,257 @@
+#!/bin/sh
+
+# Define our bail out shortcut
+exerr () { echo -e "Error: $*" >&2 ; exit 1; }
+
+# Assuming we are called to do the pair-new-device subcommand first
+
+main() {
+unset node device bdaddresses retry
+
+[ $( id -u ) -eq 0 ] || exerr "$0 must modify files that belong to root.  Re-run as root."
+
+# Get command line options
+while getopts :a:n: arg; do
+  case ${arg} in
+    n) node="$OPTARG";;
+    a) device="$OPTARG";;
+    ?) exerr "Syntax: $0 [-n node] [-a address] cmd";;
+  esac
+done
+
+known_nodes=$(/usr/sbin/hccontrol read_node_list 2>/dev/null | \
+    /usr/bin/tail -n +2 | /usr/bin/cut -d ' ' -f 1)
+
+# Check if netgraph knows about any HCI nodes
+if ! [ "${known_nodes}" ]; then
+  ng_nodes=$(/usr/sbin/ngctl list 2>/dev/null | \
+    /usr/bin/grep -o "Name: .* Type: ubt" | /usr/bin/cut -d ' ' -f 2)
+
+  [ "${ng_nodes}" ] || exerr "No Bluetooth host controllers found."
+
+  unset found
+  for n in ${ng_nodes}; do
+    if [ "${n}" = "${node%hci}" ]; then
+      # If we found the node but its stack is not set up, do it now
+      /usr/sbin/service bluetooth start ${node%hci} || exit 1
+      found="YES"
+    fi
+  done
+
+  # If we have Bluetooth controller nodes without a set up stack,
+  # ask the user if we shall start it up
+  if ! [ "${found}" ]; then
+    printf "No usable Bluetooth host controllers were found.\n"
+    printf "These host controllers exist in the system:\n  %s" " ${ng_nodes}"
+    read -p "Choose a host controller to set up: [${ng_nodes%% *}]" node
+    : ${node:="${ng_nodes%% *}"}
+    /usr/sbin/service bluetooth start ${node} || exit 1
+  fi
+
+  # Re-read known nodes
+  known_nodes=$(/usr/sbin/hccontrol read_node_list 2>/dev/null | \
+    /usr/bin/tail -n +2 | /usr/bin/cut -d ' ' -f 1)
+  # check if we succeeded in bringing it up
+  [ "${known_nodes}" ] || exerr "Failed to set up Bluetooth stack"
+fi
+
+# If a node was requested on command line, check if it is there
+if [ "${node}" ]; then
+  unset found
+  for n in ${known_nodes}; do
+    [ "${n}" = "${node}" ] && found="YES"
+    [ "${n}" = "${node}hci" ] && node="${node}hci" && found="YES"
+  done
+  [ "${found}" ] || exerr "Node ${node} not found"
+fi
+
+[ "${node}" ] && node="-n ${node}"
+
+while ! [ "${bdaddresses}" ]; do
+  retry=X${retry}
+  printf "Scanning for new Bluetooth devices (Attempt %d of 5) ... " ${#retry}
+  bdaddresses=$( /usr/sbin/hccontrol -N ${node} inquiry 2>/dev/null | \
+    /usr/bin/grep -o "BD_ADDR: .*" | /usr/bin/cut -d ' ' -f 2 )
+
+  # Count entries and, if a device was requested on command line,
+  # try to find it
+  unset found count
+  for bdaddress in ${bdaddresses}; do
+    count=X${count}
+    if [ "${bdaddress}" = "${device}" ]; then
+      found=YES
+      bdaddresses="${device}"
+      count=X
+      break
+    fi
+  done
+
+  # If device was requested on command line but is not found,
+  # or no devices found at all, rescan until retry is exhausted
+  if ! [ "${found}" -o "${count}" -a -z "${device}" ]; then
+    printf "failed.\n"
+    if [ "${#retry}" -eq 5 ]; then
+      [ "${device}" ] && exerr "Device ${device} not found"
+      exerr "No new Bluetooth devices found"
+    fi
+    unset bdaddresses
+    sleep 2
+    continue
+  fi
+
+  printf "done.\nFound %d new Bluetooth devic%.*s (scanning for names):\n" ${#count} ${#count} es
+
+  # Looping again for the faster feedback
+  unset count
+  for bdaddress in ${bdaddresses}; do
+    count=X${count}
+    bdname=$( /usr/bin/bthost -b "${bdaddress}" 2>/dev/null )
+    friendlyname=$( /usr/sbin/hccontrol Remote_Name_Request ${bdaddress} 2> /dev/null | \
+      /usr/bin/grep -o "Name: .*" | /usr/bin/cut -d ' ' -f 2- )
+
+    # sdpcontrol should be able to pull vendor and product id via sdp
+    printf "[%2d] %s\t\"%s\" (%s)\n" ${#count} "${bdaddress}" "${friendlyname}" "${bdname}"
+
+    eval bdaddress_${#count}=\${bdaddress}
+    eval bdname_${#count}=\${bdname}
+    eval friendlyname_${#count}=\${friendlyname}
+  done
+
+  # If a device was pre-selected, do not query the user
+  [ "${device}" ] && topair=1 || unset topair
+
+  # Even if only one device was found, user may chose 0 to rescan
+  while ! [ "${topair}" ]; do
+    read -p "Select device to pair with [1-${#count}, 0 to rescan]: " topair
+    if ! [ "${topair}" -ge 0 -a "${topair}" -le "${#count}" ] 2>/dev/null ; then
+      printf "Value out of range: %s.\n" {topair}
+      unset topair
+    fi
+  done
+
+  [ "${topair}" -eq "0" ] && unset bdaddresses retry
+done
+
+eval bdaddress=\${bdaddress_${topair}}
+eval bdname=\${bdname_${topair}}
+eval friendlyname=\${friendlyname_${topair}}
+
+# Do we need to add an entry to /etc/bluetooth/hosts?
+if ! [ "${bdname}" ]; then
+  printf "\nAdding device ${bdaddress} to /etc/bluetooth/hosts.\n"
+
+  while ! [ "${bdname}" ]; do
+    read -p "Enter friendly name. [${friendlyname}]: " REPLY
+    : ${REPLY:="${friendlyname}"}
+
+    if [ "${REPLY}" ]; then
+      # Remove white space and non-friendly characters
+      bdname=$( printf "%s" "${REPLY}" | tr -c '[:alnum:]-,.' _ )
+      [ "${REPLY}" != "${bdname}" ] && printf "Notice: Using sanitized name \"%s\" in /etc/bluetooth/hosts.\n" "${bdname}"
+    fi
+  done
+
+  printf "%s\t%s\n" "${bdaddress}" "${bdname}" >> /etc/bluetooth/hosts
+fi
+
+# If scanning for the name did not succeed, resort to bdname
+: ${friendlyname:="${bdname}"}
+
+# Now over to hcsecd
+
+# Since hcsecd does not allow querying for known devices, we need to
+# check for bdaddr entries manually.
+#
+# Also we cannot really modify the PIN in an existing entry. So we
+# need to prompt the user to manually do it and restart this script.
+if ! /usr/sbin/service hcsecd enabled; then
+  printf "\nWarning: hcsecd is not enabled.\nThis daemon manages pairing requests.\n"
+  read -p "Enable hcsecd? [yes]: " REPLY
+  case "${REPLY}" in no|n|NO|N|No|nO) ;; *) /usr/sbin/sysrc hcsecd_enable="YES";; esac
+fi
+secd_config=$( /usr/sbin/sysrc -n hcsecd_config )
+secd_entries=$( /usr/bin/grep -Eo "bdaddr[[:space:]]+(${bdaddress}|${bdname})" ${secd_config} | awk '{ print $2; }' )
+
+if [ "${secd_entries}" ]; then
+  printf "\nWarning: An entry for device %s is already present in %s.\n" ${secd_entries} ${secd_config}
+  printf "To modify pairing information, edit this file and run\n  service hcsecd restart\n"
+  read -p "Continue? [yes]: " REPLY
+  case "${REPLY}" in no|n|NO|N|No|nO) exit;; esac
+else
+  printf "\nWriting pairing information description block to %s.\n" ${secd_config}
+  printf "(To get PIN, put device in pairing mode first.)\n"
+  read -p "Enter PIN [nopin]: " pin
+  [ "${pin}" ] && pin=\""${pin}"\" || pin="nopin"
+
+  # Write out new hcsecd config block
+  printf "\ndevice {\n\tbdaddr\t%s;\n\tname\t\"%s\";\n\tkey\tnokey\;\n\tpin\t%s\;\n}\n" \
+    "${bdaddress}" "${friendlyname}" "${pin}" >> ${secd_config}
+
+  # ... and make daemon reload config, TODO: hcsecd should provide a reload hook
+  /usr/sbin/service hcsecd restart
+
+  # TODO: We should check if hcsecd succeeded pairing and revert to an old version
+  # of hcsecd.conf so we can undo adding the block above and retry with a new PIN
+  # also, if there's a way to force devices to re-pair, try this
+fi
+
+# Now check for specific services to be provided by the device
+# First up: HID
+
+if /usr/sbin/sdpcontrol -a "${bdaddress}" search HID | \
+   /usr/bin/grep -q "^Record Handle: "; then
+
+  printf "\nThis device provides human interface device services.\n"
+  read -p "Set it up? [yes]: " REPLY
+  case "${REPLY}" in no|n|NO|N|No|nO) ;;
+  *)
+    if ! /usr/sbin/service bthidd enabled; then
+      printf "\nWarning: bthidd is not enabled."
+      printf "\nThis daemon manages Bluetooth HID devices.\n"
+      read -p "Enable bthidd? [yes]: " REPLY
+      case "${REPLY}" in no|n|NO|N|No|nO) ;; *) /usr/sbin/sysrc bthidd_enable="YES";; esac
+    fi
+
+    # Check if bthidd already knows about this device
+    bthidd_known=$( /usr/sbin/bthidcontrol -a "${bdaddress}" known | \
+      /usr/bin/grep "${bdaddress}" )
+    if [ "${bthidd_known}" ]; then
+      printf "Notice: Device %s already known to bthidd.\n" "${bdaddress}"
+    else
+      bthidd_config=$( /usr/sbin/sysrc -n bthidd_config )
+      printf "Writing HID descriptor block to %s ... " "${bthidd_config}"
+      /usr/sbin/bthidcontrol -a "${bdaddress}" query >> "${bthidd_config}"
+
+      # Re-read config to see if we succeeded adding the device
+      bthidd_known=$( /usr/sbin/bthidcontrol -a "${bdaddress}" known | \
+        grep "${bdaddress}" )
+      if ! [ "${bthidd_known}" ]; then
+        printf "failed.\n"
+      else
+        printf "success.\nTo re-read its config, bthidd must be restarted.\n"
+        printf "Warning: If a Bluetooth keyboard is being used, the connection might be lost.\n"
+        printf "It can be manually restarted later with\n  service bthidd restart\n"
+        read -p "Restart bthidd now? [yes]: " REPLY
+        case "${REPLY}" in no|n|NO|N|No|nO) ;; *) /usr/sbin/service bthidd restart;; esac
+      fi
+    fi
+  ;;
+  esac
+fi
+
+}
+
+# After function definitions, main() can use them
+main "$@"
+
+exit
+
+# TODO
+# * If device is a keyboard, offer a text entry test field and if it does
+#   not succeed, leave some clues for debugging (i.e. if the node responds
+#   to pings, maybe switch keyboard on/off, etc)
+# * Same if device is a mouse, i.e. hexdump /dev/sysmouse.
+# * If device offers DUN profiles, ask the user if an entry in
+#   /etc/ppp/ppp.conf should be created
+# * If OPUSH or SPP is offered, refer to the respective man pages to give
+#   some clues how to continue