/* radare - LGPL - Copyright 2010-2022 pancake, nibble */

#include <r_util.h>
#include "pe.h"

bool PE_(r_bin_pe_section_perms)(RBinFile *bf, const char *name, int perms) {
	RBinPEObj *pe = PE_(get)(bf);
	if (!pe) {
		return false;
	}
	PE_(image_section_header) *shdr = pe->section_header;
	if (!shdr) {
		return false;
	}
	int i;
	for (i = 0; i < pe->num_sections; i++) {
		const char *sname = (const char*) shdr[i].Name;
		if (!strncmp (name, sname, PE_IMAGE_SIZEOF_SHORT_NAME)) {
			ut32 newperms = shdr[i].Characteristics;
			ut8 newperms_le[4];

			/* Apply permission flags */
			if (perms & R_PERM_X) {
				newperms |=  PE_IMAGE_SCN_MEM_EXECUTE;
			} else {
				newperms &= ~PE_IMAGE_SCN_MEM_EXECUTE;
			}
			if (perms & R_PERM_W) {
				newperms |=  PE_IMAGE_SCN_MEM_WRITE;
			} else {
				newperms &= ~PE_IMAGE_SCN_MEM_WRITE;
			}
			if (perms & R_PERM_R) {
				newperms |=  PE_IMAGE_SCN_MEM_READ;
			} else {
				newperms &= ~PE_IMAGE_SCN_MEM_READ;
			}
			if (perms & R_PERM_SHAR) {
				newperms |=  PE_IMAGE_SCN_MEM_SHARED;
			} else {
				newperms &= ~PE_IMAGE_SCN_MEM_SHARED;
			}

			int patchoff = pe->section_header_offset;
			patchoff += i * sizeof (PE_(image_section_header));
			patchoff += r_offsetof (PE_(image_section_header), Characteristics);
			r_write_le32 (newperms_le, newperms);
			printf ("wx %08x @ 0x%x\n", newperms, patchoff);
			int res = r_buf_write_at (bf->buf, patchoff, newperms_le, sizeof (newperms_le));
			if (res != sizeof (newperms_le)) {
				return false;
			}
			return true;
		}
	}
	return false;
}