wiiuse/io_wii.c

   1 #ifdef GEKKO
   2
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <unistd.h>
   6 #include <string.h>
   7
   8 #include "definitions.h"
   9 #include "wiiuse_internal.h"
  10 #include "events.h"
  11 #include "io.h"
  12 #include "lwp_wkspace.h"
  13
  14 #define MAX_COMMANDS                                    0x100
  15
  16 static vu32* const _ipcReg = (u32*)0xCD000000;
  17
  18 extern void parse_event(struct wiimote_t *wm);
  19 extern void idle_cycle(struct wiimote_t* wm);
  20 extern void hexdump(void *d, int len);
  21
  22 static __inline__ u32 ACR_ReadReg(u32 reg)
  23 {
  24         return _ipcReg[reg>>2];
  25 }
  26
  27 static __inline__ void ACR_WriteReg(u32 reg,u32 val)
  28 {
  29         _ipcReg[reg>>2] = val;
  30 }
  31
  32 static s32 __wiiuse_disconnected(void *arg,struct bte_pcb *pcb,u8 err)
  33 {
  34         struct wiimote_listen_t *wml = (struct wiimote_listen_t*)arg;
  35         struct wiimote_t *wm = wml->wm;
  36
  37         if(!wm) return ERR_OK;
  38
  39         //printf("wiimote disconnected\n");
  40         WIIMOTE_DISABLE_STATE(wm, (WIIMOTE_STATE_IR|WIIMOTE_STATE_IR_INIT));
  41         WIIMOTE_DISABLE_STATE(wm, (WIIMOTE_STATE_SPEAKER|WIIMOTE_STATE_SPEAKER_INIT));
  42         WIIMOTE_DISABLE_STATE(wm, (WIIMOTE_STATE_EXP|WIIMOTE_STATE_EXP_HANDSHAKE|WIIMOTE_STATE_EXP_FAILED));
  43         WIIMOTE_DISABLE_STATE(wm,(WIIMOTE_STATE_CONNECTED|WIIMOTE_STATE_HANDSHAKE|WIIMOTE_STATE_HANDSHAKE_COMPLETE));
  44
  45         while(wm->cmd_head) {
  46                 __lwp_queue_append(&wm->cmdq,&wm->cmd_head->node);
  47                 wm->cmd_head = wm->cmd_head->next;
  48         }
  49         wm->cmd_tail = NULL;
  50
  51         if(wm->event_cb) wm->event_cb(wm,WIIUSE_DISCONNECT);
  52
  53         wml->wm = NULL;
  54         return ERR_OK;
  55 }
  56
  57 static s32 __wiiuse_receive(void *arg,void *buffer,u16 len)
  58 {
  59         struct wiimote_listen_t *wml = (struct wiimote_listen_t*)arg;
  60         struct wiimote_t *wm = wml->wm;
  61
  62         if(!wm || !buffer || len==0) return ERR_OK;
  63
  64         //printf("__wiiuse_receive[%02x]\n",*(char*)buffer);
  65         wm->event = WIIUSE_NONE;
  66
  67         memcpy(wm->event_buf,buffer,len);
  68         memset(&(wm->event_buf[len]),0,(MAX_PAYLOAD - len));
  69         parse_event(wm);
  70
  71         if(wm->event!=WIIUSE_NONE) {
  72                 if(wm->event_cb) wm->event_cb(wm,wm->event);
  73         }
  74
  75         return ERR_OK;
  76 }
  77
  78 static s32 __wiiuse_connected(void *arg,struct bte_pcb *pcb,u8 err)
  79 {
  80         struct wiimote_listen_t *wml = (struct wiimote_listen_t*)arg;
  81         struct wiimote_t *wm;
  82
  83         wm = wml->assign_cb(&wml->bdaddr);
  84
  85         if(!wm) {
  86                 bte_disconnect(wml->sock);
  87                 return ERR_OK;
  88         }
  89
  90         wml->wm = wm;
  91
  92         wm->sock = wml->sock;
  93         wm->bdaddr = wml->bdaddr;
  94
  95         //printf("__wiiuse_connected()\n");
  96         WIIMOTE_ENABLE_STATE(wm,(WIIMOTE_STATE_CONNECTED|WIIMOTE_STATE_HANDSHAKE));
  97
  98         wm->handshake_state = 0;
  99         wiiuse_handshake(wm,NULL,0);
 100
 101         return ERR_OK;
 102 }
 103
 104 static s32 __wiiuse_sent(void *arg,struct bte_pcb *pcb,u8 err)
 105 {
 106         struct cmd_blk_t *cmd = NULL;
 107         struct wiimote_listen_t *wml = (struct wiimote_listen_t*)arg;
 108         struct wiimote_t *wm = wml->wm;
 109
 110         if(!wm) return ERR_OK;
 111
 112         cmd = wm->cmd_head;
 113
 114         if(!cmd) return ERR_OK;
 115         if(cmd->state!=CMD_SENT) return ERR_OK;
 116
 117         switch(cmd->data[0]) {
 118                 case WM_CMD_CTRL_STATUS:
 119                 case WM_CMD_WRITE_DATA:
 120                 case WM_CMD_READ_DATA:
 121                         return ERR_OK;
 122                 default:
 123                         wm->cmd_head = cmd->next;
 124
 125                         cmd->state = CMD_DONE;
 126                         if(cmd->cb) cmd->cb(wm,NULL,0);
 127
 128                         __lwp_queue_append(&wm->cmdq,&cmd->node);
 129                         wiiuse_send_next_command(wm);
 130                         break;
 131         }
 132         return ERR_OK;
 133 }
 134
 135 void __wiiuse_sensorbar_enable(int enable)
 136 {
 137         u32 val;
 138         u32 level;
 139
 140         level = IRQ_Disable();
 141         val = (ACR_ReadReg(0xc0)&~0x100);
 142         if(enable) val |= 0x100;
 143         ACR_WriteReg(0xc0,val);
 144         IRQ_Restore(level);
 145 }
 146
 147 int wiiuse_register(struct wiimote_listen_t *wml, struct bd_addr *bdaddr, struct wiimote_t *(*assign_cb)(struct bd_addr *bdaddr))
 148 {
 149         s32 err;
 150
 151         if(!wml || !bdaddr || !assign_cb) return 0;
 152
 153         wml->wm = NULL;
 154         wml->bdaddr = *bdaddr;
 155         wml->sock = bte_new();
 156         wml->assign_cb = assign_cb;
 157         if(wml->sock==NULL) return 0;
 158
 159         bte_arg(wml->sock,wml);
 160         bte_received(wml->sock,__wiiuse_receive);
 161         bte_disconnected(wml->sock,__wiiuse_disconnected);
 162
 163         err = bte_registerdeviceasync(wml->sock,bdaddr,__wiiuse_connected);
 164         if(err==ERR_OK) return 1;
 165
 166         return 0;
 167 }
 168
 169 void wiiuse_disconnect(struct wiimote_t *wm)
 170 {
 171         if(wm==NULL || wm->sock==NULL) return;
 172
 173         WIIMOTE_DISABLE_STATE(wm,WIIMOTE_STATE_CONNECTED);
 174         bte_disconnect(wm->sock);
 175 }
 176
 177 void wiiuse_sensorbar_enable(int enable)
 178 {
 179         __wiiuse_sensorbar_enable(enable);
 180 }
 181
 182 void wiiuse_init_cmd_queue(struct wiimote_t *wm)
 183 {
 184         u8 *buffer;
 185         u32 size;
 186
 187         size = (MAX_COMMANDS*sizeof(struct cmd_blk_t));
 188         buffer = __lwp_wkspace_allocate(size);
 189         __lwp_queue_initialize(&wm->cmdq,buffer,MAX_COMMANDS,sizeof(struct cmd_blk_t));
 190 }
 191
 192 int wiiuse_io_write(struct wiimote_t *wm,ubyte *buf,int len)
 193 {
 194         if(wm->sock)
 195                 return bte_sendmessageasync(wm->sock,buf,len,__wiiuse_sent);
 196
 197         return ERR_CONN;
 198 }
 199
 200 #endif