mf_info.c

 
#include <freerdp/config.h>
 
#include <stdlib.h>
#include <errno.h>
 
#include "mf_info.h"
#include "mf_mountain_lion.h"
 
#define MF_INFO_DEFAULT_FPS 30
#define MF_INFO_MAXPEERS 32
 
static mfInfo* mfInfoInstance = nullptr;
 
int mf_info_lock(mfInfo* mfi)
{
  int status = pthread_mutex_lock(&mfi->mutex);
 
  switch (status)
  {
    case 0:
      return TRUE;
      break;
 
    default:
      return -1;
      break;
  }
 
  return 1;
}
 
int mf_info_try_lock(mfInfo* mfi, UINT32 ms)
{
  int status = pthread_mutex_trylock(&mfi->mutex);
 
  switch (status)
  {
    case 0:
      return TRUE;
      break;
 
    case EBUSY:
      return FALSE;
      break;
 
    default:
      return -1;
      break;
  }
 
  return 1;
}
 
int mf_info_unlock(mfInfo* mfi)
{
  int status = pthread_mutex_unlock(&mfi->mutex);
 
  switch (status)
  {
    case 0:
      return TRUE;
      break;
 
    default:
      return -1;
      break;
  }
 
  return 1;
}
 
WINPR_ATTR_NODISCARD
static mfInfo* mf_info_init(void)
{
  mfInfo* mfi = (mfInfo*)calloc(1, sizeof(mfInfo));
 
  if (mfi != nullptr)
  {
    pthread_mutex_init(&mfi->mutex, nullptr);
 
    mfi->peers = (freerdp_peer**)calloc(MF_INFO_MAXPEERS, sizeof(freerdp_peer*));
    if (!mfi->peers)
    {
      free(mfi);
      return nullptr;
    }
 
    mfi->framesPerSecond = MF_INFO_DEFAULT_FPS;
    mfi->input_disabled = FALSE;
  }
 
  return mfi;
}
 
mfInfo* mf_info_get_instance(void)
{
  if (mfInfoInstance == nullptr)
    mfInfoInstance = mf_info_init();
 
  return mfInfoInstance;
}
 
void mf_info_peer_register(mfInfo* mfi, mfPeerContext* context)
{
  if (mf_info_lock(mfi) > 0)
  {
    int peerId;
 
    if (mfi->peerCount == MF_INFO_MAXPEERS)
    {
      mf_info_unlock(mfi);
      return;
    }
 
    context->info = mfi;
 
    if (mfi->peerCount == 0)
    {
      mf_mlion_display_info(&mfi->servscreen_width, &mfi->servscreen_height, &mfi->scale);
      mf_mlion_screen_updates_init();
      mf_mlion_start_getting_screen_updates();
    }
 
    peerId = 0;
 
    for (int i = 0; i < MF_INFO_MAXPEERS; ++i)
    {
      // empty index will be our peer id
      if (mfi->peers[i] == nullptr)
      {
        peerId = i;
        break;
      }
    }
 
    mfi->peers[peerId] = ((rdpContext*)context)->peer;
    mfi->peers[peerId]->pId = peerId;
    mfi->peerCount++;
 
    mf_info_unlock(mfi);
  }
}
 
void mf_info_peer_unregister(mfInfo* mfi, mfPeerContext* context)
{
  if (mf_info_lock(mfi) > 0)
  {
    int peerId;
 
    peerId = ((rdpContext*)context)->peer->pId;
    mfi->peers[peerId] = nullptr;
    mfi->peerCount--;
 
    if (mfi->peerCount == 0)
      mf_mlion_stop_getting_screen_updates();
 
    mf_info_unlock(mfi);
  }
}
 
BOOL mf_info_have_updates(mfInfo* mfi)
{
  if (mfi->framesWaiting == 0)
    return FALSE;
 
  return TRUE;
}
 
void mf_info_update_changes(mfInfo* mfi)
{
}
 
void mf_info_find_invalid_region(mfInfo* mfi)
{
  mf_mlion_get_dirty_region(&mfi->invalid);
}
 
void mf_info_clear_invalid_region(mfInfo* mfi)
{
  mf_mlion_clear_dirty_region();
  mfi->invalid.height = 0;
  mfi->invalid.width = 0;
}
 
void mf_info_invalidate_full_screen(mfInfo* mfi)
{
  mfi->invalid.x = 0;
  mfi->invalid.y = 0;
  mfi->invalid.height = mfi->servscreen_height;
  mfi->invalid.width = mfi->servscreen_width;
}
 
BOOL mf_info_have_invalid_region(mfInfo* mfi)
{
  if (mfi->invalid.width * mfi->invalid.height == 0)
    return FALSE;
 
  return TRUE;
}
 
void mf_info_getScreenData(mfInfo* mfi, long* width, long* height, BYTE** pBits, int* pitch)
{
  *width = mfi->invalid.width / mfi->scale;
  *height = mfi->invalid.height / mfi->scale;
  *pitch = mfi->servscreen_width * mfi->scale * 4;
 
  mf_mlion_get_pixelData(mfi->invalid.x / mfi->scale, mfi->invalid.y / mfi->scale, *width,
                         *height, pBits);
 
  *pBits = *pBits + (mfi->invalid.x * 4) + (*pitch * mfi->invalid.y);
}