#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define RSRC_DATA_SIZE      256
#define NUM_SOURCE_FILES    12000
#define ITERATION_LIMIT     1000000

static char g_basedir[1024];
static char g_src[NUM_SOURCE_FILES][1024];
static char g_src_rsrc[NUM_SOURCE_FILES][1024];

static void fill_pattern(void *buf, size_t len, unsigned char seed)
{
  unsigned char *p = (unsigned char*) buf;
  for (size_t i = 0; i < len; i++)
    p[i] = (unsigned char)(seed + (i & 0xff));
}

static void create_source_files(void)
{
  char *data = (char*) malloc(RSRC_DATA_SIZE);
  if (!data) { perror("malloc"); exit(1); }
  
  for (int i = 0; i < NUM_SOURCE_FILES; i++) {
    snprintf(g_src[i], sizeof(g_src[i]),
             "%s/source_%d", g_basedir, i);
    snprintf(g_src_rsrc[i], sizeof(g_src_rsrc[i]),
             "%s/source_%d/..namedfork/rsrc", g_basedir, i);

    int fd = open(g_src[i], O_CREAT | O_WRONLY | O_TRUNC, 0644);
    if (fd < 0) { perror("open src"); exit(1); }
    fill_pattern(data, 8192, 'D' + i);
    write(fd, data, 8192);
    close(fd);

    fd = open(g_src_rsrc[i], O_CREAT | O_WRONLY | O_TRUNC, 0644);
    if (fd >= 0) {
      fill_pattern(data, RSRC_DATA_SIZE, 'R' + i);
      write(fd, data, RSRC_DATA_SIZE);
      close(fd);
    } else {
      fill_pattern(data, RSRC_DATA_SIZE, 'R' + i);
      if (setxattr(g_src[i], "com.apple.ResourceFork",
                   data, RSRC_DATA_SIZE, 0, 0) != 0) {
        perror("setxattr ResourceFork");
        exit(1);
      }
    }

    printf("  [+] source_%d  data=8KB  rsrc=%dKB\n", i, RSRC_DATA_SIZE / 1024);
  }
  
  free(data);
}

static void *rsrc_write_worker(void *arg)
{
  int id = (int)(long)arg;
  char buf[8192];
  long n = 0;
  
  fill_pattern(buf, sizeof(buf), 'W' + id);
  
  while (n < ITERATION_LIMIT) {
    int file_idx = n % NUM_SOURCE_FILES;
    int fd = open(g_src_rsrc[file_idx], O_WRONLY | O_CREAT, 0644);
    if (fd >= 0) {
      off_t off = ((n * 4096) % RSRC_DATA_SIZE);
      pwrite(fd, buf, sizeof(buf), off);
      if ((n & 0x7) == 0)
        fsync(fd);
      
      close(fd);
    }
    /* Fallback: xattr write (always goes through the namedstream path) */
    else {
      setxattr(g_src[file_idx], "com.apple.ResourceFork",
               buf, sizeof(buf), 0, 0);
    }
    
    n++;
  }
  printf("[rsrc_wr_%d] done (%ld ops)\n", id, n);
  return NULL;
}

static void cleanup_files(void)
{
  printf("[*] Cleaning up files...\n");
  for (int i = 0; i < NUM_SOURCE_FILES; i++) {
    unlink(g_src[i]);
    if ((i % 1000) == 0 && i > 0)
      printf("  [+] removed %d files\n", i);
  }
  char cmd[1100];
  snprintf(cmd, sizeof(cmd), "rm -f %s/clone_* %s/fclone_* %s/cross_* 2>/dev/null", g_basedir, g_basedir, g_basedir);
  system(cmd);
  printf("[*] Cleanup complete.\n");
}

int main(int argc, char *argv[])
{
  if (argc < 2) {
    fprintf(stderr, "Usage: %s \n", argv[0]);
    return 1;
  }
  
  strncpy(g_basedir, argv[1], sizeof(g_basedir) - 1);
  for (int i = 0; i < NUM_SOURCE_FILES; i++) {
    snprintf(g_src[i], sizeof(g_src[i]), "%s/source_%d", g_basedir, i);
    snprintf(g_src_rsrc[i], sizeof(g_src_rsrc[i]), "%s/source_%d/..namedfork/rsrc", g_basedir, i);
  }
  
  cleanup_files();
  while (1)
  {
    {
      mkdir(g_basedir, 0755);
      printf("[*] Creating source files...\n");
      create_source_files();
      printf("\n[*] File creation complete.\n");
    }

    {
      printf("[*] Starting tests...\n");
      pthread_t tids[1];
      int tc = 0;
      pthread_create(&tids[tc++], NULL, rsrc_write_worker,  (void *)(long)0);
      
      for (int i = 0; i < tc; i++)
        pthread_join(tids[i], NULL);

      printf("[*] Tests done...\n");
    }
    
    {
      printf("[*] Cleaning up files...\n");
      cleanup_files();
      printf("\n[*] File cleanup complete.\n");
    }
  }

  return 0;
}