+static void fio_unpin_memory(void *pinned)
+{
+ if (pinned) {
+ if (munlock(pinned, mlock_size) < 0)
+ perror("munlock");
+ munmap(pinned, mlock_size);
+ }
+}
+
+static void *fio_pin_memory(void)
+{
+ long pagesize, pages;
+ void *ptr;
+
+ if (!mlock_size)
+ return NULL;
+
+ /*
+ * Don't allow mlock of more than real_mem-128MB
+ */
+ pagesize = sysconf(_SC_PAGESIZE);
+ pages = sysconf(_SC_PHYS_PAGES);
+ if (pages != -1 && pagesize != -1) {
+ unsigned long long real_mem = pages * pagesize;
+
+ if ((mlock_size + 128 * 1024 * 1024) > real_mem) {
+ mlock_size = real_mem - 128 * 1024 * 1024;
+ printf("fio: limiting mlocked memory to %lluMiB\n",
+ mlock_size >> 20);
+ }
+ }
+
+ ptr = mmap(NULL, mlock_size, PROT_READ | PROT_WRITE, MAP_PRIVATE | OS_MAP_ANON, 0, 0);
+ if (!ptr) {
+ perror("malloc locked mem");
+ return NULL;
+ }
+ if (mlock(ptr, mlock_size) < 0) {
+ munmap(ptr, mlock_size);
+ perror("mlock");
+ return NULL;
+ }
+
+ return ptr;
+}
+