@@ -358,6 +358,7 @@ struct ext4_new_group_data {
#define EXT4_IOC_MOVE_EXT _IOWR('f', 15, struct move_extent)
#define EXT4_IOC_ADD_GLOBAL_ALLOC_RULE _IOW('f', 16, struct ext4_alloc_rule)
#define EXT4_IOC_CLR_GLOBAL_ALLOC_RULE _IOW('f', 17, struct ext4_alloc_rule)
+#define EXT4_IOC_ADD_INODE_ALLOC_RULE _IOW('f', 18, struct ext4_alloc_rule)
/*
* ioctl commands in 32 bit emulation
@@ -403,6 +404,11 @@ struct ext4_bg_alloc_rule {
int alloc_flag; /* 0(mandatory) or 1(advisory) */
};
+struct ext4_inode_alloc_rule {
+ struct ext4_alloc_rule alloc_rule;
+ pid_t alloc_pid;
+};
+
/*
* Mount options
*/
@@ -689,6 +695,8 @@ struct ext4_inode_info {
__u16 i_extra_isize;
spinlock_t i_block_reservation_lock;
+
+ struct ext4_inode_alloc_rule *i_alloc_rule;
};
/*
@@ -1389,6 +1397,9 @@ extern void ext4_mb_put_buddy_cache_lock(struct super_block *,
extern int ext4_mb_add_global_arule(struct inode *, struct ext4_alloc_rule *);
extern int ext4_mb_del_global_arule(struct inode *, struct ext4_alloc_rule *);
extern void ext4_mb_release_arule_list(struct ext4_sb_info *);
+extern int ext4_mb_add_inode_arule(struct inode *inode,
+ struct ext4_alloc_rule *arule);
+extern void ext4_mb_del_inode_arule(struct inode *inode);
/* inode.c */
int ext4_forget(handle_t *handle, int is_metadata, struct inode *inode,
@@ -46,6 +46,7 @@ static int ext4_release_file(struct inode *inode, struct file *filp)
{
down_write(&EXT4_I(inode)->i_data_sem);
ext4_discard_preallocations(inode);
+ ext4_mb_del_inode_arule(inode);
up_write(&EXT4_I(inode)->i_data_sem);
}
if (is_dx(inode) && filp->private_data)
@@ -1033,6 +1033,8 @@ got:
ei->i_extra_isize = EXT4_SB(sb)->s_want_extra_isize;
+ ei->i_alloc_rule = NULL;
+
ret = inode;
if (vfs_dq_alloc_inode(inode)) {
err = -EDQUOT;
@@ -292,6 +292,32 @@ setversion_out:
return err;
}
+ case EXT4_IOC_ADD_INODE_ALLOC_RULE: {
+ struct ext4_alloc_rule arule;
+ int err;
+
+ /*
+ * If we do not have the permission to access the inode,
+ * just skip it.
+ */
+ if (!is_owner_or_cap(inode))
+ return -EACCES;
+
+ if (!(filp->f_mode & FMODE_WRITE))
+ return -EBADF;
+
+ if (copy_from_user(&arule,
+ (struct ext4_alloc_rule __user *)arg,
+ sizeof(arule)))
+ return -EFAULT;
+
+ down_write(&ei->i_data_sem);
+ err = ext4_mb_add_inode_arule(inode, &arule);
+ up_write(&ei->i_data_sem);
+
+ return err;
+ }
+
case EXT4_IOC_GROUP_ADD: {
struct ext4_new_group_data input;
struct super_block *sb = inode->i_sb;
@@ -5482,3 +5482,36 @@ out:
write_unlock(&sbi->s_bg_arule_lock);
return ret;
}
+
+int ext4_mb_add_inode_arule(struct inode *inode,
+ struct ext4_alloc_rule *arule)
+{
+ struct ext4_inode_info *ei = EXT4_I(inode);
+ int err;
+
+ err = ext4_mb_check_arule(inode, arule);
+ if (err)
+ return err;
+ if (!ei->i_alloc_rule) {
+ ei->i_alloc_rule = kmalloc(
+ sizeof(struct ext4_inode_alloc_rule), GFP_KERNEL);
+ if (!ei->i_alloc_rule)
+ return -ENOMEM;
+ } else if (ei->i_alloc_rule->alloc_pid != current->pid) {
+ ext4_debug("using allocation rule\n");
+ return -EINVAL;
+ }
+
+ memcpy(&ei->i_alloc_rule->alloc_rule, arule, sizeof(*arule));
+ ei->i_alloc_rule->alloc_pid = current->pid;
+
+ return 0;
+}
+
+void ext4_mb_del_inode_arule(struct inode *inode)
+{
+ struct ext4_inode_info *ei = EXT4_I(inode);
+
+ kfree(ei->i_alloc_rule);
+ ei->i_alloc_rule = NULL;
+}
@@ -686,6 +686,7 @@ static struct inode *ext4_alloc_inode(struct super_block *sb)
ei->i_reserved_meta_blocks = 0;
ei->i_allocated_meta_blocks = 0;
ei->i_delalloc_reserved_flag = 0;
+ ei->i_alloc_rule = NULL;
spin_lock_init(&(ei->i_block_reservation_lock));
return &ei->vfs_inode;
@@ -749,6 +750,7 @@ static void ext4_clear_inode(struct inode *inode)
}
#endif
ext4_discard_preallocations(inode);
+ ext4_mb_del_inode_arule(inode);
if (EXT4_JOURNAL(inode))
jbd2_journal_release_jbd_inode(EXT4_SB(inode->i_sb)->s_journal,
&EXT4_I(inode)->jinode);