48 static int set_update_marker(struct ubi_device *ubi, struct ubi_volume *vol)  in set_update_marker()  argument
53 	dbg_gen("set update marker for volume %d", vol->vol_id);  in set_update_marker()
55 	if (vol->upd_marker) {  in set_update_marker()
56 		ubi_assert(ubi->vtbl[vol->vol_id].upd_marker);  in set_update_marker()
61 	vtbl_rec = ubi->vtbl[vol->vol_id];  in set_update_marker()
65 	err = ubi_change_vtbl_record(ubi, vol->vol_id, &vtbl_rec);  in set_update_marker()
66 	vol->upd_marker = 1;  in set_update_marker()
81 static int clear_update_marker(struct ubi_device *ubi, struct ubi_volume *vol,  in clear_update_marker()  argument
87 	dbg_gen("clear update marker for volume %d", vol->vol_id);  in clear_update_marker()
89 	vtbl_rec = ubi->vtbl[vol->vol_id];  in clear_update_marker()
90 	ubi_assert(vol->upd_marker && vtbl_rec.upd_marker);  in clear_update_marker()
93 	if (vol->vol_type == UBI_STATIC_VOLUME) {  in clear_update_marker()
94 		vol->corrupted = 0;  in clear_update_marker()
95 		vol->used_bytes = bytes;  in clear_update_marker()
96 		vol->used_ebs = div_u64_rem(bytes, vol->usable_leb_size,  in clear_update_marker()
97 					    &vol->last_eb_bytes);  in clear_update_marker()
98 		if (vol->last_eb_bytes)  in clear_update_marker()
99 			vol->used_ebs += 1;  in clear_update_marker()
101 			vol->last_eb_bytes = vol->usable_leb_size;  in clear_update_marker()
105 	err = ubi_change_vtbl_record(ubi, vol->vol_id, &vtbl_rec);  in clear_update_marker()
106 	vol->upd_marker = 0;  in clear_update_marker()
121 int ubi_start_update(struct ubi_device *ubi, struct ubi_volume *vol,  in ubi_start_update()  argument
126 	dbg_gen("start update of volume %d, %llu bytes", vol->vol_id, bytes);  in ubi_start_update()
127 	ubi_assert(!vol->updating && !vol->changing_leb);  in ubi_start_update()
128 	vol->updating = 1;  in ubi_start_update()
130 	vol->upd_buf = vmalloc(ubi->leb_size);  in ubi_start_update()
131 	if (!vol->upd_buf)  in ubi_start_update()
134 	err = set_update_marker(ubi, vol);  in ubi_start_update()
139 	for (i = 0; i < vol->reserved_pebs; i++) {  in ubi_start_update()
140 		err = ubi_eba_unmap_leb(ubi, vol, i);  in ubi_start_update()
150 		err = clear_update_marker(ubi, vol, 0);  in ubi_start_update()
154 		vfree(vol->upd_buf);  in ubi_start_update()
155 		vol->updating = 0;  in ubi_start_update()
159 	vol->upd_ebs = div_u64(bytes + vol->usable_leb_size - 1,  in ubi_start_update()
160 			       vol->usable_leb_size);  in ubi_start_update()
161 	vol->upd_bytes = bytes;  in ubi_start_update()
162 	vol->upd_received = 0;  in ubi_start_update()
175 int ubi_start_leb_change(struct ubi_device *ubi, struct ubi_volume *vol,  in ubi_start_leb_change()  argument
178 	ubi_assert(!vol->updating && !vol->changing_leb);  in ubi_start_leb_change()
181 		vol->vol_id, req->lnum, req->bytes);  in ubi_start_leb_change()
183 		return ubi_eba_atomic_leb_change(ubi, vol, req->lnum, NULL, 0);  in ubi_start_leb_change()
185 	vol->upd_bytes = req->bytes;  in ubi_start_leb_change()
186 	vol->upd_received = 0;  in ubi_start_leb_change()
187 	vol->changing_leb = 1;  in ubi_start_leb_change()
188 	vol->ch_lnum = req->lnum;  in ubi_start_leb_change()
190 	vol->upd_buf = vmalloc(req->bytes);  in ubi_start_leb_change()
191 	if (!vol->upd_buf)  in ubi_start_leb_change()
226 static int write_leb(struct ubi_device *ubi, struct ubi_volume *vol, int lnum,  in write_leb()  argument
231 	if (vol->vol_type == UBI_DYNAMIC_VOLUME) {  in write_leb()
241 		err = ubi_eba_write_leb(ubi, vol, lnum, buf, 0, len);  in write_leb()
252 		memset(buf + len, 0, vol->usable_leb_size - len);  in write_leb()
253 		err = ubi_eba_write_leb_st(ubi, vol, lnum, buf, len, used_ebs);  in write_leb()
272 int ubi_more_update_data(struct ubi_device *ubi, struct ubi_volume *vol,  in ubi_more_update_data()  argument
283 		count, vol->upd_bytes, vol->upd_received);  in ubi_more_update_data()
288 	lnum = div_u64_rem(vol->upd_received,  vol->usable_leb_size, &offs);  in ubi_more_update_data()
289 	if (vol->upd_received + count > vol->upd_bytes)  in ubi_more_update_data()
290 		to_write = count = vol->upd_bytes - vol->upd_received;  in ubi_more_update_data()
304 		len = vol->usable_leb_size - offs;  in ubi_more_update_data()
308 		err = copy_from_user(vol->upd_buf + offs, buf, len);  in ubi_more_update_data()
312 		if (offs + len == vol->usable_leb_size ||  in ubi_more_update_data()
313 		    vol->upd_received + len == vol->upd_bytes) {  in ubi_more_update_data()
320 			ubi_assert(flush_len <= vol->usable_leb_size);  in ubi_more_update_data()
321 			err = write_leb(ubi, vol, lnum, vol->upd_buf, flush_len,  in ubi_more_update_data()
322 					vol->upd_ebs);  in ubi_more_update_data()
327 		vol->upd_received += len;  in ubi_more_update_data()
338 		if (count > vol->usable_leb_size)  in ubi_more_update_data()
339 			len = vol->usable_leb_size;  in ubi_more_update_data()
343 		err = copy_from_user(vol->upd_buf, buf, len);  in ubi_more_update_data()
347 		if (len == vol->usable_leb_size ||  in ubi_more_update_data()
348 		    vol->upd_received + len == vol->upd_bytes) {  in ubi_more_update_data()
349 			err = write_leb(ubi, vol, lnum, vol->upd_buf,  in ubi_more_update_data()
350 					len, vol->upd_ebs);  in ubi_more_update_data()
355 		vol->upd_received += len;  in ubi_more_update_data()
361 	ubi_assert(vol->upd_received <= vol->upd_bytes);  in ubi_more_update_data()
362 	if (vol->upd_received == vol->upd_bytes) {  in ubi_more_update_data()
367 		err = clear_update_marker(ubi, vol, vol->upd_bytes);  in ubi_more_update_data()
370 		vol->updating = 0;  in ubi_more_update_data()
372 		vfree(vol->upd_buf);  in ubi_more_update_data()
392 int ubi_more_leb_change_data(struct ubi_device *ubi, struct ubi_volume *vol,  in ubi_more_leb_change_data()  argument
398 		count, vol->upd_bytes, vol->upd_received);  in ubi_more_leb_change_data()
403 	if (vol->upd_received + count > vol->upd_bytes)  in ubi_more_leb_change_data()
404 		count = vol->upd_bytes - vol->upd_received;  in ubi_more_leb_change_data()
406 	err = copy_from_user(vol->upd_buf + vol->upd_received, buf, count);  in ubi_more_leb_change_data()
410 	vol->upd_received += count;  in ubi_more_leb_change_data()
412 	if (vol->upd_received == vol->upd_bytes) {  in ubi_more_leb_change_data()
413 		int len = ALIGN((int)vol->upd_bytes, ubi->min_io_size);  in ubi_more_leb_change_data()
415 		memset(vol->upd_buf + vol->upd_bytes, 0xFF,  in ubi_more_leb_change_data()
416 		       len - vol->upd_bytes);  in ubi_more_leb_change_data()
417 		len = ubi_calc_data_len(ubi, vol->upd_buf, len);  in ubi_more_leb_change_data()
418 		err = ubi_eba_atomic_leb_change(ubi, vol, vol->ch_lnum,  in ubi_more_leb_change_data()
419 						vol->upd_buf, len);  in ubi_more_leb_change_data()
424 	ubi_assert(vol->upd_received <= vol->upd_bytes);  in ubi_more_leb_change_data()
425 	if (vol->upd_received == vol->upd_bytes) {  in ubi_more_leb_change_data()
426 		vol->changing_leb = 0;  in ubi_more_leb_change_data()
428 		vfree(vol->upd_buf);  in ubi_more_leb_change_data()