source: code/branches/wiimote/src/external/wiicpp/wiic/io_nix.c @ 10422

/*
 *    io_nix.c
 *
 *        This file is part of WiiC, written by:
 *              Gabriele Randelli
 *              Email: randelli@dis.uniroma1.it
 *
 *    Copyright 2010
 *              
 *        This file is based on Wiiuse, written By:
 *              Michael Laforest        < para >
 *              Email: < thepara (--AT--) g m a i l [--DOT--] com >
 *
 *        Copyright 2006-2007
 *
 *    This program is free software; you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation; either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 *        $Header$
 */

/**
 *      @file
 *      @brief Handles device I/O for *nix.
 */

#ifndef __APPLE__

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include <bluetooth/bluetooth.h>
#include <bluetooth/hci.h>
#include <bluetooth/hci_lib.h>
#include <bluetooth/l2cap.h>

#include "definitions.h"
#include "wiic_internal.h"
#include "io.h"

/**
 *      @brief Find a wiimote or wiimotes.
 *
 *      @param wm                       An array of wiimote_t structures.
 *      @param max_wiimotes     The number of wiimote structures in \a wm.
 *      @param timeout          The number of seconds before the search times out.
 *
 *      @return The number of wiimotes found.
 *
 *      @see wiimote_connect()
 *
 *      This function will only look for wiimote devices.                                               \n
 *      When a device is found the address in the structures will be set.               \n
 *      You can then call wiimote_connect() to connect to the found                             \n
 *      devices.
 */
int wiic_find(struct wiimote_t** wm, int max_wiimotes, int timeout) {
        int device_id;
        int device_sock;
        int found_devices;
        int found_wiimotes;

        /* reset all wiimote bluetooth device addresses */
        for (found_wiimotes = 0; found_wiimotes < max_wiimotes; ++found_wiimotes)
                wm[found_wiimotes]->bdaddr = *BDADDR_ANY;
        found_wiimotes = 0;

        /* get the id of the first bluetooth device. */
        device_id = hci_get_route(NULL);
        if (device_id < 0) {
                perror("hci_get_route");
                return 0;
        }

        /* create a socket to the device */
        device_sock = hci_open_dev(device_id);
        if (device_sock < 0) {
                perror("hci_open_dev");
                return 0;
        }

        inquiry_info scan_info_arr[128];
        inquiry_info* scan_info = scan_info_arr;
        memset(&scan_info_arr, 0, sizeof(scan_info_arr));

        /* scan for bluetooth devices for 'timeout' seconds */
        if(timeout)
                found_devices = hci_inquiry(device_id, timeout, 128, NULL, &scan_info, IREQ_CACHE_FLUSH);
        else {
                while(!(found_devices = hci_inquiry(device_id, 5, 128, NULL, &scan_info, IREQ_CACHE_FLUSH))) ; // Unlimited inquiry
        }
                
        if (found_devices < 0) {
                perror("hci_inquiry");
                return 0;
        }

        WIIC_INFO("Found %i bluetooth device(s).", found_devices);

        int i = 0;

        /* display discovered devices */
        for (; (i < found_devices) && (found_wiimotes < max_wiimotes); ++i) {
                if ((scan_info[i].dev_class[0] == WM_DEV_CLASS_0) &&
                        (scan_info[i].dev_class[1] == WM_DEV_CLASS_1) &&
                        (scan_info[i].dev_class[2] == WM_DEV_CLASS_2))
                {
                        /* found a device */
                        ba2str(&scan_info[i].bdaddr, wm[found_wiimotes]->bdaddr_str);

                        WIIC_INFO("Found wiimote (%s) [id %i].", wm[found_wiimotes]->bdaddr_str, wm[found_wiimotes]->unid);

                        wm[found_wiimotes]->bdaddr = scan_info[i].bdaddr;
                        WIIMOTE_ENABLE_STATE(wm[found_wiimotes], WIIMOTE_STATE_DEV_FOUND);
                        ++found_wiimotes;
                }
        }

        close(device_sock);
        return found_wiimotes;
}


/**
 *      @brief Connect to a wiimote or wiimotes once an address is known.
 *
 *      @param wm                       An array of wiimote_t structures.
 *      @param wiimotes         The number of wiimote structures in \a wm.
 *  @param autoreconnect        Re-connects the device in case of unexpected disconnection.
 *
 *      @return The number of wiimotes that successfully connected.
 *
 *      @see wiic_find()
 *      @see wiic_connect_single()
 *      @see wiic_disconnect()
 *
 *      Connect to a number of wiimotes when the address is already set
 *      in the wiimote_t structures.  These addresses are normally set
 *      by the wiic_find() function, but can also be set manually.
 */
int wiic_connect(struct wiimote_t** wm, int wiimotes, int autoreconnect) 
{
        int connected = 0;
        int i = 0;

        for (; i < wiimotes; ++i) {
                if (!WIIMOTE_IS_SET(wm[i], WIIMOTE_STATE_DEV_FOUND))
                        /* if the device address is not set, skip it */
                        continue;
                
                if (wiic_connect_single(wm[i], NULL, autoreconnect)) {
                        ++connected;
                }
        }

        return connected;
}

/**
 *      @brief Load Wii devices registered in the wiimotes.config file.
 *
 *      @param wm                       An array of wiimote_t structures.
 *
 *      @return The number of wiimotes successfully loaded.
 *
 *      @see wiic_find()
 *  @see wiic_connect()
 *      @see wiic_connect_single()
 *      @see wiic_disconnect()
 *
 *      Up to version 0.53, it is possible to register the MAC address of your 
 *  Wii devices. This allows to automatically load them, without waiting for any
 *  search timeout. To register a new device, go to: <HOME_DIR>/.wiic/ and
 *  edit the file wiimotes.config, by adding the MAC address of the device 
 *  you want to register (one line per MAC address).
 */
int wiic_load(struct wiimote_t** wm) 
{
        int loaded = 0;
        int i = 0;
        char str[200];
        char* str_ptr = 0;
        char configPath[100];
        char* tmp = 0;
        
        // Retrieve the HOME environment variable
        tmp = getenv("HOME");
        strcpy(configPath,tmp);
        strncat(configPath,"/.wiic/wiimotes.config",22);

        // Open the config file
        FILE* fd = 0;
        fd = fopen(configPath,"r");
        if(!fd) 
                return loaded;

        // Read line by line
        while(fgets(str,sizeof(str),fd) != NULL && loaded < 1) {
                int len = strlen(str)-1;
        if(str[len] == '\n') 
                str[len] = 0;
                loaded++;
        }
        
        // We initialize the device structure
        for (; i < loaded; ++i) {
                /* found a device */
                strncpy(wm[i]->bdaddr_str,str,18);
                str_ptr = str;
                str2ba(str_ptr,&(wm[i]->bdaddr));
                WIIMOTE_ENABLE_STATE(wm[i], WIIMOTE_STATE_DEV_FOUND);
                WIIC_INFO("Loaded Wiimote (%s) [id %i].",wm[i]->bdaddr_str,wm[i]->unid);
        }

        return loaded;
}

/**
 *      @brief Connect to a wiimote with a known address.
 *
 *      @param wm               Pointer to a wiimote_t structure.
 *      @param address  The address of the device to connect to.
 *                                      If NULL, use the address in the struct set by wiic_find().
 *      @param autoreconnect    Re-connect to the device in case of unexpected disconnection.
 *
 *      @return 1 on success, 0 on failure
 */
int wiic_connect_single(struct wiimote_t* wm, char* address, int autoreconnect) {
        struct sockaddr_l2 addr;
        memset(&addr, 0, sizeof(addr));

        if (!wm || WIIMOTE_IS_CONNECTED(wm))
                return 0;

        addr.l2_family = AF_BLUETOOTH;

        if (address) 
                /* use provided address */
                str2ba(address, &addr.l2_bdaddr);
        else
                /* use address of device discovered */
                addr.l2_bdaddr = wm->bdaddr;

        /*
         *      OUTPUT CHANNEL
         */
        wm->out_sock = socket(AF_BLUETOOTH, SOCK_SEQPACKET, BTPROTO_L2CAP);
        if (wm->out_sock == -1)
                return 0;

        addr.l2_psm = htobs(WM_OUTPUT_CHANNEL);

        /* connect to wiimote */
        if (connect(wm->out_sock, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
                perror("connect() output sock");
                return 0;
        }

        /*
         *      INPUT CHANNEL
         */
        wm->in_sock = socket(AF_BLUETOOTH, SOCK_SEQPACKET, BTPROTO_L2CAP);
        if (wm->in_sock == -1) {
                close(wm->out_sock);
                wm->out_sock = -1;
                return 0;
        }

        addr.l2_psm = htobs(WM_INPUT_CHANNEL);

        /* connect to wiimote */
        if (connect(wm->in_sock, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
                perror("connect() interrupt sock");
                close(wm->out_sock);
                wm->out_sock = -1;
                return 0;
        }
        
        /* autoreconnect flag */
        wm->autoreconnect = autoreconnect;

        WIIC_INFO("Connected to wiimote [id %i].", wm->unid);

        /* do the handshake */
        WIIMOTE_ENABLE_STATE(wm, WIIMOTE_STATE_CONNECTED);
        wiic_handshake(wm, NULL, 0);

        wiic_set_report_type(wm);

        return 1;
}


/**
 *      @brief Disconnect a wiimote.
 *
 *      @param wm               Pointer to a wiimote_t structure.
 *
 *      @see wiic_connect()
 *
 *      Note that this will not free the wiimote structure.
 */
void wiic_disconnect(struct wiimote_t* wm) {
        if (!wm || !WIIMOTE_IS_CONNECTED(wm))
                return;

        close(wm->out_sock);
        close(wm->in_sock);

        wm->out_sock = -1;
        wm->in_sock = -1;
        wm->event = WIIC_NONE;

        WIIMOTE_DISABLE_STATE(wm, WIIMOTE_STATE_CONNECTED);
        WIIMOTE_DISABLE_STATE(wm, WIIMOTE_STATE_HANDSHAKE);
}


int wiic_io_read(struct wiimote_t* wm) {
        /* not used */
        return 0;
}


int wiic_io_write(struct wiimote_t* wm, byte* buf, int len) {
        return write(wm->out_sock, buf, len);
}


#endif /* ifndef __APPLE__ */