#include "fs.h"
#include <fcntl.h>
#include <string.h>
#include "fproc.h"
#include <minix/vm.h>
#include <sys/mman.h>
#include <machine/elf.h>
#include "param.h"

#include <machine/archtypes.h>
#include "../../kernel/const.h"
#include "../../kernel/config.h"
#include "../../kernel/type.h"
#include "../../kernel/proc.h"

/* Include ELF headers */
#include <sys/elf_core.h>
#include <sys/procfs.h>

FORWARD _PROTOTYPE( void fill_elf_header, (Elf32_Ehdr *elf_header,
		int phnum)                                                 );
FORWARD _PROTOTYPE( void fill_prog_header, (Elf32_Phdr *prog_header,
	Elf32_Word p_type, Elf32_Off p_offset, Elf32_Addr p_vaddr,
	Elf32_Word p_flags, Elf32_Word p_filesz, Elf32_Word p_memsz)       );
FORWARD _PROTOTYPE( int get_memory_regions, (Elf32_Phdr phdrs[])           );
FORWARD _PROTOTYPE( void fill_note_segment_and_entries_hdrs,
	(Elf32_Phdr phdrs[], Elf32_Nhdr nhdrs[])                              );
FORWARD _PROTOTYPE( void adjust_offsets, (Elf32_Phdr phdrs[], int phnum)   );
FORWARD _PROTOTYPE( void dump_elf_header, (Elf32_Ehdr elf_header)          );
FORWARD _PROTOTYPE( void dump_notes, (Elf32_Nhdr nhdrs[], int csig,
	char *exe_name)                                                    );
FORWARD _PROTOTYPE( void dump_segments, (Elf32_Phdr phdrs[], int phnum)    );

/*===========================================================================*
 *				write_elf_core_file			     *
 *===========================================================================*/
/* First, fill in all the required headers, second, adjust the offsets,
 * third, dump everything into the core file
 */
PUBLIC void write_elf_core_file(int csig, char *exe_name)
{
#define MAX_REGIONS 20
#define NR_NOTE_ENTRIES 2
  Elf_Ehdr elf_header;
  Elf_Phdr phdrs[MAX_REGIONS + 1];
  Elf_Nhdr nhdrs[NR_NOTE_ENTRIES];
  int phnum;

  /* Fill in the NOTE Program Header - at phdrs[0] - and
   * note entries' headers
   */
  fill_note_segment_and_entries_hdrs(phdrs, nhdrs);

  /* Get the memory segments and fill in the Program headers */
  phnum = get_memory_regions(phdrs) + 1;

  /* Fill in the ELF header */
  fill_elf_header(&elf_header, phnum);

  /* Adjust offsets in program headers - The layout in the ELF core file
   * is the following: the ELF Header, the Note Program Header,
   * the rest of Program Headers (memory segments), Note contents,
   * the program segments' contents
   */
  adjust_offsets(phdrs, phnum);

  /* Write ELF header */
  dump_elf_header(elf_header);

  /* Write Program headers (Including the NOTE) */
  dump_program_headers(phdrs, phnum);

  /* Write NOTE contents */
  dump_notes(nhdrs, csig, exe_name);

  /* Write segments' contents */
  dump_segments(phdrs, phnum);
}

/*===========================================================================*
 *				fill_elf_header        			     *
 *===========================================================================*/
PRIVATE void fill_elf_header (Elf_Ehdr *elf_header, int phnum)
{
  memset((void *) elf_header, 0, sizeof(Elf_Ehdr));

  elf_header->e_ident[EI_MAG0] = ELFMAG0;
  elf_header->e_ident[EI_MAG1] = ELFMAG1;
  elf_header->e_ident[EI_MAG2] = ELFMAG2;
  elf_header->e_ident[EI_MAG3] = ELFMAG3;
  elf_header->e_ident[EI_CLASS] = ELF_TARG_CLASS;
  elf_header->e_ident[EI_DATA] = ELF_TARG_DATA;
  elf_header->e_ident[EI_VERSION] = EV_CURRENT;
  elf_header->e_ident[EI_OSABI] = ELFOSABI_FREEBSD;
  elf_header->e_type = ET_CORE;
  elf_header->e_machine = ELF_TARG_MACH;
  elf_header->e_version = EV_CURRENT;
  elf_header->e_ehsize = sizeof(Elf_Ehdr);
  elf_header->e_phoff = sizeof(Elf_Ehdr);
  elf_header->e_phentsize = sizeof(Elf_Phdr);
  elf_header->e_phnum = phnum;
}

/*===========================================================================*
 *				fill_prog_header        		     *
 *===========================================================================*/
PRIVATE void fill_prog_header (Elf_Phdr *prog_header, Elf_Word p_type,
	Elf_Off p_offset, Elf_Addr p_vaddr, Elf_Word p_flags,
	Elf_Word p_filesz, Elf_Word p_memsz)
{

  memset((void *) prog_header, 0, sizeof(Elf_Phdr));

  prog_header->p_type = p_type;
  prog_header->p_offset = p_offset;
  prog_header->p_vaddr = p_vaddr;
  prog_header->p_flags = p_flags;
  prog_header->p_filesz = p_filesz;
  prog_header->p_memsz = p_memsz;

}

#define PADBYTES    4
#define PAD_LEN(x)  ((x + (PADBYTES - 1)) & ~(PADBYTES - 1))

/*===========================================================================*
 *			fill_note_segment_and_entries_hdrs     	     	     *
 *===========================================================================*/
PRIVATE void fill_note_segment_and_entries_hdrs(Elf_Phdr phdrs[],
				Elf_Nhdr nhdrs[])
{
  int filesize;
  const char *note_name = ELF_NOTE_MINIX_ELFCORE_NAME "\0";
  int name_len, mei_len, gregs_len;

  /* Size of notes in the core file is rather fixed:
   * sizeof(minix_elfcore_info_t) +
   * 2 * sizeof(Elf_Nhdr) + the size of the padded name of the note
   * - i.e. "MINIX-CORE\0" padded to 4-byte alignment => 2 * 8 bytes
   */

  name_len = strlen(note_name) + 1;
  mei_len = sizeof(minix_elfcore_info_t);
  gregs_len = sizeof(gregset_t);

  /* Make sure to also count the padding bytes */
  filesize = PAD_LEN(mei_len) + PAD_LEN(gregs_len) +
	2 * sizeof(Elf_Nhdr) + 2 * PAD_LEN(name_len);
  fill_prog_header(&phdrs[0], PT_NOTE, 0, 0, PF_R, filesize, 0);

  /* First note entry header */
  nhdrs[0].n_namesz = name_len;
  nhdrs[0].n_descsz = sizeof(minix_elfcore_info_t);
  nhdrs[0].n_type = NT_MINIX_ELFCORE_INFO;

  /* Second note entry header */
  nhdrs[1].n_namesz = name_len;
  nhdrs[1].n_descsz = sizeof(gregset_t);
  nhdrs[1].n_type = NT_MINIX_ELFCORE_GREGS;
}

/*===========================================================================*
 *				adjust_offset   			     *
 *===========================================================================*/
PRIVATE void adjust_offsets(Elf_Phdr phdrs[], int phnum)
{
  int i;
  long offset = sizeof(Elf_Ehdr) + phnum * sizeof(Elf_Phdr);

  for (i = 0; i < phnum; i++) {
	phdrs[i].p_offset = offset;
	offset += phdrs[i].p_filesz;
  }
}

/*===========================================================================*
 *				write_buf       			     *
 *===========================================================================*/
PRIVATE void write_buf(char *buf, int size)
{
  m_in.buffer = buf;
  m_in.nbytes = size;
  read_write(WRITING);
}

/*===========================================================================*
 *				get_memory_regions			     *
 *===========================================================================*/
/* The same as dump_regions from procfs/pid.c */
PRIVATE int get_memory_regions(Elf_Phdr phdrs[])
{
  /* Print the virtual memory regions of a process. */
  struct vm_region_info vri[MAX_VRI_COUNT];
  vir_bytes next;
  int i, r, count;
  Elf_Word pflags;

  count = 0;
  next = 0;

  do {
	r = vm_info_region(fp->fp_endpoint, vri, MAX_VRI_COUNT,
			&next);
	if (r < 0) return r;
	if (r == 0) break;

	for (i = 0; i < r; i++) {

		pflags = (vri[i].vri_prot & PROT_READ ? PF_R : 0)
			| (vri[i].vri_prot & PROT_WRITE ? PF_W : 0)
			| (vri[i].vri_prot & PROT_EXEC ? PF_X : 0);

		fill_prog_header (&phdrs[count + 1], PT_LOAD,
				0, vri[i].vri_addr, pflags,
				vri[i].vri_length, vri[i].vri_length);
		count++;

		if (count >= MAX_REGIONS) {
			printf("VFS: get_memory_regions Warning: "
				"Program has too many regions\n");
			return count;
		}
	}
  } while (r == MAX_VRI_COUNT);

  return count;
}

/*===========================================================================*
 *				dump_notes			             *
 *===========================================================================*/
PRIVATE void dump_notes(Elf_Nhdr nhdrs[], int csig, char *exe_name)
{
  char *note_name = ELF_NOTE_MINIX_ELFCORE_NAME "\0";
  char pad[4];
  minix_elfcore_info_t mei;
  int mei_len = sizeof(minix_elfcore_info_t);
  int gregs_len = sizeof(gregset_t);
  struct proc p;
  char proc_name[PROC_NAME_LEN];

  /* Get process's name */
  if (sys_datacopy(PM_PROC_NR, (vir_bytes) exe_name,
		VFS_PROC_NR, (vir_bytes) proc_name, PROC_NAME_LEN) != OK)
	printf("VFS: Cannot get porcess's name\n");

  /* Dump first note entry */
  mei.mei_version = MINIX_ELFCORE_VERSION;
  mei.mei_meisize = mei_len;
  mei.mei_signo = csig;
  mei.mei_pid = fp->fp_pid;
  memcpy(mei.mei_command, proc_name, sizeof(mei.mei_command));

  write_buf((char *)&nhdrs[0], sizeof(Elf_Nhdr));
  write_buf(note_name, nhdrs[0].n_namesz);
  write_buf(pad, PAD_LEN(nhdrs[0].n_namesz) - nhdrs[0].n_namesz);
  write_buf((char *)&mei, mei_len);
  write_buf(pad, PAD_LEN(mei_len) - mei_len);

  /* XXX: Other way to read registries ? */
  /* Get registers */
  if (sys_getproc(&p, fp->fp_endpoint) != OK)
	printf("VFS: Could not read registers\n");

  /* Dump second note entry - the general registers */
  write_buf((char *)&nhdrs[1], sizeof(Elf_Nhdr));
  write_buf(note_name, nhdrs[1].n_namesz);
  write_buf(pad, PAD_LEN(nhdrs[1].n_namesz) - nhdrs[1].n_namesz);
  write_buf((char *)&(p.p_reg), gregs_len);
  write_buf(pad, PAD_LEN(gregs_len) - gregs_len);
}

/*===========================================================================*
 *				dump_elf_header			             *
 *===========================================================================*/
PRIVATE void dump_elf_header(Elf_Ehdr elf_header)
{
  write_buf((char *)&elf_header, sizeof(Elf_Ehdr));
}

/*===========================================================================*
 *			  dump_program_headers			             *
 *===========================================================================*/
PRIVATE void dump_program_headers(Elf_Phdr phdrs[], int phnum)
{
  int i;

  for (i = 0; i < phnum; i++)
	write_buf((char *)&phdrs[i], sizeof(Elf_Phdr));
}

/*===========================================================================*
 *			      dump_segments 			             *
 *===========================================================================*/
PRIVATE void dump_segments(Elf_Phdr phdrs[], int phnum)
{
  int i;
  vir_bytes len;
  off_t off, seg_off;
  int r;
  static u8_t buf[CLICK_SIZE];

  for (i = 1; i < phnum; i++) {
	len = phdrs[i].p_memsz;
	seg_off = phdrs[i].p_vaddr;

	for (off = 0; off < len; off += CLICK_SIZE) {
		r = sys_vircopy(fp->fp_endpoint, D,
			(vir_bytes) (seg_off + off),
			SELF, D, (vir_bytes) buf,
			(phys_bytes) CLICK_SIZE);

		write_buf((char *)buf, (off + CLICK_SIZE <= len) ?
					CLICK_SIZE : (len - off));
	}
  }
}