-ifneq ($(CONFIG_TARGET_ROOTFS_INITRAMFS),y)
- ifeq ($(CONFIG_TARGET_ROOTFS_JFFS2),y)
+JFFS2_BLOCKSIZE ?= 64k 128k
+
+define add_jffs2_mark
+ echo -ne '\xde\xad\xc0\xde' >> $(1)
+endef
+
+# pad to 4k, 8k, 64k, 128k 256k and add jffs2 end-of-filesystem mark
+define prepare_generic_squashfs
+ $(STAGING_DIR_HOST)/bin/padjffs2 $(1) 4 8 64 128 256
+endef
+
+
+ifneq ($(CONFIG_TARGET_ROOTFS_INITRAMFS),)
+
+ define Image/BuildKernel
+ cp $(KDIR)/vmlinux.elf $(BIN_DIR)/$(IMG_PREFIX)-vmlinux.elf
+ $(call Image/Build/Initramfs)
+ endef
+
+else
+
+ ifneq ($(CONFIG_TARGET_ROOTFS_JFFS2),)
+ define Image/mkfs/jffs2/sub
+ # FIXME: removing this line will cause strange behaviour in the foreach loop below
+ $(STAGING_DIR_HOST)/bin/mkfs.jffs2 $(JFFS2OPTS) -e $(patsubst %k,%KiB,$(1)) -o $(KDIR)/root.jffs2-$(1) -d $(TARGET_DIR) -v 2>&1 1>/dev/null | awk '/^.+$$$$/'
+ $(call add_jffs2_mark,$(KDIR)/root.jffs2-$(1))
+ $(call Image/Build,jffs2-$(1))
+ endef