acts_tests/acts_contrib/test_utils/bt/rfcomm_lib.py

#!/usr/bin/env python3
#
# Copyright (C) 2016 The Android Open Source Project
#
# 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
#
# http://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.
"""
Bluetooth adapter libraries
"""

from acts_contrib.test_utils.bt.bt_constants import bt_rfcomm_uuids
from acts_contrib.test_utils.bt.bt_test_utils import set_bt_scan_mode


class RfcommLib():
    def __init__(self, log, dut, target_mac_addr=None):
        self.advertisement_list = []
        self.dut = dut
        self.log = log
        self.target_mac_addr = target_mac_addr

    def set_target_mac_addr(self, mac_addr):
        self.target_mac_addr = mac_addr

    def connect(self, line):
        """Perform an RFCOMM connect"""
        uuid = None
        if len(line) > 0:
            uuid = line
        if uuid:
            self.dut.droid.bluetoothRfcommBeginConnectThread(
                self.target_mac_addr, uuid)
        else:
            self.dut.droid.bluetoothRfcommBeginConnectThread(
                self.target_mac_addr)

    def open_rfcomm_socket(self):
        """Open rfcomm socket"""
        self.dut.droid.rfcommCreateRfcommSocket(self.target_mac_addr, 1)

    def open_l2cap_socket(self):
        """Open L2CAP socket"""
        self.dut.droid.rfcommCreateL2capSocket(self.target_mac_addr, 1)

    def write(self, line):
        """Write String data over an RFCOMM connection"""
        self.dut.droid.bluetoothRfcommWrite(line)

    def write_binary(self, line):
        """Write String data over an RFCOMM connection"""
        self.dut.droid.bluetoothRfcommWriteBinary(line)

    def end_connect(self):
        """End RFCOMM connection"""
        self.dut.droid.bluetoothRfcommEndConnectThread()

    def accept(self, line):
        """Accept RFCOMM connection"""
        uuid = None
        if len(line) > 0:
            uuid = line
        if uuid:
            self.dut.droid.bluetoothRfcommBeginAcceptThread(uuid)
        else:
            self.dut.droid.bluetoothRfcommBeginAcceptThread(
                bt_rfcomm_uuids['base_uuid'])

    def stop(self):
        """Stop RFCOMM Connection"""
        self.dut.droid.bluetoothRfcommStop()

    def open_l2cap_socket(self):
        """Open L2CAP socket"""
        self.dut.droid.rfcommCreateL2capSocket(self.target_mac_addr, 1)