looking-glass/host/MultiMemcpy.cpp

/*
Looking Glass - KVM FrameRelay (KVMFR) Client
Copyright (C) 2017 Geoffrey McRae <geoff@hostfission.com>
https://looking-glass.hostfission.com

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 2 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, write to the Free Software Foundation, Inc., 59 Temple
Place, Suite 330, Boston, MA 02111-1307 USA
*/

#include "MultiMemcpy.h"
#include "common/memcpySSE.h"

MultiMemcpy::MultiMemcpy()
{
  for (int i = 0; i < MULTIMEMCPY_THREADS; ++i)
  {
    m_workers[i].id      = (1 << i);
    m_workers[i].running = &m_running;
    m_workers[i].abort   = false;
    m_workers[i].start   = CreateSemaphore(NULL, 0, 1, NULL);

    m_workers[i].thread = CreateThread(0, 0, WorkerFunction, &m_workers[i], 0, NULL);
  }
}

MultiMemcpy::~MultiMemcpy()
{
  for(int i = 0; i < MULTIMEMCPY_THREADS; ++i)
  {
    TerminateThread(m_workers[i].thread, 0);
    CloseHandle(m_workers[i].start);
  }
}

void MultiMemcpy::Copy(void * dst, void * src, size_t size)
{
  const size_t block = (size / MULTIMEMCPY_THREADS) & ~0x7F;
  if (block == 0)
  {
    Abort();
    memcpySSE(dst, src, size);
    return;
  }

  Wake();
  for (int i = 0; i < MULTIMEMCPY_THREADS; ++i)
  {
    m_workers[i].dst  = (uint8_t *)dst + i * block;
    m_workers[i].src  = (uint8_t *)src + i * block;
    if (i == MULTIMEMCPY_THREADS - 1)
      m_workers[i].size = size - (block * i);
    else
      m_workers[i].size = block;
  }

  INTERLOCKED_OR8(&m_running, (1 << MULTIMEMCPY_THREADS) - 1);
  while(m_running) {}

  m_awake = false;
}

DWORD WINAPI MultiMemcpy::WorkerFunction(LPVOID param)
{
  struct Worker * w = (struct Worker *)param;
  
  for(;;)
  {
    WaitForSingleObject(w->start, INFINITE);
    while(!(*w->running & w->id)) {}
    if (w->abort)
    {
      w->abort = false;
      INTERLOCKED_AND8(w->running, ~w->id);
      continue;
    }

    memcpySSE(w->dst, w->src, w->size);
    INTERLOCKED_AND8(w->running, ~w->id);
  }
}