Ps2Mouse.cc

/*
 * Copyright (c) 2008-2014, Pedigree Developers
 *
 * Please see the CONTRIB file in the root of the source tree for a full
 * list of contributors.
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */
 
#include "Ps2Mouse.h"
#include "pedigree/kernel/machine/InputManager.h"
#include "pedigree/kernel/process/Thread.h"
#include "pedigree/kernel/processor/Processor.h"
#include "pedigree/kernel/processor/ProcessorInformation.h"
#include "pedigree/kernel/time/Time.h"
#include "pedigree/kernel/utilities/new"
 
#include "system/kernel/machine/mach_pc/Ps2Controller.h"
 
class Process;
 
Ps2Mouse::Ps2Mouse(Device* pDev)
    : m_pController(0), m_PacketDecoder(), m_IrqWait(0), m_ReaderThread(), m_Callbacks() {
  setSpecificType(String("ps2-mouse"));
}
 
Ps2Mouse::~Ps2Mouse() {
  m_ReaderThread.stop();
}
 
bool Ps2Mouse::initialise(Ps2Controller* pController) {
  if (!pController || !pController->hasSecondPort())
    return false;
  m_pController = pController;
 
  // Command responses use the controller's split-IRQ buffer once its worker
  // is active, so IRQ12 must be enabled before waiting for either ACK.
  m_pController->setIrqEnable(true, true);
 
  // Missing devices and lost command replies must not block module loading.
  const uint8_t commands[] = {SetDefaults, MouseStream};
  for (uint8_t command : commands) {
    bool acknowledged = false;
    for (size_t retry = 0; retry < 3 && !acknowledged; ++retry) {
      m_pController->writeSecondPort(command);
      uint8_t response = 0;
      bool received = false;
      for (size_t poll = 0; poll < 250 && !m_pController->readsStopping(); ++poll) {
        if (m_pController->readSecondPort(response, false)) {
          received = true;
          break;
        }
        Time::delay(Time::Multiplier::Millisecond);
      }
      acknowledged = received && response == MouseAck;
      if (!received || (response != MouseAck && response != 0xfe))
        break;
    }
    if (!acknowledged) {
      m_pController->setIrqEnable(true, false);
      WARNING("PS/2 mouse did not acknowledge command " << Hex << command);
      return false;
    }
  }
 
  Process* pProcess = Processor::information().getCurrentThread()->getParent();
  Thread* pThread = new Thread(pProcess, readerThreadTrampoline, this);
  pThread->setName("PS/2 mouse reader");
  m_ReaderThread.adopt(pThread);
 
  return true;
}
 
void Ps2Mouse::write(const char* bytes, size_t len) {
  for (size_t i = 0; i < len; ++i) {
    m_pController->writeSecondPort(bytes[i]);
  }
}
 
bool Ps2Mouse::subscribe(MouseHandlerFunction handler, void* param, Registration& registration) {
  return m_Callbacks.subscribe(handler, param, registration);
}
 
void Ps2Mouse::updateSubscribers(const void* buffer, size_t len) {
  m_Callbacks.dispatch(buffer, len);
}
 
int Ps2Mouse::readerThreadTrampoline(void* param) {
  Ps2Mouse* instance = reinterpret_cast<Ps2Mouse*>(param);
  instance->readerThread();
  return 0;
}
 
void Ps2Mouse::readerThread() {
  while (true) {
    uint8_t byte;
    if (!m_pController->readSecondPort(byte)) {
      Thread* thread = Processor::information().getCurrentThread();
      if (m_pController->readsStopping() ||
          (thread && thread->getUnwindState() != Thread::Continue)) {
        return;
      }
      continue;
    }
 
    updateSubscribers(&byte, 1);
 
    Ps2MousePacket packet = {};
    if (m_PacketDecoder.feed(byte, packet)) {
      InputManager::instance().mouseUpdate(packet.relativeX, packet.relativeY, 0, packet.buttons);
    }
  }
}