kernel/drivers/char/mouse/ps2.c

   1 /*
   2  *  ZeX/OS
   3  *  Copyright (C) 2009  Martin 'povik' Poviser (martin.povik@gmail.com)
   4  *
   5  *  This program is free software: you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation, either version 3 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #include <build.h>
  20
  21 #ifdef ARCH_i386
  22
  23 #include <system.h>
  24 #include <arch/io.h>
  25 #include <mouse.h>
  26
  27 #define PS2MOUSE_SAMPLES 0x64   // Samples in second - 0x50 for 80, 0x64 for 100, 0xC8 for 200
  28 #define PS2MOUSE_RESOLUTION 2
  29
  30 #define PS2MOUSE_TIMEOUT                500000
  31 #define PS2MOUSE_ACK                    0xFA
  32 #define PS2MOUSE_PORT                   0x60
  33 #define PS2MOUSE_CTRL                   0x64
  34 #define PS2MOUSE_COMMAND                0xD4
  35 #define PS2MOUSE_COMMAND_RESET          0xFF
  36 #define PS2MOUSE_COMMAND_GET_MOUSE_ID   0xF2
  37 #define PS2MOUSE_COMMAND_GET_PACKET     0xEB
  38 #define PS2MOUSE_COMMAND_ENABLE_PACKETS 0xF4
  39 #define PS2MOUSE_COMMAND_SET_SAMPLERATE 0xF3
  40 #define PS2MOUSE_COMMAND_SET_RESOLUTION 0xE8
  41 #define PS2MOUSE_COMMAND_SET_REMOTE     0xF0
  42 #define PS2MOUSE_RESET_ACK              0xAA
  43 #define PS2MOUSE_IRQ                    12
  44
  45 #define PS2MOUSE_SAMPLERATE_80          0x50
  46 #define PS2MOUSE_SAMPLERATE_100         0x64
  47 #define PS2MOUSE_SAMPLERATE_200         0xC8
  48
  49 #define PS2MOUSE_SAMPLERATE     PS2MOUSE_SAMPLERATE_100
  50
  51
  52 bool ps2mouse_extwheel = 0;
  53 bool ps2mouse_extbuttons = 0;
  54
  55 short mouse_id;
  56
  57 dev_mouse_t ps2mouse_state;
  58
  59 bool ps2mouse_reset ();
  60 bool ps2mouse_write (unsigned char command);
  61 short ps2mouse_read ();
  62
  63 /* ps2mouse_reset - reset mouse */
  64
  65 bool ps2mouse_reset ()
  66 {
  67         int timeout;
  68
  69         for (timeout = 0; timeout < PS2MOUSE_TIMEOUT; timeout ++)
  70         {
  71                 if((inb (PS2MOUSE_CTRL) && 0x02) != 0x02)
  72                         break;
  73                 schedule ();
  74         }
  75
  76         outb (PS2MOUSE_CTRL, PS2MOUSE_COMMAND);
  77         outb (PS2MOUSE_PORT, PS2MOUSE_COMMAND_RESET);
  78
  79         for (timeout = 0; timeout < PS2MOUSE_TIMEOUT; timeout ++)
  80         {
  81                 int read = ps2mouse_read ();
  82                 if (read == PS2MOUSE_RESET_ACK)
  83                         return 1;
  84
  85                 if (read == -1)
  86                         return 0;
  87
  88                 schedule ();
  89         }
  90
  91         return 0;
  92 }
  93
  94 /* ps2mouse_write - send a command to mouse */
  95
  96 bool ps2mouse_write (unsigned char command)
  97 {
  98         int timeout;
  99
 100         for (timeout = 0; timeout < PS2MOUSE_TIMEOUT; timeout ++)
 101         {
 102                 if((inb (PS2MOUSE_CTRL) && 0x02) != 0x02)
 103                         break;
 104                 schedule ();
 105         }
 106
 107         outb (PS2MOUSE_CTRL, PS2MOUSE_COMMAND);
 108         outb (PS2MOUSE_PORT, command);
 109
 110         short read = ps2mouse_read ();
 111
 112         if (read == PS2MOUSE_ACK)
 113                 return 1;
 114
 115         if (read == -1)
 116                 return 0;
 117
 118         return 0;
 119 }
 120
 121 /* ps2mouse_read - read a value from mouse */
 122
 123 short ps2mouse_read()
 124 {
 125         int timeout;
 126
 127         for (timeout = 0; timeout < PS2MOUSE_TIMEOUT; timeout ++)
 128         {
 129                 if ((inb (PS2MOUSE_CTRL) && 0x01) == 0x01)
 130                         return inb (PS2MOUSE_PORT);
 131                 schedule ();
 132         }
 133
 134         return -1;
 135 }
 136
 137 /* ps2mouse_get_state - get a mouse state */
 138
 139 bool ps2mouse_get_state ()
 140 {
 141         unsigned char b1, b2, b3, b4 = 0;
 142
 143         if (ps2mouse_write (PS2MOUSE_COMMAND_GET_PACKET) == -1)
 144                 return 0;
 145
 146         b1 = ps2mouse_read ();
 147         b2 = ps2mouse_read ();
 148         b3 = ps2mouse_read ();
 149
 150         if (ps2mouse_extwheel || ps2mouse_extbuttons)
 151                 b4 = ps2mouse_read ();
 152
 153         ps2mouse_state.flags = 0;
 154         ps2mouse_state.flags |= ((b1 & 1) ? MOUSE_FLAG_BUTTON1 : 0);
 155         ps2mouse_state.flags |= ((b1 & 2) ? MOUSE_FLAG_BUTTON2 : 0);
 156         ps2mouse_state.flags |= ((b1 & 3) ? MOUSE_FLAG_BUTTON3 : 0);
 157         ps2mouse_state.flags |= ((b1 & 4) ? MOUSE_FLAG_BUTTON4 : 0);
 158         ps2mouse_state.flags |= ((b1 & 5) ? MOUSE_FLAG_BUTTON5 : 0);
 159
 160         short dx, dy, dz;
 161
 162         dx = (b1 & 0x10) ? b2 - 256 : b2;
 163         dy = (b1 & 0x20) ? -(b3 - 256) : -b3;
 164         dz = (b4 & 0x08) ? (b4 & 7) - 8 : b4 & 7;
 165
 166         if (dx > 5 || dx < -5)
 167                 dx *= 4;
 168         if (dy > 5 || dy < -5)
 169                 dy *= 4;
 170
 171         ps2mouse_state.pos_x = dx;
 172         ps2mouse_state.pos_y = dy;
 173
 174         if (dz > 0)
 175                 ps2mouse_state.flags = ps2mouse_state.flags | MOUSE_FLAG_SCROLLUP;
 176         if (dz < 0)
 177                 ps2mouse_state.flags = ps2mouse_state.flags | MOUSE_FLAG_SCROLLDOWN;
 178
 179         return 1;
 180 }
 181
 182 void ps2mouse_get_mouseid ()
 183 {
 184         ps2mouse_write (PS2MOUSE_COMMAND_SET_SAMPLERATE);
 185         ps2mouse_write (PS2MOUSE_SAMPLERATE_200);
 186
 187         ps2mouse_write (PS2MOUSE_COMMAND_SET_SAMPLERATE);
 188         ps2mouse_write (PS2MOUSE_SAMPLERATE_100);
 189
 190         ps2mouse_write (PS2MOUSE_COMMAND_SET_SAMPLERATE);
 191         ps2mouse_write (PS2MOUSE_SAMPLERATE_80);
 192
 193         ps2mouse_write (PS2MOUSE_COMMAND_GET_MOUSE_ID);
 194         mouse_id = ps2mouse_read ();
 195
 196         if (mouse_id < 3)
 197                 return;
 198         else
 199                 ps2mouse_extwheel = true;
 200
 201         ps2mouse_write (PS2MOUSE_COMMAND_SET_SAMPLERATE);
 202         ps2mouse_write (PS2MOUSE_SAMPLERATE_200);
 203
 204         ps2mouse_write (PS2MOUSE_COMMAND_SET_SAMPLERATE);
 205         ps2mouse_write (PS2MOUSE_SAMPLERATE_200);
 206
 207         ps2mouse_write (PS2MOUSE_COMMAND_SET_SAMPLERATE);
 208         ps2mouse_write (PS2MOUSE_SAMPLERATE_80);
 209
 210         ps2mouse_write (PS2MOUSE_COMMAND_GET_MOUSE_ID);
 211         mouse_id = ps2mouse_read ();
 212
 213         if (mouse_id > 3)
 214                 ps2mouse_extbuttons = true;
 215 }
 216
 217 unsigned ps2mouse_init ()
 218 {
 219         int timeout;
 220
 221         ps2mouse_extbuttons = false;
 222         ps2mouse_extwheel = false;
 223
 224         if(ps2mouse_reset () == 0)
 225                 return 0;
 226
 227         ps2mouse_read (); // Mouse send her ID
 228
 229         ps2mouse_get_mouseid ();
 230
 231         if (mouse_id == -1)
 232                 return 0;
 233
 234         ps2mouse_write (PS2MOUSE_COMMAND_SET_REMOTE);
 235         ps2mouse_write (PS2MOUSE_COMMAND_ENABLE_PACKETS);
 236
 237         return 1;
 238 }
 239
 240 bool ps2mouse_acthandler (unsigned act, char *block, unsigned block_len)
 241 {
 242         switch (act) {
 243                 case DEV_ACT_INIT:
 244                 {
 245                         return ps2mouse_init ();
 246                 }
 247                 break;
 248                 case DEV_ACT_READ:
 249                 {
 250                         if(!ps2mouse_get_state ())
 251                                 return 0;
 252                         memcpy (block, &ps2mouse_state, sizeof (dev_mouse_t));
 253                         return 1;
 254                 }
 255                 break;
 256                 case DEV_ACT_WRITE:
 257                 {
 258                         return 0;
 259                 }
 260                 break;
 261         }
 262
 263         return 0;
 264 }
 265
 266 #endif