/*
  VitaShell
  Copyright (C) 2015-2018, TheFloW

  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 3 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, see <http://www.gnu.org/licenses/>.
*/

#include "main.h"
#include "init.h"
#include "theme.h"
#include "language.h"
#include "utils.h"
#include "adhoc_dialog.h"
#include "message_dialog.h"
#include "uncommon_dialog.h"
#include "io_process.h"

#define SHARE_MAGIC 0x574F4C46
#define SHARE_TYPE_FOLDER 0
#define SHARE_TYPE_FILE 1

// 700 KB/s
#define SOCKET_BUFSIZE 128 * 1024
#define SHARE_SIZE 4096
#define RECEIVE_WAIT 100 * 1000

#define PEERLIST_MAX 4

typedef struct {
  int status;
  int result;
  int sel;
  float x;
  float y;
  float width;
  float height;
  float scale;
} AdhocDialog;

static AdhocDialog adhoc_dialog;

static SceNetAdhocctlPeerInfo  peer_list[PEERLIST_MAX];
static int peer_count = 0;

static SceNetAdhocctlPeerInfo server_peer_info;
static SceNetEtherAddr client_addr, server_addr;
static int client_socket = -1, server_socket = -1;
static SceUID client_waiting_thid = -1;
static int server_request_result = 0;
static char client_response[4];

typedef struct {
  int magic;
  int type;
  size_t path_len;
  uint64_t file_size;
} ShareInfo;

int adhocSend(int socket, const void *buf, int size) {
  return sceNetAdhocPtpSend(socket, buf, &size, 0, 0);
}

int adhocRecv(int socket, void *buf, int *size) {
  return sceNetAdhocPtpRecv(socket, buf, size, 0, SCE_NET_ADHOC_F_NONBLOCK);
}

int sendFile(const char *src_path, FileProcessParam *param) {
  SceUID fdsrc = sceIoOpen(src_path, SCE_O_RDONLY, 0);
  if (fdsrc < 0)
    return fdsrc;

  SceOff size = sceIoLseek(fdsrc, 0, SCE_SEEK_END);
  sceIoLseek(fdsrc, 0, SCE_SEEK_SET);

  // Send info
  ShareInfo info;
  info.magic = SHARE_MAGIC;
  info.type = SHARE_TYPE_FILE;
  info.path_len = strlen(src_path);
  info.file_size = size;
  int ret = adhocSend(client_socket, &info, sizeof(ShareInfo));
  if (ret < 0) {
    sceIoDclose(fdsrc);
    return ret;
  }
  
  // Send path
  ret = adhocSend(client_socket, src_path, info.path_len);
  if (ret < 0) {
    sceIoDclose(fdsrc);
    return ret;
  }
  
  void *buf = memalign(4096, SHARE_SIZE);
  
  while (1) {
    int read = sceIoRead(fdsrc, buf, SHARE_SIZE);

    if (read < 0) {
      free(buf);
      sceIoClose(fdsrc);
      return read;
    }

    if (read == 0)
      break;

    int sended = adhocSend(client_socket, buf, read);
    if (sended < 0) {
      free(buf);
      sceIoClose(fdsrc);
      return sended;
    }

    if (param) {
      if (param->value)
        (*param->value) += read;

      if (param->SetProgress)
        param->SetProgress(param->value ? *param->value : 0, param->max);

      if (param->cancelHandler && param->cancelHandler()) {
        free(buf);
        sceIoClose(fdsrc);
        return 0;
      }
    }
  }

  free(buf);
  sceIoClose(fdsrc);

  return 1;
}

int sendPath(const char *src_path, FileProcessParam *param) {  
  SceUID dfd = sceIoDopen(src_path);
  if (dfd >= 0) {
    // Send info
    ShareInfo info;
    info.magic = SHARE_MAGIC;
    info.type = SHARE_TYPE_FOLDER;
    info.path_len = strlen(src_path);
    info.file_size = 0;
    int ret = adhocSend(client_socket, &info, sizeof(ShareInfo));
    if (ret < 0) {
      sceIoDclose(dfd);
      return ret;
    }
    
    // Send path
    ret = adhocSend(client_socket, src_path, info.path_len);
    if (ret < 0) {
      sceIoDclose(dfd);
      return ret;
    }
    
    if (param) {
      if (param->value)
        (*param->value) += DIRECTORY_SIZE;

      if (param->SetProgress)
        param->SetProgress(param->value ? *param->value : 0, param->max);

      if (param->cancelHandler && param->cancelHandler()) {
        sceIoDclose(dfd);
        return 0;
      }
    }

    int res = 0;

    do {
      SceIoDirent dir;
      memset(&dir, 0, sizeof(SceIoDirent));

      res = sceIoDread(dfd, &dir);
      if (res > 0) {
        char *new_src_path = malloc(strlen(src_path) + strlen(dir.d_name) + 2);
        snprintf(new_src_path, MAX_PATH_LENGTH, "%s%s%s", src_path, hasEndSlash(src_path) ? "" : "/", dir.d_name);

        int ret = 0;

        if (SCE_S_ISDIR(dir.d_stat.st_mode)) {
          ret = sendPath(new_src_path, param);
        } else {
          ret = sendFile(new_src_path, param);
        }

        free(new_src_path);

        if (ret <= 0) {
          sceIoDclose(dfd);
          return ret;
        }
      }
    } while (res > 0);

    sceIoDclose(dfd);
  } else {
    return sendFile(src_path, param);
  }

  return 1;
}

int send_thread(SceSize args_size, SendArguments *args) {
  int res;
  SceUID thid = -1;

  // Lock power timers
  powerLock();

  // Set progress to 0%
  sceMsgDialogProgressBarSetValue(SCE_MSG_DIALOG_PROGRESSBAR_TARGET_BAR_DEFAULT, 0);
  sceKernelDelayThread(DIALOG_WAIT); // Needed to see the percentage

  FileListEntry *file_entry = fileListGetNthEntry(args->file_list, args->index);

  int count = 0;
  FileListEntry *head = NULL;
  FileListEntry *mark_entry_one = NULL;

  if (fileListFindEntry(args->mark_list, file_entry->name)) { // On marked entry
    count = args->mark_list->length;
    head = args->mark_list->head;
  } else {
    count = 1;
    mark_entry_one = fileListCopyEntry(file_entry);
    head = mark_entry_one;
  }

  char path[MAX_PATH_LENGTH];
  FileListEntry *mark_entry = NULL;

  // Get paths info
  uint64_t size = 0;
  uint32_t folders = 0, files = 0;

  mark_entry = head;

  int i;
  for (i = 0; i < count; i++) {
    snprintf(path, MAX_PATH_LENGTH, "%s%s", args->file_list->path, mark_entry->name);
    getPathInfo(path, &size, &folders, &files, NULL);
    mark_entry = mark_entry->next;
  }

  // Send total size and path length of cwd
  ShareInfo info;
  info.magic = SHARE_MAGIC;
  info.path_len = strlen(args->file_list->path);
  info.file_size = size + folders * DIRECTORY_SIZE;
  res = adhocSend(client_socket, &info, sizeof(ShareInfo));
  if (res < 0) {
    goto CANCELED;
  }

  // Update thread
  thid = createStartUpdateThread(size + folders * DIRECTORY_SIZE, 1);

  // Send process
  uint64_t value = 0;

  mark_entry = head;

  for (i = 0; i < count; i++) {
    snprintf(path, MAX_PATH_LENGTH, "%s%s", args->file_list->path, mark_entry->name);

    FileProcessParam param;
    param.value = &value;
    param.max = folders + files;
    param.SetProgress = SetProgress;
    param.cancelHandler = cancelHandler;
    res = sendPath(path, &param);
    if (res <= 0) {
      goto CANCELED;
    }

    mark_entry = mark_entry->next;
  }

  // In case server finished earlier
  while (1) {
    // Cancel
    if (cancelHandler()) {
      res = 0;
      goto CANCELED;
    }
    
    uint32_t sync;
    int len = sizeof(uint32_t);
    res = adhocRecv(client_socket, &sync, &len);
    if (res < 0 && res != SCE_ERROR_NET_ADHOC_WOULD_BLOCK) {
      goto CANCELED;
    }
    
    if (res == 0) {
      break;
    }

    sceKernelDelayThread(RECEIVE_WAIT);
  }

  // Set progress to 100%
  sceMsgDialogProgressBarSetValue(SCE_MSG_DIALOG_PROGRESSBAR_TARGET_BAR_DEFAULT, 100);
  sceKernelDelayThread(COUNTUP_WAIT);

  // Close
  sceMsgDialogClose();

  setDialogStep(DIALOG_STEP_ADHOC_SENDED);
  goto EXIT;
  
CANCELED:
  closeWaitDialog();
  setDialogStep(DIALOG_STEP_CANCELED);
  errorDialog(res);

EXIT:
  if (mark_entry_one)
    free(mark_entry_one);

  if (thid >= 0)
    sceKernelWaitThreadEnd(thid, NULL, NULL);

  // Unlock power timers
  powerUnlock();

  // Close sockets and disconnect
  adhocCloseSockets();
  sceNetCtlAdhocDisconnect();
 
  return sceKernelExitDeleteThread(0);
}

int receive_thread(SceSize args_size, ReceiveArguments *args) {
  int res;
  SceUID thid = -1;
  uint64_t timeout;

  // Lock power timers
  powerLock();

  // Set progress to 0%
  sceMsgDialogProgressBarSetValue(SCE_MSG_DIALOG_PROGRESSBAR_TARGET_BAR_DEFAULT, 0);
  sceKernelDelayThread(DIALOG_WAIT); // Needed to see the percentage

  // Receive total size and and path length of cwd
  ShareInfo server_info;

  while (1) {
    // Cancel
    if (cancelHandler()) {
      res = 0;
      goto CANCELED;
    }
    
    int len = sizeof(ShareInfo);
    res = adhocRecv(server_socket, &server_info, &len);
    if (res < 0 && res != SCE_ERROR_NET_ADHOC_WOULD_BLOCK) {
      goto CANCELED;
    }
    
    if (res == 0) {
      // Wrong magic
      if (server_info.magic != SHARE_MAGIC || server_info.path_len >= MAX_PATH_LENGTH) {
        res = -1;
        goto CANCELED;
      }
      
      break;
    }

    sceKernelDelayThread(RECEIVE_WAIT);
  }

  // Update thread
  thid = createStartUpdateThread(server_info.file_size, 1);

  // Receive process
  uint64_t value = 0;
  while (value < server_info.file_size) {
    ShareInfo info;
    
    char path[MAX_PATH_LENGTH];
    memset(path, 0, sizeof(path));

    // Receive share info
    while (1) {
      // Cancel
      if (cancelHandler()) {
        res = 0;
        goto CANCELED;
      }
      
      int len = sizeof(ShareInfo);
      res = adhocRecv(server_socket, &info, &len);
      if (res < 0 && res != SCE_ERROR_NET_ADHOC_WOULD_BLOCK) {
        goto CANCELED;
      }
      
      if (res == 0) {
        // Wrong magic
        if (info.magic != SHARE_MAGIC || info.path_len >= MAX_PATH_LENGTH) {
          res = -1;
          goto CANCELED;
        }
        
        break;
      }

      sceKernelDelayThread(RECEIVE_WAIT);
    }

    // Receive path    
    while (1) {
      // Cancel
      if (cancelHandler()) {
        res = 0;
        goto CANCELED;
      }
      
      int len = (int)info.path_len;
      res = adhocRecv(server_socket, path, &len);
      if (res < 0 && res != SCE_ERROR_NET_ADHOC_WOULD_BLOCK) {
        goto CANCELED;
      }
      
      if (res == 0) {
        break;
      }

      sceKernelDelayThread(RECEIVE_WAIT);
    }
    
    // New path
    char dst_path[MAX_PATH_LENGTH];
    snprintf(dst_path, MAX_PATH_LENGTH, "%s%s", args->file_list->path, path + server_info.path_len);

    // Folder
    if (info.type == SHARE_TYPE_FOLDER) {
      res = sceIoMkdir(dst_path, 0777);
      if (res < 0 && res != SCE_ERROR_ERRNO_EEXIST) {
        goto CANCELED;
      }
      
      value += DIRECTORY_SIZE;
      SetProgress(value, server_info.file_size);
      continue;
    }
    
    // Receive file
    if (info.type == SHARE_TYPE_FILE) {
      SceUID fddst = sceIoOpen(dst_path, SCE_O_WRONLY | SCE_O_CREAT | SCE_O_TRUNC, 0777);
      if (fddst < 0) {
        res = fddst;
        goto CANCELED;
      }

      void *buf = memalign(4096, SHARE_SIZE);

      uint64_t recv_offset = 0;
      while (recv_offset < info.file_size) {
        // Cancel
        if (cancelHandler()) {
          res = 0;
          free(buf);
          sceIoClose(fddst);
          sceIoRemove(dst_path);
          goto CANCELED;
        }
        
        uint64_t remain = info.file_size - recv_offset;
        int len = (remain < SHARE_SIZE) ? (int)remain : SHARE_SIZE;
        res = adhocRecv(server_socket, buf, &len);
        if (res < 0 && res != SCE_ERROR_NET_ADHOC_WOULD_BLOCK) {
          free(buf);
          sceIoClose(fddst);
          sceIoRemove(dst_path);
          goto CANCELED;
        }
        
        if (res == 0) {
          int written = sceIoWrite(fddst, buf, len);
          if (written < 0) {
            res = written;
            free(buf);
            sceIoClose(fddst);
            sceIoRemove(dst_path);
            goto CANCELED;
          }

          recv_offset += len;
          value += len;
          SetProgress(value, server_info.file_size);
          continue;
        }

        sceKernelDelayThread(RECEIVE_WAIT);
      }
      
      free(buf);
      sceIoClose(fddst);
    }
  }

  // In case server finished earlier
  uint32_t sync = 0x12345678;
  adhocSend(server_socket, &sync, sizeof(uint32_t));

  // Set progress to 100%
  sceMsgDialogProgressBarSetValue(SCE_MSG_DIALOG_PROGRESSBAR_TARGET_BAR_DEFAULT, 100);
  sceKernelDelayThread(COUNTUP_WAIT);

  // Close
  sceMsgDialogClose();

  setDialogStep(DIALOG_STEP_ADHOC_RECEIVED);
  goto EXIT;
  
CANCELED:
  closeWaitDialog();
  setDialogStep(DIALOG_STEP_CANCELED);
  errorDialog(res);

EXIT:
  if (thid >= 0)
    sceKernelWaitThreadEnd(thid, NULL, NULL);

  // Unlock power timers
  powerUnlock();

  // Close sockets and disconnect
  adhocCloseSockets();
  sceNetCtlAdhocDisconnect();

  return sceKernelExitDeleteThread(0);
}

int client_waiting_thread(SceSize args, void *argp) {
  // Create PTP socket and wait for connection
  sceNetAdhocctlGetEtherAddr(&client_addr);
  client_socket = sceNetAdhocPtpListen(&client_addr, 1, SOCKET_BUFSIZE, RECEIVE_WAIT, 50, 1, 0);
  if (client_socket < 0) {
    server_request_result = client_socket;
    goto EXIT;
  }

  // Establish PTP connection
  SceUShort16 server_port;
  server_socket = sceNetAdhocPtpAccept(client_socket, &server_addr, &server_port, 0, 0);
  if (server_socket < 0) {
    server_request_result = server_socket;
    goto EXIT;
  }
  
  // Check if we have successfully received a request from a server
  int res = sceNetAdhocctlGetPeerInfo(&server_addr, sizeof(SceNetAdhocctlPeerInfo), &server_peer_info);
  if (res < 0) {
    server_request_result = res;
    goto EXIT;
  }
  
  server_request_result = 1;

EXIT:
  return sceKernelExitDeleteThread(0);
}

void adhocAlertSockets() {
  if (client_socket >= 0)
    sceNetAdhocSetSocketAlert(client_socket, SCE_NET_ADHOC_F_ALERTALL);

  if (server_socket >= 0)
    sceNetAdhocSetSocketAlert(server_socket, SCE_NET_ADHOC_F_ALERTALL);  
}

void adhocCloseSockets() {
  adhocAlertSockets();

  if (client_waiting_thid >= 0) {
    sceKernelWaitThreadEnd(client_waiting_thid, NULL, NULL);
    client_waiting_thid = -1;
  }

  if (client_socket >= 0) {
    sceNetAdhocPtpClose(client_socket, 0);
    client_socket = -1;
  }

  if (server_socket >= 0) {
    sceNetAdhocPtpClose(server_socket, 0);
    server_socket = -1;
  }
}

int adhocUpdatePeerList() {
  int buflen = 0;
  sceNetAdhocctlGetPeerList(&buflen, NULL);

  memset(peer_list, 0, PEERLIST_MAX * sizeof(SceNetAdhocctlPeerInfo));
  peer_count = 0;
  if (buflen > 0) {
    if (buflen > PEERLIST_MAX * sizeof(SceNetAdhocctlPeerInfo)) {
      buflen = PEERLIST_MAX * sizeof(SceNetAdhocctlPeerInfo);
    }
    
    peer_count = buflen / sizeof(SceNetAdhocctlPeerInfo);
    sceNetAdhocctlGetPeerList(&buflen, peer_list);
  }

  return peer_count;
}

int adhocSendServerResponse(char *response) {
  return adhocSend(server_socket, response, 4);  
}

char *adhocReceiveClientReponse() {
  int len = 4;
  adhocRecv(client_socket, client_response, &len);
  return client_response;
}

int adhocReceiveServerRequest() {
  return server_request_result;
}

char *adhocGetServerNickname() {
  return (char *)server_peer_info.nickname.data;
}

void adhocWaitingForServerRequest() {
  server_request_result = 0;
  
  client_waiting_thid = sceKernelCreateThread("client_waiting_thread", client_waiting_thread, 0x10000100, 0x4000, 0, 0, NULL);
  if (client_waiting_thid >= 0)
    sceKernelStartThread(client_waiting_thid, 0, NULL);
}

int getAdhocDialogStatus() {
  return adhoc_dialog.status;
}

int initAdhocDialog() {
  memset(&adhoc_dialog, 0, sizeof(AdhocDialog));
  
  // Opening status
  adhoc_dialog.result = ADHOC_DIALOG_RESULT_NONE;
  adhoc_dialog.status = ADHOC_DIALOG_OPENING;

  // Selector
  adhoc_dialog.sel = 0;

  // Width and height
  adhoc_dialog.width = 340.0f;
  adhoc_dialog.height = FONT_Y_SPACE * (PEERLIST_MAX + 1);

  // For buttons
  adhoc_dialog.height += 2.0f * FONT_Y_SPACE;

  // Margin
  adhoc_dialog.width += 2.0f * SHELL_MARGIN_X;
  adhoc_dialog.height += 2.0f * SHELL_MARGIN_Y;

  // Position
  adhoc_dialog.x = ALIGN_CENTER(SCREEN_WIDTH, adhoc_dialog.width);
  adhoc_dialog.y = ALIGN_CENTER(SCREEN_HEIGHT, adhoc_dialog.height);

  // Align
  int y_n = (int)((float)(adhoc_dialog.y - 2.0f) / FONT_Y_SPACE);
  adhoc_dialog.y = (float)y_n * FONT_Y_SPACE + 2.0f;

  // Scale
  adhoc_dialog.scale = 0.0f;

  // Reset
  memset(peer_list, 0, PEERLIST_MAX * sizeof(SceNetAdhocctlPeerInfo));
  peer_count = 0;

  return 0;
}

void adhocDialogCtrl() {
  adhocUpdatePeerList();
  
  if (pressed_pad[PAD_ENTER]) {
    if (peer_count > 0) {
      SceNetAdhocctlPeerInfo *curr = peer_list;
      int i;
      for (i = 0; i < PEERLIST_MAX; i++) {
        if (curr) {
          if (adhoc_dialog.sel == i) {
            memcpy(&client_addr, &curr->macAddr, sizeof(SceNetEtherAddr));
            adhoc_dialog.result = ADHOC_DIALOG_RESULT_WAITING_FOR_RESPONSE;
            adhoc_dialog.status = ADHOC_DIALOG_CLOSING;
            break;
          }
          
          curr = curr->next;
        }
      }
    }
  }
  
  if (pressed_pad[PAD_CANCEL]) {
    sceNetCtlAdhocDisconnect();
    adhoc_dialog.status = ADHOC_DIALOG_CLOSING;
  }
  
  if (hold_pad[PAD_UP] || hold2_pad[PAD_LEFT_ANALOG_UP]) {
    if (adhoc_dialog.sel > 0)
      adhoc_dialog.sel--;
  } else if (hold_pad[PAD_DOWN] || hold2_pad[PAD_LEFT_ANALOG_DOWN]) {
    if (adhoc_dialog.sel < peer_count - 1)
      adhoc_dialog.sel++;
  }
  
  // Easing out
  if (adhoc_dialog.status == ADHOC_DIALOG_CLOSING) {
    if (adhoc_dialog.scale > 0.0f) {
      adhoc_dialog.scale -= easeOut(0.0f, adhoc_dialog.scale, 0.25f, 0.01f);
    } else {
      adhoc_dialog.status = ADHOC_DIALOG_CLOSED;
    }
  }

  if (adhoc_dialog.status == ADHOC_DIALOG_OPENING) {
    if (adhoc_dialog.scale < 1.0f) {
      adhoc_dialog.scale += easeOut(adhoc_dialog.scale, 1.0f, 0.25f, 0.01f);
    } else {
      adhoc_dialog.status = ADHOC_DIALOG_OPENED;
    }
  }

  // Waiting state
  if (adhoc_dialog.status == ADHOC_DIALOG_CLOSED) {
    if (adhoc_dialog.result == ADHOC_DIALOG_RESULT_WAITING_FOR_RESPONSE) {      
      // Create PTP socket and start connection
      sceNetAdhocctlGetEtherAddr(&server_addr);
      client_socket = sceNetAdhocPtpOpen(&server_addr, 0, &client_addr, 1, SOCKET_BUFSIZE, RECEIVE_WAIT, 50, 0);
      if (client_socket < 0) {
        sceNetCtlAdhocDisconnect();
        errorDialog(client_socket);
        return;
      }
      
      // Establish PTP connection
      /*int res = sceNetAdhocPtpConnect(client_socket, 0, SCE_NET_ADHOC_F_NONBLOCK);
      if (res < 0) {
        sceNetAdhocPtpClose(client_socket, 0);
        sceNetCtlAdhocDisconnect();
        errorDialog(res);
        return;
      }*/
      
      memset(client_response, 0, sizeof(client_response));
      
      initMessageDialog(SCE_MSG_DIALOG_BUTTON_TYPE_CANCEL, language_container[PLEASE_WAIT]);
      setDialogStep(DIALOG_STEP_ADHOC_SEND_WAITING);
    }
  }
}

void drawAdhocDialog() {
  if (adhoc_dialog.status == ADHOC_DIALOG_CLOSED)
    return;

  // Dialog background
  float dialog_width = vita2d_texture_get_width(dialog_image);
  float dialog_height = vita2d_texture_get_height(dialog_image);
  vita2d_draw_texture_scale_rotate_hotspot(dialog_image, adhoc_dialog.x + adhoc_dialog.width / 2.0f,
                                           adhoc_dialog.y + adhoc_dialog.height / 2.0f,
                                           adhoc_dialog.scale * (adhoc_dialog.width/dialog_width),
                                           adhoc_dialog.scale * (adhoc_dialog.height/dialog_height),
                                           0.0f, dialog_width / 2.0f, dialog_height / 2.0f);

  if (adhoc_dialog.status == ADHOC_DIALOG_OPENED) {
    float string_y = adhoc_dialog.y + SHELL_MARGIN_Y - 2.0f;

    // Select PS Vita
    pgf_draw_text(ALIGN_CENTER(SCREEN_WIDTH, pgf_text_width(language_container[ADHOC_SELECT_PSVITA])), string_y, DIALOG_COLOR, language_container[ADHOC_SELECT_PSVITA]);
    string_y += FONT_Y_SPACE;

    // Peer list
    SceNetAdhocctlPeerInfo *curr = peer_list;
    int i;
    for (i = 0; i < PEERLIST_MAX; i++) {
      if (curr) {
        char *user = (char *)curr->nickname.data;
        pgf_draw_text(ALIGN_CENTER(SCREEN_WIDTH, pgf_text_width(user)), string_y, (i == adhoc_dialog.sel) ? FOCUS_COLOR : DIALOG_COLOR, user);
        curr = curr->next;
      }
      
      string_y += FONT_Y_SPACE;
    }

    // Buttons
    char button_string[128];
    sprintf(button_string, "%s %s    %s %s", enter_button == SCE_SYSTEM_PARAM_ENTER_BUTTON_CIRCLE ? CIRCLE : CROSS, language_container[OK],
                                             enter_button == SCE_SYSTEM_PARAM_ENTER_BUTTON_CIRCLE ? CROSS : CIRCLE, language_container[CANCEL]);
    pgf_draw_text(ALIGN_CENTER(SCREEN_WIDTH, pgf_text_width(button_string)), string_y + FONT_Y_SPACE, DIALOG_COLOR, button_string);
  }
}