/*
   *  linux/fs/hpfs/ea.c
   *
   *  Mikulas Patocka (mikulas@artax.karlin.mff.cuni.cz), 1998-1999
   *
   *  handling extended attributes
   */
  
  #include "hpfs_fn.h"
  
  /* Remove external extended attributes. ano specifies whether a is a 
     direct sector where eas starts or an anode */
  
  void hpfs_ea_ext_remove(struct super_block *s, secno a, int ano, unsigned len)
  {
  	unsigned pos = 0;
  	while (pos < len) {
  		char ex[4 + 255 + 1 + 8];
  		struct extended_attribute *ea = (struct extended_attribute *)ex;
  		if (pos + 4 > len) {
  			hpfs_error(s, "EAs don't end correctly, %s %08x, len %08x",
  				ano ? "anode" : "sectors", a, len);
  			return;
  		}
  		if (hpfs_ea_read(s, a, ano, pos, 4, ex)) return;

  		if (ea_indirect(ea)) {

  			if (ea_valuelen(ea) != 8) {

  				hpfs_error(s, "ea_indirect(ea) set while ea->valuelen!=8, %s %08x, pos %08x",

  					ano ? "anode" : "sectors", a, pos);
  				return;
  			}
  			if (hpfs_ea_read(s, a, ano, pos + 4, ea->namelen + 9, ex+4))
  				return;

  			hpfs_ea_remove(s, ea_sec(ea), ea_in_anode(ea), ea_len(ea));

  		}

  		pos += ea->namelen + ea_valuelen(ea) + 5;

  	}
  	if (!ano) hpfs_free_sectors(s, a, (len+511) >> 9);
  	else {
  		struct buffer_head *bh;
  		struct anode *anode;
  		if ((anode = hpfs_map_anode(s, a, &bh))) {
  			hpfs_remove_btree(s, &anode->btree);
  			brelse(bh);
  			hpfs_free_sectors(s, a, 1);
  		}
  	}
  }
  
  static char *get_indirect_ea(struct super_block *s, int ano, secno a, int size)
  {
  	char *ret;
  	if (!(ret = kmalloc(size + 1, GFP_NOFS))) {

  		pr_err("out of memory for EA
  ");

  		return NULL;
  	}
  	if (hpfs_ea_read(s, a, ano, 0, size, ret)) {
  		kfree(ret);
  		return NULL;
  	}
  	ret[size] = 0;
  	return ret;
  }

  static void set_indirect_ea(struct super_block *s, int ano, secno a,
  			    const char *data, int size)

  {
  	hpfs_ea_write(s, a, ano, 0, size, data);
  }
  
  /* Read an extended attribute named 'key' into the provided buffer */
  
  int hpfs_read_ea(struct super_block *s, struct fnode *fnode, char *key,
  		char *buf, int size)
  {
  	unsigned pos;
  	int ano, len;
  	secno a;

  	char ex[4 + 255 + 1 + 8];

  	struct extended_attribute *ea;
  	struct extended_attribute *ea_end = fnode_end_ea(fnode);
  	for (ea = fnode_ea(fnode); ea < ea_end; ea = next_ea(ea))
  		if (!strcmp(ea->name, key)) {

  			if (ea_indirect(ea))

  				goto indirect;

  			if (ea_valuelen(ea) >= size)

  				return -EINVAL;

  			memcpy(buf, ea_data(ea), ea_valuelen(ea));
  			buf[ea_valuelen(ea)] = 0;

  			return 0;
  		}

  	a = le32_to_cpu(fnode->ea_secno);
  	len = le32_to_cpu(fnode->ea_size_l);

  	ano = fnode_in_anode(fnode);

  	pos = 0;
  	while (pos < len) {

  		ea = (struct extended_attribute *)ex;
  		if (pos + 4 > len) {
  			hpfs_error(s, "EAs don't end correctly, %s %08x, len %08x",
  				ano ? "anode" : "sectors", a, len);
  			return -EIO;
  		}
  		if (hpfs_ea_read(s, a, ano, pos, 4, ex)) return -EIO;

  		if (hpfs_ea_read(s, a, ano, pos + 4, ea->namelen + 1 + (ea_indirect(ea) ? 8 : 0), ex + 4))

  			return -EIO;
  		if (!strcmp(ea->name, key)) {

  			if (ea_indirect(ea))

  				goto indirect;

  			if (ea_valuelen(ea) >= size)

  				return -EINVAL;

  			if (hpfs_ea_read(s, a, ano, pos + 4 + ea->namelen + 1, ea_valuelen(ea), buf))

  				return -EIO;

  			buf[ea_valuelen(ea)] = 0;

  			return 0;
  		}

  		pos += ea->namelen + ea_valuelen(ea) + 5;

  	}
  	return -ENOENT;
  indirect:
  	if (ea_len(ea) >= size)
  		return -EINVAL;

  	if (hpfs_ea_read(s, ea_sec(ea), ea_in_anode(ea), 0, ea_len(ea), buf))

  		return -EIO;
  	buf[ea_len(ea)] = 0;
  	return 0;
  }
  
  /* Read an extended attribute named 'key' */
  char *hpfs_get_ea(struct super_block *s, struct fnode *fnode, char *key, int *size)
  {
  	char *ret;
  	unsigned pos;
  	int ano, len;
  	secno a;
  	struct extended_attribute *ea;
  	struct extended_attribute *ea_end = fnode_end_ea(fnode);
  	for (ea = fnode_ea(fnode); ea < ea_end; ea = next_ea(ea))
  		if (!strcmp(ea->name, key)) {

  			if (ea_indirect(ea))
  				return get_indirect_ea(s, ea_in_anode(ea), ea_sec(ea), *size = ea_len(ea));

  			if (!(ret = kmalloc((*size = ea_valuelen(ea)) + 1, GFP_NOFS))) {

  				pr_err("out of memory for EA
  ");

  				return NULL;
  			}

  			memcpy(ret, ea_data(ea), ea_valuelen(ea));
  			ret[ea_valuelen(ea)] = 0;

  			return ret;
  		}

  	a = le32_to_cpu(fnode->ea_secno);
  	len = le32_to_cpu(fnode->ea_size_l);

  	ano = fnode_in_anode(fnode);

  	pos = 0;
  	while (pos < len) {
  		char ex[4 + 255 + 1 + 8];
  		ea = (struct extended_attribute *)ex;
  		if (pos + 4 > len) {
  			hpfs_error(s, "EAs don't end correctly, %s %08x, len %08x",
  				ano ? "anode" : "sectors", a, len);
  			return NULL;
  		}
  		if (hpfs_ea_read(s, a, ano, pos, 4, ex)) return NULL;

  		if (hpfs_ea_read(s, a, ano, pos + 4, ea->namelen + 1 + (ea_indirect(ea) ? 8 : 0), ex + 4))

  			return NULL;
  		if (!strcmp(ea->name, key)) {

  			if (ea_indirect(ea))
  				return get_indirect_ea(s, ea_in_anode(ea), ea_sec(ea), *size = ea_len(ea));

  			if (!(ret = kmalloc((*size = ea_valuelen(ea)) + 1, GFP_NOFS))) {

  				pr_err("out of memory for EA
  ");

  				return NULL;
  			}

  			if (hpfs_ea_read(s, a, ano, pos + 4 + ea->namelen + 1, ea_valuelen(ea), ret)) {

  				kfree(ret);
  				return NULL;
  			}

  			ret[ea_valuelen(ea)] = 0;

  			return ret;
  		}

  		pos += ea->namelen + ea_valuelen(ea) + 5;

  	}
  	return NULL;
  }
  
  /* 
   * Update or create extended attribute 'key' with value 'data'. Note that
   * when this ea exists, it MUST have the same size as size of data.
   * This driver can't change sizes of eas ('cause I just don't need it).
   */

  void hpfs_set_ea(struct inode *inode, struct fnode *fnode, const char *key,
  		 const char *data, int size)

  {
  	fnode_secno fno = inode->i_ino;
  	struct super_block *s = inode->i_sb;
  	unsigned pos;
  	int ano, len;
  	secno a;
  	unsigned char h[4];
  	struct extended_attribute *ea;
  	struct extended_attribute *ea_end = fnode_end_ea(fnode);
  	for (ea = fnode_ea(fnode); ea < ea_end; ea = next_ea(ea))
  		if (!strcmp(ea->name, key)) {

  			if (ea_indirect(ea)) {

  				if (ea_len(ea) == size)

  					set_indirect_ea(s, ea_in_anode(ea), ea_sec(ea), data, size);

  			} else if (ea_valuelen(ea) == size) {

  				memcpy(ea_data(ea), data, size);
  			}
  			return;
  		}

  	a = le32_to_cpu(fnode->ea_secno);
  	len = le32_to_cpu(fnode->ea_size_l);

  	ano = fnode_in_anode(fnode);

  	pos = 0;
  	while (pos < len) {
  		char ex[4 + 255 + 1 + 8];
  		ea = (struct extended_attribute *)ex;
  		if (pos + 4 > len) {
  			hpfs_error(s, "EAs don't end correctly, %s %08x, len %08x",
  				ano ? "anode" : "sectors", a, len);
  			return;
  		}
  		if (hpfs_ea_read(s, a, ano, pos, 4, ex)) return;

  		if (hpfs_ea_read(s, a, ano, pos + 4, ea->namelen + 1 + (ea_indirect(ea) ? 8 : 0), ex + 4))

  			return;
  		if (!strcmp(ea->name, key)) {

  			if (ea_indirect(ea)) {

  				if (ea_len(ea) == size)

  					set_indirect_ea(s, ea_in_anode(ea), ea_sec(ea), data, size);

  			}
  			else {

  				if (ea_valuelen(ea) == size)

  					hpfs_ea_write(s, a, ano, pos + 4 + ea->namelen + 1, size, data);
  			}
  			return;
  		}

  		pos += ea->namelen + ea_valuelen(ea) + 5;

  	}

  	if (!le16_to_cpu(fnode->ea_offs)) {
  		/*if (le16_to_cpu(fnode->ea_size_s)) {

  			hpfs_error(s, "fnode %08x: ea_size_s == %03x, ea_offs == 0",

  				inode->i_ino, le16_to_cpu(fnode->ea_size_s));

  			return;
  		}*/

  		fnode->ea_offs = cpu_to_le16(0xc4);

  	}

  	if (le16_to_cpu(fnode->ea_offs) < 0xc4 || le16_to_cpu(fnode->ea_offs) + le16_to_cpu(fnode->acl_size_s) + le16_to_cpu(fnode->ea_size_s) > 0x200) {

  		hpfs_error(s, "fnode %08lx: ea_offs == %03x, ea_size_s == %03x",
  			(unsigned long)inode->i_ino,

  			le16_to_cpu(fnode->ea_offs), le16_to_cpu(fnode->ea_size_s));

  		return;
  	}

  	if ((le16_to_cpu(fnode->ea_size_s) || !le32_to_cpu(fnode->ea_size_l)) &&
  	     le16_to_cpu(fnode->ea_offs) + le16_to_cpu(fnode->acl_size_s) + le16_to_cpu(fnode->ea_size_s) + strlen(key) + size + 5 <= 0x200) {

  		ea = fnode_end_ea(fnode);
  		*(char *)ea = 0;
  		ea->namelen = strlen(key);

  		ea->valuelen_lo = size;
  		ea->valuelen_hi = size >> 8;

  		strcpy(ea->name, key);
  		memcpy(ea_data(ea), data, size);

  		fnode->ea_size_s = cpu_to_le16(le16_to_cpu(fnode->ea_size_s) + strlen(key) + size + 5);

  		goto ret;
  	}
  	/* Most the code here is 99.9993422% unused. I hope there are no bugs.
  	   But what .. HPFS.IFS has also bugs in ea management. */

  	if (le16_to_cpu(fnode->ea_size_s) && !le32_to_cpu(fnode->ea_size_l)) {

  		secno n;
  		struct buffer_head *bh;
  		char *data;

  		if (!(n = hpfs_alloc_sector(s, fno, 1, 0))) return;

  		if (!(data = hpfs_get_sector(s, n, &bh))) {
  			hpfs_free_sectors(s, n, 1);
  			return;
  		}

  		memcpy(data, fnode_ea(fnode), le16_to_cpu(fnode->ea_size_s));
  		fnode->ea_size_l = cpu_to_le32(le16_to_cpu(fnode->ea_size_s));
  		fnode->ea_size_s = cpu_to_le16(0);
  		fnode->ea_secno = cpu_to_le32(n);

  		fnode->flags &= ~FNODE_anode;

  		mark_buffer_dirty(bh);
  		brelse(bh);
  	}

  	pos = le32_to_cpu(fnode->ea_size_l) + 5 + strlen(key) + size;
  	len = (le32_to_cpu(fnode->ea_size_l) + 511) >> 9;

  	if (pos >= 30000) goto bail;
  	while (((pos + 511) >> 9) > len) {
  		if (!len) {

  			secno q = hpfs_alloc_sector(s, fno, 1, 0);
  			if (!q) goto bail;
  			fnode->ea_secno = cpu_to_le32(q);

  			fnode->flags &= ~FNODE_anode;

  			len++;

  		} else if (!fnode_in_anode(fnode)) {

  			if (hpfs_alloc_if_possible(s, le32_to_cpu(fnode->ea_secno) + len)) {

  				len++;
  			} else {
  				/* Aargh... don't know how to create ea anodes :-( */
  				/*struct buffer_head *bh;
  				struct anode *anode;
  				anode_secno a_s;
  				if (!(anode = hpfs_alloc_anode(s, fno, &a_s, &bh)))
  					goto bail;

  				anode->up = cpu_to_le32(fno);

  				anode->btree.fnode_parent = 1;
  				anode->btree.n_free_nodes--;
  				anode->btree.n_used_nodes++;

  				anode->btree.first_free = cpu_to_le16(le16_to_cpu(anode->btree.first_free) + 12);
  				anode->u.external[0].disk_secno = cpu_to_le32(le32_to_cpu(fnode->ea_secno));
  				anode->u.external[0].file_secno = cpu_to_le32(0);
  				anode->u.external[0].length = cpu_to_le32(len);

  				mark_buffer_dirty(bh);
  				brelse(bh);

  				fnode->flags |= FNODE_anode;

  				fnode->ea_secno = cpu_to_le32(a_s);*/

  				secno new_sec;
  				int i;

  				if (!(new_sec = hpfs_alloc_sector(s, fno, 1, 1 - ((pos + 511) >> 9))))

  					goto bail;
  				for (i = 0; i < len; i++) {
  					struct buffer_head *bh1, *bh2;
  					void *b1, *b2;

  					if (!(b1 = hpfs_map_sector(s, le32_to_cpu(fnode->ea_secno) + i, &bh1, len - i - 1))) {

  						hpfs_free_sectors(s, new_sec, (pos + 511) >> 9);
  						goto bail;
  					}
  					if (!(b2 = hpfs_get_sector(s, new_sec + i, &bh2))) {
  						brelse(bh1);
  						hpfs_free_sectors(s, new_sec, (pos + 511) >> 9);
  						goto bail;
  					}
  					memcpy(b2, b1, 512);
  					brelse(bh1);
  					mark_buffer_dirty(bh2);
  					brelse(bh2);
  				}

  				hpfs_free_sectors(s, le32_to_cpu(fnode->ea_secno), len);
  				fnode->ea_secno = cpu_to_le32(new_sec);

  				len = (pos + 511) >> 9;
  			}
  		}

  		if (fnode_in_anode(fnode)) {

  			if (hpfs_add_sector_to_btree(s, le32_to_cpu(fnode->ea_secno),

  						     0, len) != -1) {
  				len++;
  			} else {
  				goto bail;
  			}
  		}
  	}
  	h[0] = 0;
  	h[1] = strlen(key);
  	h[2] = size & 0xff;
  	h[3] = size >> 8;

  	if (hpfs_ea_write(s, le32_to_cpu(fnode->ea_secno), fnode_in_anode(fnode), le32_to_cpu(fnode->ea_size_l), 4, h)) goto bail;
  	if (hpfs_ea_write(s, le32_to_cpu(fnode->ea_secno), fnode_in_anode(fnode), le32_to_cpu(fnode->ea_size_l) + 4, h[1] + 1, key)) goto bail;
  	if (hpfs_ea_write(s, le32_to_cpu(fnode->ea_secno), fnode_in_anode(fnode), le32_to_cpu(fnode->ea_size_l) + 5 + h[1], size, data)) goto bail;

  	fnode->ea_size_l = cpu_to_le32(pos);

  	ret:
  	hpfs_i(inode)->i_ea_size += 5 + strlen(key) + size;
  	return;
  	bail:

  	if (le32_to_cpu(fnode->ea_secno))

  		if (fnode_in_anode(fnode)) hpfs_truncate_btree(s, le32_to_cpu(fnode->ea_secno), 1, (le32_to_cpu(fnode->ea_size_l) + 511) >> 9);

  		else hpfs_free_sectors(s, le32_to_cpu(fnode->ea_secno) + ((le32_to_cpu(fnode->ea_size_l) + 511) >> 9), len - ((le32_to_cpu(fnode->ea_size_l) + 511) >> 9));
  	else fnode->ea_secno = fnode->ea_size_l = cpu_to_le32(0);

  }