| #!/bin/bash |
| # |
| # Copyright 2019 Google LLC |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # https://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| set -e |
| |
| readonly RULES_FILE=/etc/udev/rules.d/99-edgetpu-accelerator.rules |
| |
| if [[ ! -x "$(command -v udevadm)" ]]; then |
| echo "Your system does not support device rules" |
| exit 1 |
| fi |
| |
| if [[ "${EUID}" != 0 ]]; then |
| echo "Please use sudo to run as root" |
| exit 1 |
| fi |
| |
| if [[ "$1" == "install" ]]; then |
| cat << EOF > "${RULES_FILE}" |
| SUBSYSTEM=="usb",ATTRS{idVendor}=="1a6e",ATTRS{idProduct}=="089a",GROUP="plugdev" |
| SUBSYSTEM=="usb",ATTRS{idVendor}=="18d1",ATTRS{idProduct}=="9302",GROUP="plugdev" |
| EOF |
| udevadm control --reload-rules && udevadm trigger |
| cat "${RULES_FILE}" |
| elif [[ "$1" == "uninstall" ]]; then |
| rm -f "${RULES_FILE}" |
| udevadm control --reload-rules && udevadm trigger |
| else |
| echo "$0 install|uninstall" |
| exit 1 |
| fi |
| |