#define _GNU_SOURCE

#include <libvirt/libvirt.h>
#include <libvirt/virterror.h>
#include <stdio.h>
#include <pthread.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/syscall.h>

#define NUM_THREADS 8
#define MAX_VMS 1000

const char *XML = "<domain type='kvm'>"
"  <name>vm%06d</name>"
"  <memory unit='KiB'>220160</memory>"
"  <currentMemory unit='KiB'>60000</currentMemory>"
"  <vcpu placement='static'>1</vcpu>"
"  <os>"
"    <type arch='x86_64'>hvm</type>"
"    <boot dev='hd'/>"
"  </os>"
"  <features>"
"    <acpi/>"
"  </features>"
"  <devices>"
"    <emulator>/bin/qemu-kvm</emulator>"
"    <disk type='file' device='disk'>"
"      <driver name='qemu' type='qcow2'/>"
"      <source file='/home/berrange/VirtualMachines/vm%06d.qcow'/>"
"      <target dev='vda' bus='virtio'/>"
"      <address type='pci' domain='0x0000' bus='0x00' slot='0x04' function='0x0'/>"
"      <shareable/>"
"    </disk>"
"    <console type='pty'>"
"      <target type='serial' port='0'/>"
"    </console>"
"    <input type='mouse' bus='ps2'/>"
"    <graphics type='vnc' port='-1' autoport='yes' listen='0.0.0.0'/>"
"  </devices>"
"</domain>";


static pid_t gettid(void)
{
  return syscall(SYS_gettid);
}

static void *one_thread(void *arg)
{
  int i;
  virConnectPtr conn = arg;
  virConnectRef(conn);
  char *cmd;
  if (asprintf(&cmd, "qemu-img create -f qcow2 /home/berrange/VirtualMachines/vm%06d.qcow 1G", gettid()) < 0)
    abort();

  if (system(cmd)  != 0)
    abort();

  char *xml;
  if (asprintf(&xml, XML, gettid(), gettid()) < 0)
    abort();

  for (i = 0 ; i < MAX_VMS ; i++) {
    fprintf(stderr, "%d: %d of %d \n", gettid(), i, MAX_VMS);
    virDomainPtr dom = virDomainCreateXML(conn, xml, VIR_DOMAIN_START_AUTODESTROY);

    virDomainDestroy(dom);

  }
  free(xml);
  free(cmd);
  virConnectClose(conn);
  return NULL;
}

int main(int argc, char **argv) {
  virConnectPtr conn;
  int ret = -1;
  pthread_t threads[NUM_THREADS];
  int i;

  if (!(conn = virConnectOpen("qemu:///session")))
    goto error;

  for (i = 0 ; i < NUM_THREADS ; i++) {
    if (pthread_create(&threads[i], NULL,
		       one_thread, conn) < 0)
      goto error;
  }

  for (i = 0 ; i < NUM_THREADS ; i++) {
    if (pthread_join(threads[i], NULL) < 0)
      goto error;
  }

  ret = 0;
 error:
  if (ret != 0) {
    virErrorPtr err = virGetLastError();
    fprintf(stderr, "error: %s\n", err ? err->message : "<unknown>");
  }
  if (conn)
    virConnectClose(conn);
  exit(ret);
}