1 // SPDX-License-Identifier: GPL-2.0 2 /* 3 * 4 * Copyright (C) 2019-2021 Paragon Software GmbH, All rights reserved. 5 * 6 */ 7 8 #include <linux/blkdev.h> 9 #include <linux/fs.h> 10 #include <linux/random.h> 11 #include <linux/slab.h> 12 13 #include "debug.h" 14 #include "ntfs.h" 15 #include "ntfs_fs.h" 16 17 /* 18 * LOG FILE structs 19 */ 20 21 // clang-format off 22 23 #define MaxLogFileSize 0x100000000ull 24 #define DefaultLogPageSize 4096 25 #define MinLogRecordPages 0x30 26 27 struct RESTART_HDR { 28 struct NTFS_RECORD_HEADER rhdr; // 'RSTR' 29 __le32 sys_page_size; // 0x10: Page size of the system which initialized the log. 30 __le32 page_size; // 0x14: Log page size used for this log file. 31 __le16 ra_off; // 0x18: 32 __le16 minor_ver; // 0x1A: 33 __le16 major_ver; // 0x1C: 34 __le16 fixups[]; 35 }; 36 37 #define LFS_NO_CLIENT 0xffff 38 #define LFS_NO_CLIENT_LE cpu_to_le16(0xffff) 39 40 struct CLIENT_REC { 41 __le64 oldest_lsn; 42 __le64 restart_lsn; // 0x08: 43 __le16 prev_client; // 0x10: 44 __le16 next_client; // 0x12: 45 __le16 seq_num; // 0x14: 46 u8 align[6]; // 0x16: 47 __le32 name_bytes; // 0x1C: In bytes. 48 __le16 name[32]; // 0x20: Name of client. 49 }; 50 51 static_assert(sizeof(struct CLIENT_REC) == 0x60); 52 53 /* Two copies of these will exist at the beginning of the log file */ 54 struct RESTART_AREA { 55 __le64 current_lsn; // 0x00: Current logical end of log file. 56 __le16 log_clients; // 0x08: Maximum number of clients. 57 __le16 client_idx[2]; // 0x0A: Free/use index into the client record arrays. 58 __le16 flags; // 0x0E: See RESTART_SINGLE_PAGE_IO. 59 __le32 seq_num_bits; // 0x10: The number of bits in sequence number. 60 __le16 ra_len; // 0x14: 61 __le16 client_off; // 0x16: 62 __le64 l_size; // 0x18: Usable log file size. 63 __le32 last_lsn_data_len; // 0x20: 64 __le16 rec_hdr_len; // 0x24: Log page data offset. 65 __le16 data_off; // 0x26: Log page data length. 66 __le32 open_log_count; // 0x28: 67 __le32 align[5]; // 0x2C: 68 struct CLIENT_REC clients[]; // 0x40: 69 }; 70 71 struct LOG_REC_HDR { 72 __le16 redo_op; // 0x00: NTFS_LOG_OPERATION 73 __le16 undo_op; // 0x02: NTFS_LOG_OPERATION 74 __le16 redo_off; // 0x04: Offset to Redo record. 75 __le16 redo_len; // 0x06: Redo length. 76 __le16 undo_off; // 0x08: Offset to Undo record. 77 __le16 undo_len; // 0x0A: Undo length. 78 __le16 target_attr; // 0x0C: 79 __le16 lcns_follow; // 0x0E: 80 __le16 record_off; // 0x10: 81 __le16 attr_off; // 0x12: 82 __le16 cluster_off; // 0x14: 83 __le16 reserved; // 0x16: 84 __le64 target_vcn; // 0x18: 85 __le64 page_lcns[]; // 0x20: 86 }; 87 88 static_assert(sizeof(struct LOG_REC_HDR) == 0x20); 89 90 #define RESTART_ENTRY_ALLOCATED 0xFFFFFFFF 91 #define RESTART_ENTRY_ALLOCATED_LE cpu_to_le32(0xFFFFFFFF) 92 93 struct RESTART_TABLE { 94 __le16 size; // 0x00: In bytes 95 __le16 used; // 0x02: Entries 96 __le16 total; // 0x04: Entries 97 __le16 res[3]; // 0x06: 98 __le32 free_goal; // 0x0C: 99 __le32 first_free; // 0x10: 100 __le32 last_free; // 0x14: 101 102 }; 103 104 static_assert(sizeof(struct RESTART_TABLE) == 0x18); 105 106 struct ATTR_NAME_ENTRY { 107 __le16 off; // Offset in the Open attribute Table. 108 __le16 name_bytes; 109 __le16 name[]; 110 }; 111 112 struct OPEN_ATTR_ENRTY { 113 __le32 next; // 0x00: RESTART_ENTRY_ALLOCATED if allocated 114 __le32 bytes_per_index; // 0x04: 115 enum ATTR_TYPE type; // 0x08: 116 u8 is_dirty_pages; // 0x0C: 117 u8 is_attr_name; // 0x0B: Faked field to manage 'ptr' 118 u8 name_len; // 0x0C: Faked field to manage 'ptr' 119 u8 res; 120 struct MFT_REF ref; // 0x10: File Reference of file containing attribute 121 __le64 open_record_lsn; // 0x18: 122 void *ptr; // 0x20: 123 }; 124 125 /* 32 bit version of 'struct OPEN_ATTR_ENRTY' */ 126 struct OPEN_ATTR_ENRTY_32 { 127 __le32 next; // 0x00: RESTART_ENTRY_ALLOCATED if allocated 128 __le32 ptr; // 0x04: 129 struct MFT_REF ref; // 0x08: 130 __le64 open_record_lsn; // 0x10: 131 u8 is_dirty_pages; // 0x18: 132 u8 is_attr_name; // 0x19: 133 u8 res1[2]; 134 enum ATTR_TYPE type; // 0x1C: 135 u8 name_len; // 0x20: In wchar 136 u8 res2[3]; 137 __le32 AttributeName; // 0x24: 138 __le32 bytes_per_index; // 0x28: 139 }; 140 141 #define SIZEOF_OPENATTRIBUTEENTRY0 0x2c 142 // static_assert( 0x2C == sizeof(struct OPEN_ATTR_ENRTY_32) ); 143 static_assert(sizeof(struct OPEN_ATTR_ENRTY) < SIZEOF_OPENATTRIBUTEENTRY0); 144 145 /* 146 * One entry exists in the Dirty Pages Table for each page which is dirty at 147 * the time the Restart Area is written. 148 */ 149 struct DIR_PAGE_ENTRY { 150 __le32 next; // 0x00: RESTART_ENTRY_ALLOCATED if allocated 151 __le32 target_attr; // 0x04: Index into the Open attribute Table 152 __le32 transfer_len; // 0x08: 153 __le32 lcns_follow; // 0x0C: 154 __le64 vcn; // 0x10: Vcn of dirty page 155 __le64 oldest_lsn; // 0x18: 156 __le64 page_lcns[]; // 0x20: 157 }; 158 159 static_assert(sizeof(struct DIR_PAGE_ENTRY) == 0x20); 160 161 /* 32 bit version of 'struct DIR_PAGE_ENTRY' */ 162 struct DIR_PAGE_ENTRY_32 { 163 __le32 next; // 0x00: RESTART_ENTRY_ALLOCATED if allocated 164 __le32 target_attr; // 0x04: Index into the Open attribute Table 165 __le32 transfer_len; // 0x08: 166 __le32 lcns_follow; // 0x0C: 167 __le32 reserved; // 0x10: 168 __le32 vcn_low; // 0x14: Vcn of dirty page 169 __le32 vcn_hi; // 0x18: Vcn of dirty page 170 __le32 oldest_lsn_low; // 0x1C: 171 __le32 oldest_lsn_hi; // 0x1C: 172 __le32 page_lcns_low; // 0x24: 173 __le32 page_lcns_hi; // 0x24: 174 }; 175 176 static_assert(offsetof(struct DIR_PAGE_ENTRY_32, vcn_low) == 0x14); 177 static_assert(sizeof(struct DIR_PAGE_ENTRY_32) == 0x2c); 178 179 enum transact_state { 180 TransactionUninitialized = 0, 181 TransactionActive, 182 TransactionPrepared, 183 TransactionCommitted 184 }; 185 186 struct TRANSACTION_ENTRY { 187 __le32 next; // 0x00: RESTART_ENTRY_ALLOCATED if allocated 188 u8 transact_state; // 0x04: 189 u8 reserved[3]; // 0x05: 190 __le64 first_lsn; // 0x08: 191 __le64 prev_lsn; // 0x10: 192 __le64 undo_next_lsn; // 0x18: 193 __le32 undo_records; // 0x20: Number of undo log records pending abort 194 __le32 undo_len; // 0x24: Total undo size 195 }; 196 197 static_assert(sizeof(struct TRANSACTION_ENTRY) == 0x28); 198 199 struct NTFS_RESTART { 200 __le32 major_ver; // 0x00: 201 __le32 minor_ver; // 0x04: 202 __le64 check_point_start; // 0x08: 203 __le64 open_attr_table_lsn; // 0x10: 204 __le64 attr_names_lsn; // 0x18: 205 __le64 dirty_pages_table_lsn; // 0x20: 206 __le64 transact_table_lsn; // 0x28: 207 __le32 open_attr_len; // 0x30: In bytes 208 __le32 attr_names_len; // 0x34: In bytes 209 __le32 dirty_pages_len; // 0x38: In bytes 210 __le32 transact_table_len; // 0x3C: In bytes 211 }; 212 213 static_assert(sizeof(struct NTFS_RESTART) == 0x40); 214 215 struct NEW_ATTRIBUTE_SIZES { 216 __le64 alloc_size; 217 __le64 valid_size; 218 __le64 data_size; 219 __le64 total_size; 220 }; 221 222 struct BITMAP_RANGE { 223 __le32 bitmap_off; 224 __le32 bits; 225 }; 226 227 struct LCN_RANGE { 228 __le64 lcn; 229 __le64 len; 230 }; 231 232 /* The following type defines the different log record types. */ 233 #define LfsClientRecord cpu_to_le32(1) 234 #define LfsClientRestart cpu_to_le32(2) 235 236 /* This is used to uniquely identify a client for a particular log file. */ 237 struct CLIENT_ID { 238 __le16 seq_num; 239 __le16 client_idx; 240 }; 241 242 /* This is the header that begins every Log Record in the log file. */ 243 struct LFS_RECORD_HDR { 244 __le64 this_lsn; // 0x00: 245 __le64 client_prev_lsn; // 0x08: 246 __le64 client_undo_next_lsn; // 0x10: 247 __le32 client_data_len; // 0x18: 248 struct CLIENT_ID client; // 0x1C: Owner of this log record. 249 __le32 record_type; // 0x20: LfsClientRecord or LfsClientRestart. 250 __le32 transact_id; // 0x24: 251 __le16 flags; // 0x28: LOG_RECORD_MULTI_PAGE 252 u8 align[6]; // 0x2A: 253 }; 254 255 #define LOG_RECORD_MULTI_PAGE cpu_to_le16(1) 256 257 static_assert(sizeof(struct LFS_RECORD_HDR) == 0x30); 258 259 struct LFS_RECORD { 260 __le16 next_record_off; // 0x00: Offset of the free space in the page, 261 u8 align[6]; // 0x02: 262 __le64 last_end_lsn; // 0x08: lsn for the last log record which ends on the page, 263 }; 264 265 static_assert(sizeof(struct LFS_RECORD) == 0x10); 266 267 struct RECORD_PAGE_HDR { 268 struct NTFS_RECORD_HEADER rhdr; // 'RCRD' 269 __le32 rflags; // 0x10: See LOG_PAGE_LOG_RECORD_END 270 __le16 page_count; // 0x14: 271 __le16 page_pos; // 0x16: 272 struct LFS_RECORD record_hdr; // 0x18: 273 __le16 fixups[10]; // 0x28: 274 __le32 file_off; // 0x3c: Used when major version >= 2 275 }; 276 277 // clang-format on 278 279 // Page contains the end of a log record. 280 #define LOG_PAGE_LOG_RECORD_END cpu_to_le32(0x00000001) 281 282 static inline bool is_log_record_end(const struct RECORD_PAGE_HDR *hdr) 283 { 284 return hdr->rflags & LOG_PAGE_LOG_RECORD_END; 285 } 286 287 static_assert(offsetof(struct RECORD_PAGE_HDR, file_off) == 0x3c); 288 289 /* 290 * END of NTFS LOG structures 291 */ 292 293 /* Define some tuning parameters to keep the restart tables a reasonable size. */ 294 #define INITIAL_NUMBER_TRANSACTIONS 5 295 296 enum NTFS_LOG_OPERATION { 297 298 Noop = 0x00, 299 CompensationLogRecord = 0x01, 300 InitializeFileRecordSegment = 0x02, 301 DeallocateFileRecordSegment = 0x03, 302 WriteEndOfFileRecordSegment = 0x04, 303 CreateAttribute = 0x05, 304 DeleteAttribute = 0x06, 305 UpdateResidentValue = 0x07, 306 UpdateNonresidentValue = 0x08, 307 UpdateMappingPairs = 0x09, 308 DeleteDirtyClusters = 0x0A, 309 SetNewAttributeSizes = 0x0B, 310 AddIndexEntryRoot = 0x0C, 311 DeleteIndexEntryRoot = 0x0D, 312 AddIndexEntryAllocation = 0x0E, 313 DeleteIndexEntryAllocation = 0x0F, 314 WriteEndOfIndexBuffer = 0x10, 315 SetIndexEntryVcnRoot = 0x11, 316 SetIndexEntryVcnAllocation = 0x12, 317 UpdateFileNameRoot = 0x13, 318 UpdateFileNameAllocation = 0x14, 319 SetBitsInNonresidentBitMap = 0x15, 320 ClearBitsInNonresidentBitMap = 0x16, 321 HotFix = 0x17, 322 EndTopLevelAction = 0x18, 323 PrepareTransaction = 0x19, 324 CommitTransaction = 0x1A, 325 ForgetTransaction = 0x1B, 326 OpenNonresidentAttribute = 0x1C, 327 OpenAttributeTableDump = 0x1D, 328 AttributeNamesDump = 0x1E, 329 DirtyPageTableDump = 0x1F, 330 TransactionTableDump = 0x20, 331 UpdateRecordDataRoot = 0x21, 332 UpdateRecordDataAllocation = 0x22, 333 334 UpdateRelativeDataInIndex = 335 0x23, // NtOfsRestartUpdateRelativeDataInIndex 336 UpdateRelativeDataInIndex2 = 0x24, 337 ZeroEndOfFileRecord = 0x25, 338 }; 339 340 /* 341 * Array for log records which require a target attribute. 342 * A true indicates that the corresponding restart operation 343 * requires a target attribute. 344 */ 345 static const u8 AttributeRequired[] = { 346 0xFC, 0xFB, 0xFF, 0x10, 0x06, 347 }; 348 349 static inline bool is_target_required(u16 op) 350 { 351 bool ret = op <= UpdateRecordDataAllocation && 352 (AttributeRequired[op >> 3] >> (op & 7) & 1); 353 return ret; 354 } 355 356 static inline bool can_skip_action(enum NTFS_LOG_OPERATION op) 357 { 358 switch (op) { 359 case Noop: 360 case DeleteDirtyClusters: 361 case HotFix: 362 case EndTopLevelAction: 363 case PrepareTransaction: 364 case CommitTransaction: 365 case ForgetTransaction: 366 case CompensationLogRecord: 367 case OpenNonresidentAttribute: 368 case OpenAttributeTableDump: 369 case AttributeNamesDump: 370 case DirtyPageTableDump: 371 case TransactionTableDump: 372 return true; 373 default: 374 return false; 375 } 376 } 377 378 enum { lcb_ctx_undo_next, lcb_ctx_prev, lcb_ctx_next }; 379 380 /* Bytes per restart table. */ 381 static inline u32 bytes_per_rt(const struct RESTART_TABLE *rt) 382 { 383 return le16_to_cpu(rt->used) * le16_to_cpu(rt->size) + 384 sizeof(struct RESTART_TABLE); 385 } 386 387 /* Log record length. */ 388 static inline u32 lrh_length(const struct LOG_REC_HDR *lr) 389 { 390 u16 t16 = le16_to_cpu(lr->lcns_follow); 391 392 return struct_size(lr, page_lcns, max_t(u16, 1, t16)); 393 } 394 395 struct lcb { 396 struct LFS_RECORD_HDR *lrh; // Log record header of the current lsn. 397 struct LOG_REC_HDR *log_rec; 398 u32 ctx_mode; // lcb_ctx_undo_next/lcb_ctx_prev/lcb_ctx_next 399 struct CLIENT_ID client; 400 bool alloc; // If true the we should deallocate 'log_rec'. 401 }; 402 403 static void lcb_put(struct lcb *lcb) 404 { 405 if (lcb->alloc) 406 kfree(lcb->log_rec); 407 kfree(lcb->lrh); 408 kfree(lcb); 409 } 410 411 /* Find the oldest lsn from active clients. */ 412 static inline void oldest_client_lsn(const struct CLIENT_REC *ca, 413 __le16 next_client, u64 *oldest_lsn) 414 { 415 while (next_client != LFS_NO_CLIENT_LE) { 416 const struct CLIENT_REC *cr = ca + le16_to_cpu(next_client); 417 u64 lsn = le64_to_cpu(cr->oldest_lsn); 418 419 /* Ignore this block if it's oldest lsn is 0. */ 420 if (lsn && lsn < *oldest_lsn) 421 *oldest_lsn = lsn; 422 423 next_client = cr->next_client; 424 } 425 } 426 427 static inline bool is_rst_page_hdr_valid(u32 file_off, 428 const struct RESTART_HDR *rhdr) 429 { 430 u32 sys_page = le32_to_cpu(rhdr->sys_page_size); 431 u32 page_size = le32_to_cpu(rhdr->page_size); 432 u32 end_usa; 433 u16 ro; 434 435 if (sys_page < SECTOR_SIZE || page_size < SECTOR_SIZE || 436 sys_page & (sys_page - 1) || page_size & (page_size - 1)) { 437 return false; 438 } 439 440 /* Check that if the file offset isn't 0, it is the system page size. */ 441 if (file_off && file_off != sys_page) 442 return false; 443 444 /* Check support version 1.1+. */ 445 if (le16_to_cpu(rhdr->major_ver) <= 1 && !rhdr->minor_ver) 446 return false; 447 448 if (le16_to_cpu(rhdr->major_ver) > 2) 449 return false; 450 451 ro = le16_to_cpu(rhdr->ra_off); 452 if (!IS_ALIGNED(ro, 8) || ro > sys_page) 453 return false; 454 455 end_usa = ((sys_page >> SECTOR_SHIFT) + 1) * sizeof(short); 456 end_usa += le16_to_cpu(rhdr->rhdr.fix_off); 457 458 if (ro < end_usa) 459 return false; 460 461 return true; 462 } 463 464 static inline bool is_rst_area_valid(const struct RESTART_HDR *rhdr) 465 { 466 const struct RESTART_AREA *ra; 467 u16 cl, fl, ul; 468 u32 off, l_size, seq_bits; 469 u16 ro = le16_to_cpu(rhdr->ra_off); 470 u32 sys_page = le32_to_cpu(rhdr->sys_page_size); 471 472 if (ro + offsetof(struct RESTART_AREA, l_size) > 473 SECTOR_SIZE - sizeof(short)) 474 return false; 475 476 ra = Add2Ptr(rhdr, ro); 477 cl = le16_to_cpu(ra->log_clients); 478 479 if (cl > 1) 480 return false; 481 482 off = le16_to_cpu(ra->client_off); 483 484 if (!IS_ALIGNED(off, 8) || ro + off > SECTOR_SIZE - sizeof(short)) 485 return false; 486 487 off += cl * sizeof(struct CLIENT_REC); 488 489 if (off > sys_page) 490 return false; 491 492 /* 493 * Check the restart length field and whether the entire 494 * restart area is contained that length. 495 */ 496 if (le16_to_cpu(rhdr->ra_off) + le16_to_cpu(ra->ra_len) > sys_page || 497 off > le16_to_cpu(ra->ra_len)) { 498 return false; 499 } 500 501 /* 502 * As a final check make sure that the use list and the free list 503 * are either empty or point to a valid client. 504 */ 505 fl = le16_to_cpu(ra->client_idx[0]); 506 ul = le16_to_cpu(ra->client_idx[1]); 507 if ((fl != LFS_NO_CLIENT && fl >= cl) || 508 (ul != LFS_NO_CLIENT && ul >= cl)) 509 return false; 510 511 /* Make sure the sequence number bits match the log file size. */ 512 l_size = le64_to_cpu(ra->l_size); 513 514 seq_bits = sizeof(u64) * 8 + 3; 515 while (l_size) { 516 l_size >>= 1; 517 seq_bits -= 1; 518 } 519 520 if (seq_bits != le32_to_cpu(ra->seq_num_bits)) 521 return false; 522 523 /* The log page data offset and record header length must be quad-aligned. */ 524 if (!IS_ALIGNED(le16_to_cpu(ra->data_off), 8) || 525 !IS_ALIGNED(le16_to_cpu(ra->rec_hdr_len), 8)) 526 return false; 527 528 return true; 529 } 530 531 static inline bool is_client_area_valid(const struct RESTART_HDR *rhdr, 532 bool usa_error) 533 { 534 u16 ro = le16_to_cpu(rhdr->ra_off); 535 const struct RESTART_AREA *ra = Add2Ptr(rhdr, ro); 536 u16 ra_len = le16_to_cpu(ra->ra_len); 537 const struct CLIENT_REC *ca; 538 u32 i; 539 540 if (usa_error && ra_len + ro > SECTOR_SIZE - sizeof(short)) 541 return false; 542 543 /* Find the start of the client array. */ 544 ca = Add2Ptr(ra, le16_to_cpu(ra->client_off)); 545 546 /* 547 * Start with the free list. 548 * Check that all the clients are valid and that there isn't a cycle. 549 * Do the in-use list on the second pass. 550 */ 551 for (i = 0; i < 2; i++) { 552 u16 client_idx = le16_to_cpu(ra->client_idx[i]); 553 bool first_client = true; 554 u16 clients = le16_to_cpu(ra->log_clients); 555 556 while (client_idx != LFS_NO_CLIENT) { 557 const struct CLIENT_REC *cr; 558 559 if (!clients || 560 client_idx >= le16_to_cpu(ra->log_clients)) 561 return false; 562 563 clients -= 1; 564 cr = ca + client_idx; 565 566 client_idx = le16_to_cpu(cr->next_client); 567 568 if (first_client) { 569 first_client = false; 570 if (cr->prev_client != LFS_NO_CLIENT_LE) 571 return false; 572 } 573 } 574 } 575 576 return true; 577 } 578 579 /* 580 * remove_client 581 * 582 * Remove a client record from a client record list an restart area. 583 */ 584 static inline void remove_client(struct CLIENT_REC *ca, 585 const struct CLIENT_REC *cr, __le16 *head) 586 { 587 if (cr->prev_client == LFS_NO_CLIENT_LE) 588 *head = cr->next_client; 589 else 590 ca[le16_to_cpu(cr->prev_client)].next_client = cr->next_client; 591 592 if (cr->next_client != LFS_NO_CLIENT_LE) 593 ca[le16_to_cpu(cr->next_client)].prev_client = cr->prev_client; 594 } 595 596 /* 597 * add_client - Add a client record to the start of a list. 598 */ 599 static inline void add_client(struct CLIENT_REC *ca, u16 index, __le16 *head) 600 { 601 struct CLIENT_REC *cr = ca + index; 602 603 cr->prev_client = LFS_NO_CLIENT_LE; 604 cr->next_client = *head; 605 606 if (*head != LFS_NO_CLIENT_LE) 607 ca[le16_to_cpu(*head)].prev_client = cpu_to_le16(index); 608 609 *head = cpu_to_le16(index); 610 } 611 612 /* 613 * Enumerate restart table. 614 * 615 * @t - table to enumerate. 616 * @c - current enumerated element. 617 * 618 * enumeration starts with @c == NULL 619 * returns next element or NULL 620 */ 621 static inline void *enum_rstbl(struct RESTART_TABLE *t, void *c) 622 { 623 __le32 *e; 624 u32 bprt; 625 u16 rsize; 626 627 if (!t) 628 return NULL; 629 630 rsize = le16_to_cpu(t->size); 631 632 if (!c) { 633 /* start enumeration. */ 634 if (!t->total) 635 return NULL; 636 e = Add2Ptr(t, sizeof(struct RESTART_TABLE)); 637 } else { 638 e = Add2Ptr(c, rsize); 639 } 640 641 /* Loop until we hit the first one allocated, or the end of the list. */ 642 for (bprt = bytes_per_rt(t); PtrOffset(t, e) < bprt; 643 e = Add2Ptr(e, rsize)) { 644 if (*e == RESTART_ENTRY_ALLOCATED_LE) 645 return e; 646 } 647 return NULL; 648 } 649 650 /* 651 * find_dp - Search for a @vcn in Dirty Page Table. 652 */ 653 static inline struct DIR_PAGE_ENTRY *find_dp(struct RESTART_TABLE *dptbl, 654 u32 target_attr, u64 vcn) 655 { 656 __le32 ta = cpu_to_le32(target_attr); 657 struct DIR_PAGE_ENTRY *dp = NULL; 658 659 while ((dp = enum_rstbl(dptbl, dp))) { 660 u64 dp_vcn = le64_to_cpu(dp->vcn); 661 662 if (dp->target_attr == ta && vcn >= dp_vcn && 663 vcn < dp_vcn + le32_to_cpu(dp->lcns_follow)) { 664 return dp; 665 } 666 } 667 return NULL; 668 } 669 670 static inline u32 norm_file_page(u32 page_size, u32 *l_size, bool use_default) 671 { 672 if (use_default) 673 page_size = DefaultLogPageSize; 674 675 /* Round the file size down to a system page boundary. */ 676 *l_size &= ~(page_size - 1); 677 678 /* File should contain at least 2 restart pages and MinLogRecordPages pages. */ 679 if (*l_size < (MinLogRecordPages + 2) * page_size) 680 return 0; 681 682 return page_size; 683 } 684 685 static bool check_log_rec(const struct LOG_REC_HDR *lr, u32 bytes, u32 tr, 686 u32 bytes_per_attr_entry) 687 { 688 u16 t16; 689 690 if (bytes < sizeof(struct LOG_REC_HDR)) 691 return false; 692 if (!tr) 693 return false; 694 695 if ((tr - sizeof(struct RESTART_TABLE)) % 696 sizeof(struct TRANSACTION_ENTRY)) 697 return false; 698 699 if (le16_to_cpu(lr->redo_off) & 7) 700 return false; 701 702 if (le16_to_cpu(lr->undo_off) & 7) 703 return false; 704 705 if (lr->target_attr) 706 goto check_lcns; 707 708 if (is_target_required(le16_to_cpu(lr->redo_op))) 709 return false; 710 711 if (is_target_required(le16_to_cpu(lr->undo_op))) 712 return false; 713 714 check_lcns: 715 if (!lr->lcns_follow) 716 goto check_length; 717 718 t16 = le16_to_cpu(lr->target_attr); 719 if ((t16 - sizeof(struct RESTART_TABLE)) % bytes_per_attr_entry) 720 return false; 721 722 check_length: 723 if (bytes < lrh_length(lr)) 724 return false; 725 726 return true; 727 } 728 729 static bool check_rstbl(const struct RESTART_TABLE *rt, size_t bytes) 730 { 731 u32 ts; 732 u32 i, off; 733 u16 rsize = le16_to_cpu(rt->size); 734 u16 ne = le16_to_cpu(rt->used); 735 u32 ff = le32_to_cpu(rt->first_free); 736 u32 lf = le32_to_cpu(rt->last_free); 737 738 ts = rsize * ne + sizeof(struct RESTART_TABLE); 739 740 if (!rsize || rsize > bytes || 741 rsize + sizeof(struct RESTART_TABLE) > bytes || bytes < ts || 742 le16_to_cpu(rt->total) > ne || ff > ts - sizeof(__le32) || 743 lf > ts - sizeof(__le32) || 744 (ff && ff < sizeof(struct RESTART_TABLE)) || 745 (lf && lf < sizeof(struct RESTART_TABLE))) { 746 return false; 747 } 748 749 /* 750 * Verify each entry is either allocated or points 751 * to a valid offset the table. 752 */ 753 for (i = 0; i < ne; i++) { 754 off = le32_to_cpu(*(__le32 *)Add2Ptr( 755 rt, i * rsize + sizeof(struct RESTART_TABLE))); 756 757 if (off != RESTART_ENTRY_ALLOCATED && off && 758 (off < sizeof(struct RESTART_TABLE) || 759 ((off - sizeof(struct RESTART_TABLE)) % rsize))) { 760 return false; 761 } 762 } 763 764 /* 765 * Walk through the list headed by the first entry to make 766 * sure none of the entries are currently being used. 767 */ 768 for (off = ff; off;) { 769 if (off == RESTART_ENTRY_ALLOCATED) 770 return false; 771 772 off = le32_to_cpu(*(__le32 *)Add2Ptr(rt, off)); 773 774 if (off > ts - sizeof(__le32)) 775 return false; 776 } 777 778 return true; 779 } 780 781 /* 782 * free_rsttbl_idx - Free a previously allocated index a Restart Table. 783 */ 784 static inline void free_rsttbl_idx(struct RESTART_TABLE *rt, u32 off) 785 { 786 __le32 *e; 787 u32 lf = le32_to_cpu(rt->last_free); 788 __le32 off_le = cpu_to_le32(off); 789 790 e = Add2Ptr(rt, off); 791 792 if (off < le32_to_cpu(rt->free_goal)) { 793 *e = rt->first_free; 794 rt->first_free = off_le; 795 if (!lf) 796 rt->last_free = off_le; 797 } else { 798 if (lf) 799 *(__le32 *)Add2Ptr(rt, lf) = off_le; 800 else 801 rt->first_free = off_le; 802 803 rt->last_free = off_le; 804 *e = 0; 805 } 806 807 le16_sub_cpu(&rt->total, 1); 808 } 809 810 static inline struct RESTART_TABLE *init_rsttbl(u16 esize, u16 used) 811 { 812 __le32 *e, *last_free; 813 u32 off; 814 u32 bytes = esize * used + sizeof(struct RESTART_TABLE); 815 u32 lf = sizeof(struct RESTART_TABLE) + (used - 1) * esize; 816 struct RESTART_TABLE *t = kzalloc(bytes, GFP_NOFS); 817 818 if (!t) 819 return NULL; 820 821 t->size = cpu_to_le16(esize); 822 t->used = cpu_to_le16(used); 823 t->free_goal = cpu_to_le32(~0u); 824 t->first_free = cpu_to_le32(sizeof(struct RESTART_TABLE)); 825 t->last_free = cpu_to_le32(lf); 826 827 e = (__le32 *)(t + 1); 828 last_free = Add2Ptr(t, lf); 829 830 for (off = sizeof(struct RESTART_TABLE) + esize; e < last_free; 831 e = Add2Ptr(e, esize), off += esize) { 832 *e = cpu_to_le32(off); 833 } 834 return t; 835 } 836 837 static inline struct RESTART_TABLE *extend_rsttbl(struct RESTART_TABLE *tbl, 838 u32 add, u32 free_goal) 839 { 840 u16 esize = le16_to_cpu(tbl->size); 841 __le32 osize = cpu_to_le32(bytes_per_rt(tbl)); 842 u32 used = le16_to_cpu(tbl->used); 843 struct RESTART_TABLE *rt; 844 845 rt = init_rsttbl(esize, used + add); 846 if (!rt) 847 return NULL; 848 849 memcpy(rt + 1, tbl + 1, esize * used); 850 851 rt->free_goal = free_goal == ~0u ? 852 cpu_to_le32(~0u) : 853 cpu_to_le32(sizeof(struct RESTART_TABLE) + 854 free_goal * esize); 855 856 if (tbl->first_free) { 857 rt->first_free = tbl->first_free; 858 *(__le32 *)Add2Ptr(rt, le32_to_cpu(tbl->last_free)) = osize; 859 } else { 860 rt->first_free = osize; 861 } 862 863 rt->total = tbl->total; 864 865 kfree(tbl); 866 return rt; 867 } 868 869 /* 870 * alloc_rsttbl_idx 871 * 872 * Allocate an index from within a previously initialized Restart Table. 873 */ 874 static inline void *alloc_rsttbl_idx(struct RESTART_TABLE **tbl) 875 { 876 u32 off; 877 __le32 *e; 878 struct RESTART_TABLE *t = *tbl; 879 880 if (!t->first_free) { 881 *tbl = t = extend_rsttbl(t, 16, ~0u); 882 if (!t) 883 return NULL; 884 } 885 886 off = le32_to_cpu(t->first_free); 887 888 /* Dequeue this entry and zero it. */ 889 e = Add2Ptr(t, off); 890 891 t->first_free = *e; 892 893 memset(e, 0, le16_to_cpu(t->size)); 894 895 *e = RESTART_ENTRY_ALLOCATED_LE; 896 897 /* If list is going empty, then we fix the last_free as well. */ 898 if (!t->first_free) 899 t->last_free = 0; 900 901 le16_add_cpu(&t->total, 1); 902 903 return Add2Ptr(t, off); 904 } 905 906 /* 907 * alloc_rsttbl_from_idx 908 * 909 * Allocate a specific index from within a previously initialized Restart Table. 910 */ 911 static inline void *alloc_rsttbl_from_idx(struct RESTART_TABLE **tbl, u32 vbo) 912 { 913 u32 off; 914 __le32 *e; 915 struct RESTART_TABLE *rt = *tbl; 916 u32 bytes = bytes_per_rt(rt); 917 u16 esize = le16_to_cpu(rt->size); 918 919 /* If the entry is not the table, we will have to extend the table. */ 920 if (vbo >= bytes) { 921 /* 922 * Extend the size by computing the number of entries between 923 * the existing size and the desired index and adding 1 to that. 924 */ 925 u32 bytes2idx = vbo - bytes; 926 927 /* 928 * There should always be an integral number of entries 929 * being added. Now extend the table. 930 */ 931 *tbl = rt = extend_rsttbl(rt, bytes2idx / esize + 1, bytes); 932 if (!rt) 933 return NULL; 934 } 935 936 /* See if the entry is already allocated, and just return if it is. */ 937 e = Add2Ptr(rt, vbo); 938 939 if (*e == RESTART_ENTRY_ALLOCATED_LE) 940 return e; 941 942 /* 943 * Walk through the table, looking for the entry we're 944 * interested and the previous entry. 945 */ 946 off = le32_to_cpu(rt->first_free); 947 e = Add2Ptr(rt, off); 948 949 if (off == vbo) { 950 /* this is a match */ 951 rt->first_free = *e; 952 goto skip_looking; 953 } 954 955 /* 956 * Need to walk through the list looking for the predecessor 957 * of our entry. 958 */ 959 for (;;) { 960 /* Remember the entry just found */ 961 u32 last_off = off; 962 __le32 *last_e = e; 963 964 /* Should never run of entries. */ 965 966 /* Lookup up the next entry the list. */ 967 off = le32_to_cpu(*last_e); 968 e = Add2Ptr(rt, off); 969 970 /* If this is our match we are done. */ 971 if (off == vbo) { 972 *last_e = *e; 973 974 /* 975 * If this was the last entry, we update that 976 * table as well. 977 */ 978 if (le32_to_cpu(rt->last_free) == off) 979 rt->last_free = cpu_to_le32(last_off); 980 break; 981 } 982 } 983 984 skip_looking: 985 /* If the list is now empty, we fix the last_free as well. */ 986 if (!rt->first_free) 987 rt->last_free = 0; 988 989 /* Zero this entry. */ 990 memset(e, 0, esize); 991 *e = RESTART_ENTRY_ALLOCATED_LE; 992 993 le16_add_cpu(&rt->total, 1); 994 995 return e; 996 } 997 998 struct restart_info { 999 u64 last_lsn; 1000 struct RESTART_HDR *r_page; 1001 u32 vbo; 1002 bool chkdsk_was_run; 1003 bool valid_page; 1004 bool initialized; 1005 bool restart; 1006 }; 1007 1008 #define RESTART_SINGLE_PAGE_IO cpu_to_le16(0x0001) 1009 1010 #define NTFSLOG_WRAPPED 0x00000001 1011 #define NTFSLOG_MULTIPLE_PAGE_IO 0x00000002 1012 #define NTFSLOG_NO_LAST_LSN 0x00000004 1013 #define NTFSLOG_REUSE_TAIL 0x00000010 1014 #define NTFSLOG_NO_OLDEST_LSN 0x00000020 1015 1016 /* Helper struct to work with NTFS $LogFile. */ 1017 struct ntfs_log { 1018 struct ntfs_inode *ni; 1019 1020 u32 l_size; 1021 u32 orig_file_size; 1022 u32 sys_page_size; 1023 u32 sys_page_mask; 1024 u32 page_size; 1025 u32 page_mask; // page_size - 1 1026 u8 page_bits; 1027 struct RECORD_PAGE_HDR *one_page_buf; 1028 1029 struct RESTART_TABLE *open_attr_tbl; 1030 u32 transaction_id; 1031 u32 clst_per_page; 1032 1033 u32 first_page; 1034 u32 next_page; 1035 u32 ra_off; 1036 u32 data_off; 1037 u32 restart_size; 1038 u32 data_size; 1039 u16 record_header_len; 1040 u64 seq_num; 1041 u32 seq_num_bits; 1042 u32 file_data_bits; 1043 u32 seq_num_mask; /* (1 << file_data_bits) - 1 */ 1044 1045 struct RESTART_AREA *ra; /* In-memory image of the next restart area. */ 1046 u32 ra_size; /* The usable size of the restart area. */ 1047 1048 /* 1049 * If true, then the in-memory restart area is to be written 1050 * to the first position on the disk. 1051 */ 1052 bool init_ra; 1053 bool set_dirty; /* True if we need to set dirty flag. */ 1054 1055 u64 oldest_lsn; 1056 1057 u32 oldest_lsn_off; 1058 u64 last_lsn; 1059 1060 u32 total_avail; 1061 u32 total_avail_pages; 1062 u32 total_undo_commit; 1063 u32 max_current_avail; 1064 u32 current_avail; 1065 u32 reserved; 1066 1067 short major_ver; 1068 short minor_ver; 1069 1070 u32 l_flags; /* See NTFSLOG_XXX */ 1071 u32 current_openlog_count; /* On-disk value for open_log_count. */ 1072 1073 struct CLIENT_ID client_id; 1074 u32 client_undo_commit; 1075 1076 struct restart_info rst_info, rst_info2; 1077 }; 1078 1079 static inline u32 lsn_to_vbo(struct ntfs_log *log, const u64 lsn) 1080 { 1081 u32 vbo = (lsn << log->seq_num_bits) >> (log->seq_num_bits - 3); 1082 1083 return vbo; 1084 } 1085 1086 /* Compute the offset in the log file of the next log page. */ 1087 static inline u32 next_page_off(struct ntfs_log *log, u32 off) 1088 { 1089 off = (off & ~log->sys_page_mask) + log->page_size; 1090 return off >= log->l_size ? log->first_page : off; 1091 } 1092 1093 static inline u32 lsn_to_page_off(struct ntfs_log *log, u64 lsn) 1094 { 1095 return (((u32)lsn) << 3) & log->page_mask; 1096 } 1097 1098 static inline u64 vbo_to_lsn(struct ntfs_log *log, u32 off, u64 Seq) 1099 { 1100 return (off >> 3) + (Seq << log->file_data_bits); 1101 } 1102 1103 static inline bool is_lsn_in_file(struct ntfs_log *log, u64 lsn) 1104 { 1105 return lsn >= log->oldest_lsn && 1106 lsn <= le64_to_cpu(log->ra->current_lsn); 1107 } 1108 1109 static inline u32 hdr_file_off(struct ntfs_log *log, 1110 struct RECORD_PAGE_HDR *hdr) 1111 { 1112 if (log->major_ver < 2) 1113 return le64_to_cpu(hdr->rhdr.lsn); 1114 1115 return le32_to_cpu(hdr->file_off); 1116 } 1117 1118 static inline u64 base_lsn(struct ntfs_log *log, 1119 const struct RECORD_PAGE_HDR *hdr, u64 lsn) 1120 { 1121 u64 h_lsn = le64_to_cpu(hdr->rhdr.lsn); 1122 u64 ret = (((h_lsn >> log->file_data_bits) + 1123 (lsn < (lsn_to_vbo(log, h_lsn) & ~log->page_mask) ? 1 : 0)) 1124 << log->file_data_bits) + 1125 ((((is_log_record_end(hdr) && 1126 h_lsn <= le64_to_cpu(hdr->record_hdr.last_end_lsn)) ? 1127 le16_to_cpu(hdr->record_hdr.next_record_off) : 1128 log->page_size) + 1129 lsn) >> 1130 3); 1131 1132 return ret; 1133 } 1134 1135 static inline bool verify_client_lsn(struct ntfs_log *log, 1136 const struct CLIENT_REC *client, u64 lsn) 1137 { 1138 return lsn >= le64_to_cpu(client->oldest_lsn) && 1139 lsn <= le64_to_cpu(log->ra->current_lsn) && lsn; 1140 } 1141 1142 static int read_log_page(struct ntfs_log *log, u32 vbo, 1143 struct RECORD_PAGE_HDR **buffer, bool *usa_error) 1144 { 1145 int err = 0; 1146 u32 page_idx = vbo >> log->page_bits; 1147 u32 page_off = vbo & log->page_mask; 1148 u32 bytes = log->page_size - page_off; 1149 void *to_free = NULL; 1150 u32 page_vbo = page_idx << log->page_bits; 1151 struct RECORD_PAGE_HDR *page_buf; 1152 struct ntfs_inode *ni = log->ni; 1153 bool bBAAD; 1154 1155 if (vbo >= log->l_size) 1156 return -EINVAL; 1157 1158 if (!*buffer) { 1159 to_free = kmalloc(log->page_size, GFP_NOFS); 1160 if (!to_free) 1161 return -ENOMEM; 1162 *buffer = to_free; 1163 } 1164 1165 page_buf = page_off ? log->one_page_buf : *buffer; 1166 1167 err = ntfs_read_run_nb(ni->mi.sbi, &ni->file.run, page_vbo, page_buf, 1168 log->page_size, NULL); 1169 if (err) 1170 goto out; 1171 1172 if (page_buf->rhdr.sign != NTFS_FFFF_SIGNATURE) 1173 ntfs_fix_post_read(&page_buf->rhdr, PAGE_SIZE, false); 1174 1175 if (page_buf != *buffer) 1176 memcpy(*buffer, Add2Ptr(page_buf, page_off), bytes); 1177 1178 bBAAD = page_buf->rhdr.sign == NTFS_BAAD_SIGNATURE; 1179 1180 if (usa_error) 1181 *usa_error = bBAAD; 1182 /* Check that the update sequence array for this page is valid */ 1183 /* If we don't allow errors, raise an error status */ 1184 else if (bBAAD) 1185 err = -EINVAL; 1186 1187 out: 1188 if (err && to_free) { 1189 kfree(to_free); 1190 *buffer = NULL; 1191 } 1192 1193 return err; 1194 } 1195 1196 /* 1197 * log_read_rst 1198 * 1199 * It walks through 512 blocks of the file looking for a valid 1200 * restart page header. It will stop the first time we find a 1201 * valid page header. 1202 */ 1203 static int log_read_rst(struct ntfs_log *log, bool first, 1204 struct restart_info *info) 1205 { 1206 u32 skip; 1207 u64 vbo; 1208 struct RESTART_HDR *r_page = NULL; 1209 1210 /* Determine which restart area we are looking for. */ 1211 if (first) { 1212 vbo = 0; 1213 skip = 512; 1214 } else { 1215 vbo = 512; 1216 skip = 0; 1217 } 1218 1219 /* Loop continuously until we succeed. */ 1220 for (; vbo < log->l_size; vbo = 2 * vbo + skip, skip = 0) { 1221 bool usa_error; 1222 bool brst, bchk; 1223 struct RESTART_AREA *ra; 1224 1225 /* Read a page header at the current offset. */ 1226 if (read_log_page(log, vbo, (struct RECORD_PAGE_HDR **)&r_page, 1227 &usa_error)) { 1228 /* Ignore any errors. */ 1229 continue; 1230 } 1231 1232 /* Exit if the signature is a log record page. */ 1233 if (r_page->rhdr.sign == NTFS_RCRD_SIGNATURE) { 1234 info->initialized = true; 1235 break; 1236 } 1237 1238 brst = r_page->rhdr.sign == NTFS_RSTR_SIGNATURE; 1239 bchk = r_page->rhdr.sign == NTFS_CHKD_SIGNATURE; 1240 1241 if (!bchk && !brst) { 1242 if (r_page->rhdr.sign != NTFS_FFFF_SIGNATURE) { 1243 /* 1244 * Remember if the signature does not 1245 * indicate uninitialized file. 1246 */ 1247 info->initialized = true; 1248 } 1249 continue; 1250 } 1251 1252 ra = NULL; 1253 info->valid_page = false; 1254 info->initialized = true; 1255 info->vbo = vbo; 1256 1257 /* Let's check the restart area if this is a valid page. */ 1258 if (!is_rst_page_hdr_valid(vbo, r_page)) 1259 goto check_result; 1260 ra = Add2Ptr(r_page, le16_to_cpu(r_page->ra_off)); 1261 1262 if (!is_rst_area_valid(r_page)) 1263 goto check_result; 1264 1265 /* 1266 * We have a valid restart page header and restart area. 1267 * If chkdsk was run or we have no clients then we have 1268 * no more checking to do. 1269 */ 1270 if (bchk || ra->client_idx[1] == LFS_NO_CLIENT_LE) { 1271 info->valid_page = true; 1272 goto check_result; 1273 } 1274 1275 if (is_client_area_valid(r_page, usa_error)) { 1276 info->valid_page = true; 1277 ra = Add2Ptr(r_page, le16_to_cpu(r_page->ra_off)); 1278 } 1279 1280 check_result: 1281 /* 1282 * If chkdsk was run then update the caller's 1283 * values and return. 1284 */ 1285 if (r_page->rhdr.sign == NTFS_CHKD_SIGNATURE) { 1286 info->chkdsk_was_run = true; 1287 info->last_lsn = le64_to_cpu(r_page->rhdr.lsn); 1288 info->restart = true; 1289 info->r_page = r_page; 1290 return 0; 1291 } 1292 1293 /* 1294 * If we have a valid page then copy the values 1295 * we need from it. 1296 */ 1297 if (info->valid_page) { 1298 info->last_lsn = le64_to_cpu(ra->current_lsn); 1299 info->restart = true; 1300 info->r_page = r_page; 1301 return 0; 1302 } 1303 } 1304 1305 kfree(r_page); 1306 1307 return 0; 1308 } 1309 1310 /* 1311 * Ilog_init_pg_hdr - Init @log from restart page header. 1312 */ 1313 static void log_init_pg_hdr(struct ntfs_log *log, u16 major_ver, u16 minor_ver) 1314 { 1315 log->sys_page_size = log->page_size; 1316 log->sys_page_mask = log->page_mask; 1317 1318 log->clst_per_page = log->page_size >> log->ni->mi.sbi->cluster_bits; 1319 if (!log->clst_per_page) 1320 log->clst_per_page = 1; 1321 1322 log->first_page = major_ver >= 2 ? 0x22 * log->page_size : 1323 4 * log->page_size; 1324 log->major_ver = major_ver; 1325 log->minor_ver = minor_ver; 1326 } 1327 1328 /* 1329 * log_create - Init @log in cases when we don't have a restart area to use. 1330 */ 1331 static void log_create(struct ntfs_log *log, const u64 last_lsn, 1332 u32 open_log_count, bool wrapped, bool use_multi_page) 1333 { 1334 /* All file offsets must be quadword aligned. */ 1335 log->file_data_bits = blksize_bits(log->l_size) - 3; 1336 log->seq_num_mask = (8 << log->file_data_bits) - 1; 1337 log->seq_num_bits = sizeof(u64) * 8 - log->file_data_bits; 1338 log->seq_num = (last_lsn >> log->file_data_bits) + 2; 1339 log->next_page = log->first_page; 1340 log->oldest_lsn = log->seq_num << log->file_data_bits; 1341 log->oldest_lsn_off = 0; 1342 log->last_lsn = log->oldest_lsn; 1343 1344 log->l_flags |= NTFSLOG_NO_LAST_LSN | NTFSLOG_NO_OLDEST_LSN; 1345 1346 /* Set the correct flags for the I/O and indicate if we have wrapped. */ 1347 if (wrapped) 1348 log->l_flags |= NTFSLOG_WRAPPED; 1349 1350 if (use_multi_page) 1351 log->l_flags |= NTFSLOG_MULTIPLE_PAGE_IO; 1352 1353 /* Compute the log page values. */ 1354 log->data_off = ALIGN( 1355 offsetof(struct RECORD_PAGE_HDR, fixups) + 1356 sizeof(short) * ((log->page_size >> SECTOR_SHIFT) + 1), 1357 8); 1358 log->data_size = log->page_size - log->data_off; 1359 log->record_header_len = sizeof(struct LFS_RECORD_HDR); 1360 1361 /* Remember the different page sizes for reservation. */ 1362 log->reserved = log->data_size - log->record_header_len; 1363 1364 /* Compute the restart page values. */ 1365 log->ra_off = ALIGN( 1366 offsetof(struct RESTART_HDR, fixups) + 1367 sizeof(short) * 1368 ((log->sys_page_size >> SECTOR_SHIFT) + 1), 1369 8); 1370 log->restart_size = log->sys_page_size - log->ra_off; 1371 log->ra_size = struct_size(log->ra, clients, 1); 1372 log->current_openlog_count = open_log_count; 1373 1374 /* 1375 * The total available log file space is the number of 1376 * log file pages times the space available on each page. 1377 */ 1378 log->total_avail_pages = log->l_size - log->first_page; 1379 log->total_avail = log->total_avail_pages >> log->page_bits; 1380 1381 /* 1382 * We assume that we can't use the end of the page less than 1383 * the file record size. 1384 * Then we won't need to reserve more than the caller asks for. 1385 */ 1386 log->max_current_avail = log->total_avail * log->reserved; 1387 log->total_avail = log->total_avail * log->data_size; 1388 log->current_avail = log->max_current_avail; 1389 } 1390 1391 /* 1392 * log_create_ra - Fill a restart area from the values stored in @log. 1393 */ 1394 static struct RESTART_AREA *log_create_ra(struct ntfs_log *log) 1395 { 1396 struct CLIENT_REC *cr; 1397 struct RESTART_AREA *ra = kzalloc(log->restart_size, GFP_NOFS); 1398 1399 if (!ra) 1400 return NULL; 1401 1402 ra->current_lsn = cpu_to_le64(log->last_lsn); 1403 ra->log_clients = cpu_to_le16(1); 1404 ra->client_idx[1] = LFS_NO_CLIENT_LE; 1405 if (log->l_flags & NTFSLOG_MULTIPLE_PAGE_IO) 1406 ra->flags = RESTART_SINGLE_PAGE_IO; 1407 ra->seq_num_bits = cpu_to_le32(log->seq_num_bits); 1408 ra->ra_len = cpu_to_le16(log->ra_size); 1409 ra->client_off = cpu_to_le16(offsetof(struct RESTART_AREA, clients)); 1410 ra->l_size = cpu_to_le64(log->l_size); 1411 ra->rec_hdr_len = cpu_to_le16(log->record_header_len); 1412 ra->data_off = cpu_to_le16(log->data_off); 1413 ra->open_log_count = cpu_to_le32(log->current_openlog_count + 1); 1414 1415 cr = ra->clients; 1416 1417 cr->prev_client = LFS_NO_CLIENT_LE; 1418 cr->next_client = LFS_NO_CLIENT_LE; 1419 1420 return ra; 1421 } 1422 1423 static u32 final_log_off(struct ntfs_log *log, u64 lsn, u32 data_len) 1424 { 1425 u32 base_vbo = lsn << 3; 1426 u32 final_log_off = (base_vbo & log->seq_num_mask) & ~log->page_mask; 1427 u32 page_off = base_vbo & log->page_mask; 1428 u32 tail = log->page_size - page_off; 1429 1430 page_off -= 1; 1431 1432 /* Add the length of the header. */ 1433 data_len += log->record_header_len; 1434 1435 /* 1436 * If this lsn is contained this log page we are done. 1437 * Otherwise we need to walk through several log pages. 1438 */ 1439 if (data_len > tail) { 1440 data_len -= tail; 1441 tail = log->data_size; 1442 page_off = log->data_off - 1; 1443 1444 for (;;) { 1445 final_log_off = next_page_off(log, final_log_off); 1446 1447 /* 1448 * We are done if the remaining bytes 1449 * fit on this page. 1450 */ 1451 if (data_len <= tail) 1452 break; 1453 data_len -= tail; 1454 } 1455 } 1456 1457 /* 1458 * We add the remaining bytes to our starting position on this page 1459 * and then add that value to the file offset of this log page. 1460 */ 1461 return final_log_off + data_len + page_off; 1462 } 1463 1464 static int next_log_lsn(struct ntfs_log *log, const struct LFS_RECORD_HDR *rh, 1465 u64 *lsn) 1466 { 1467 int err; 1468 u64 this_lsn = le64_to_cpu(rh->this_lsn); 1469 u32 vbo = lsn_to_vbo(log, this_lsn); 1470 u32 end = 1471 final_log_off(log, this_lsn, le32_to_cpu(rh->client_data_len)); 1472 u32 hdr_off = end & ~log->sys_page_mask; 1473 u64 seq = this_lsn >> log->file_data_bits; 1474 struct RECORD_PAGE_HDR *page = NULL; 1475 1476 /* Remember if we wrapped. */ 1477 if (end <= vbo) 1478 seq += 1; 1479 1480 /* Log page header for this page. */ 1481 err = read_log_page(log, hdr_off, &page, NULL); 1482 if (err) 1483 return err; 1484 1485 /* 1486 * If the lsn we were given was not the last lsn on this page, 1487 * then the starting offset for the next lsn is on a quad word 1488 * boundary following the last file offset for the current lsn. 1489 * Otherwise the file offset is the start of the data on the next page. 1490 */ 1491 if (this_lsn == le64_to_cpu(page->rhdr.lsn)) { 1492 /* If we wrapped, we need to increment the sequence number. */ 1493 hdr_off = next_page_off(log, hdr_off); 1494 if (hdr_off == log->first_page) 1495 seq += 1; 1496 1497 vbo = hdr_off + log->data_off; 1498 } else { 1499 vbo = ALIGN(end, 8); 1500 } 1501 1502 /* Compute the lsn based on the file offset and the sequence count. */ 1503 *lsn = vbo_to_lsn(log, vbo, seq); 1504 1505 /* 1506 * If this lsn is within the legal range for the file, we return true. 1507 * Otherwise false indicates that there are no more lsn's. 1508 */ 1509 if (!is_lsn_in_file(log, *lsn)) 1510 *lsn = 0; 1511 1512 kfree(page); 1513 1514 return 0; 1515 } 1516 1517 /* 1518 * current_log_avail - Calculate the number of bytes available for log records. 1519 */ 1520 static u32 current_log_avail(struct ntfs_log *log) 1521 { 1522 u32 oldest_off, next_free_off, free_bytes; 1523 1524 if (log->l_flags & NTFSLOG_NO_LAST_LSN) { 1525 /* The entire file is available. */ 1526 return log->max_current_avail; 1527 } 1528 1529 /* 1530 * If there is a last lsn the restart area then we know that we will 1531 * have to compute the free range. 1532 * If there is no oldest lsn then start at the first page of the file. 1533 */ 1534 oldest_off = (log->l_flags & NTFSLOG_NO_OLDEST_LSN) ? 1535 log->first_page : 1536 (log->oldest_lsn_off & ~log->sys_page_mask); 1537 1538 /* 1539 * We will use the next log page offset to compute the next free page. 1540 * If we are going to reuse this page go to the next page. 1541 * If we are at the first page then use the end of the file. 1542 */ 1543 next_free_off = (log->l_flags & NTFSLOG_REUSE_TAIL) ? 1544 log->next_page + log->page_size : 1545 log->next_page == log->first_page ? log->l_size : 1546 log->next_page; 1547 1548 /* If the two offsets are the same then there is no available space. */ 1549 if (oldest_off == next_free_off) 1550 return 0; 1551 /* 1552 * If the free offset follows the oldest offset then subtract 1553 * this range from the total available pages. 1554 */ 1555 free_bytes = 1556 oldest_off < next_free_off ? 1557 log->total_avail_pages - (next_free_off - oldest_off) : 1558 oldest_off - next_free_off; 1559 1560 free_bytes >>= log->page_bits; 1561 return free_bytes * log->reserved; 1562 } 1563 1564 static bool check_subseq_log_page(struct ntfs_log *log, 1565 const struct RECORD_PAGE_HDR *rp, u32 vbo, 1566 u64 seq) 1567 { 1568 u64 lsn_seq; 1569 const struct NTFS_RECORD_HEADER *rhdr = &rp->rhdr; 1570 u64 lsn = le64_to_cpu(rhdr->lsn); 1571 1572 if (rhdr->sign == NTFS_FFFF_SIGNATURE || !rhdr->sign) 1573 return false; 1574 1575 /* 1576 * If the last lsn on the page occurs was written after the page 1577 * that caused the original error then we have a fatal error. 1578 */ 1579 lsn_seq = lsn >> log->file_data_bits; 1580 1581 /* 1582 * If the sequence number for the lsn the page is equal or greater 1583 * than lsn we expect, then this is a subsequent write. 1584 */ 1585 return lsn_seq >= seq || 1586 (lsn_seq == seq - 1 && log->first_page == vbo && 1587 vbo != (lsn_to_vbo(log, lsn) & ~log->page_mask)); 1588 } 1589 1590 /* 1591 * last_log_lsn 1592 * 1593 * Walks through the log pages for a file, searching for the 1594 * last log page written to the file. 1595 */ 1596 static int last_log_lsn(struct ntfs_log *log) 1597 { 1598 int err; 1599 bool usa_error = false; 1600 bool replace_page = false; 1601 bool reuse_page = log->l_flags & NTFSLOG_REUSE_TAIL; 1602 bool wrapped_file, wrapped; 1603 1604 u32 page_cnt = 1, page_pos = 1; 1605 u32 page_off = 0, page_off1 = 0, saved_off = 0; 1606 u32 final_off, second_off, final_off_prev = 0, second_off_prev = 0; 1607 u32 first_file_off = 0, second_file_off = 0; 1608 u32 part_io_count = 0; 1609 u32 tails = 0; 1610 u32 this_off, curpage_off, nextpage_off, remain_pages; 1611 1612 u64 expected_seq, seq_base = 0, lsn_base = 0; 1613 u64 best_lsn, best_lsn1, best_lsn2; 1614 u64 lsn_cur, lsn1, lsn2; 1615 u64 last_ok_lsn = reuse_page ? log->last_lsn : 0; 1616 1617 u16 cur_pos, best_page_pos; 1618 1619 struct RECORD_PAGE_HDR *page = NULL; 1620 struct RECORD_PAGE_HDR *tst_page = NULL; 1621 struct RECORD_PAGE_HDR *first_tail = NULL; 1622 struct RECORD_PAGE_HDR *second_tail = NULL; 1623 struct RECORD_PAGE_HDR *tail_page = NULL; 1624 struct RECORD_PAGE_HDR *second_tail_prev = NULL; 1625 struct RECORD_PAGE_HDR *first_tail_prev = NULL; 1626 struct RECORD_PAGE_HDR *page_bufs = NULL; 1627 struct RECORD_PAGE_HDR *best_page; 1628 1629 if (log->major_ver >= 2) { 1630 final_off = 0x02 * log->page_size; 1631 second_off = 0x12 * log->page_size; 1632 1633 // 0x10 == 0x12 - 0x2 1634 page_bufs = kmalloc(log->page_size * 0x10, GFP_NOFS); 1635 if (!page_bufs) 1636 return -ENOMEM; 1637 } else { 1638 second_off = log->first_page - log->page_size; 1639 final_off = second_off - log->page_size; 1640 } 1641 1642 next_tail: 1643 /* Read second tail page (at pos 3/0x12000). */ 1644 if (read_log_page(log, second_off, &second_tail, &usa_error) || 1645 usa_error || second_tail->rhdr.sign != NTFS_RCRD_SIGNATURE) { 1646 kfree(second_tail); 1647 second_tail = NULL; 1648 second_file_off = 0; 1649 lsn2 = 0; 1650 } else { 1651 second_file_off = hdr_file_off(log, second_tail); 1652 lsn2 = le64_to_cpu(second_tail->record_hdr.last_end_lsn); 1653 } 1654 1655 /* Read first tail page (at pos 2/0x2000). */ 1656 if (read_log_page(log, final_off, &first_tail, &usa_error) || 1657 usa_error || first_tail->rhdr.sign != NTFS_RCRD_SIGNATURE) { 1658 kfree(first_tail); 1659 first_tail = NULL; 1660 first_file_off = 0; 1661 lsn1 = 0; 1662 } else { 1663 first_file_off = hdr_file_off(log, first_tail); 1664 lsn1 = le64_to_cpu(first_tail->record_hdr.last_end_lsn); 1665 } 1666 1667 if (log->major_ver < 2) { 1668 int best_page; 1669 1670 first_tail_prev = first_tail; 1671 final_off_prev = first_file_off; 1672 second_tail_prev = second_tail; 1673 second_off_prev = second_file_off; 1674 tails = 1; 1675 1676 if (!first_tail && !second_tail) 1677 goto tail_read; 1678 1679 if (first_tail && second_tail) 1680 best_page = lsn1 < lsn2 ? 1 : 0; 1681 else if (first_tail) 1682 best_page = 0; 1683 else 1684 best_page = 1; 1685 1686 page_off = best_page ? second_file_off : first_file_off; 1687 seq_base = (best_page ? lsn2 : lsn1) >> log->file_data_bits; 1688 goto tail_read; 1689 } 1690 1691 best_lsn1 = first_tail ? base_lsn(log, first_tail, first_file_off) : 0; 1692 best_lsn2 = second_tail ? base_lsn(log, second_tail, second_file_off) : 1693 0; 1694 1695 if (first_tail && second_tail) { 1696 if (best_lsn1 > best_lsn2) { 1697 best_lsn = best_lsn1; 1698 best_page = first_tail; 1699 this_off = first_file_off; 1700 } else { 1701 best_lsn = best_lsn2; 1702 best_page = second_tail; 1703 this_off = second_file_off; 1704 } 1705 } else if (first_tail) { 1706 best_lsn = best_lsn1; 1707 best_page = first_tail; 1708 this_off = first_file_off; 1709 } else if (second_tail) { 1710 best_lsn = best_lsn2; 1711 best_page = second_tail; 1712 this_off = second_file_off; 1713 } else { 1714 goto tail_read; 1715 } 1716 1717 best_page_pos = le16_to_cpu(best_page->page_pos); 1718 1719 if (!tails) { 1720 if (best_page_pos == page_pos) { 1721 seq_base = best_lsn >> log->file_data_bits; 1722 saved_off = page_off = le32_to_cpu(best_page->file_off); 1723 lsn_base = best_lsn; 1724 1725 memmove(page_bufs, best_page, log->page_size); 1726 1727 page_cnt = le16_to_cpu(best_page->page_count); 1728 if (page_cnt > 1) 1729 page_pos += 1; 1730 1731 tails = 1; 1732 } 1733 } else if (seq_base == (best_lsn >> log->file_data_bits) && 1734 saved_off + log->page_size == this_off && 1735 lsn_base < best_lsn && 1736 (page_pos != page_cnt || best_page_pos == page_pos || 1737 best_page_pos == 1) && 1738 (page_pos >= page_cnt || best_page_pos == page_pos)) { 1739 u16 bppc = le16_to_cpu(best_page->page_count); 1740 1741 saved_off += log->page_size; 1742 lsn_base = best_lsn; 1743 1744 memmove(Add2Ptr(page_bufs, tails * log->page_size), best_page, 1745 log->page_size); 1746 1747 tails += 1; 1748 1749 if (best_page_pos != bppc) { 1750 page_cnt = bppc; 1751 page_pos = best_page_pos; 1752 1753 if (page_cnt > 1) 1754 page_pos += 1; 1755 } else { 1756 page_pos = page_cnt = 1; 1757 } 1758 } else { 1759 kfree(first_tail); 1760 kfree(second_tail); 1761 goto tail_read; 1762 } 1763 1764 kfree(first_tail_prev); 1765 first_tail_prev = first_tail; 1766 final_off_prev = first_file_off; 1767 first_tail = NULL; 1768 1769 kfree(second_tail_prev); 1770 second_tail_prev = second_tail; 1771 second_off_prev = second_file_off; 1772 second_tail = NULL; 1773 1774 final_off += log->page_size; 1775 second_off += log->page_size; 1776 1777 if (tails < 0x10) 1778 goto next_tail; 1779 tail_read: 1780 first_tail = first_tail_prev; 1781 final_off = final_off_prev; 1782 1783 second_tail = second_tail_prev; 1784 second_off = second_off_prev; 1785 1786 page_cnt = page_pos = 1; 1787 1788 curpage_off = seq_base == log->seq_num ? min(log->next_page, page_off) : 1789 log->next_page; 1790 1791 wrapped_file = 1792 curpage_off == log->first_page && 1793 !(log->l_flags & (NTFSLOG_NO_LAST_LSN | NTFSLOG_REUSE_TAIL)); 1794 1795 expected_seq = wrapped_file ? (log->seq_num + 1) : log->seq_num; 1796 1797 nextpage_off = curpage_off; 1798 1799 next_page: 1800 tail_page = NULL; 1801 /* Read the next log page. */ 1802 err = read_log_page(log, curpage_off, &page, &usa_error); 1803 1804 /* Compute the next log page offset the file. */ 1805 nextpage_off = next_page_off(log, curpage_off); 1806 wrapped = nextpage_off == log->first_page; 1807 1808 if (tails > 1) { 1809 struct RECORD_PAGE_HDR *cur_page = 1810 Add2Ptr(page_bufs, curpage_off - page_off); 1811 1812 if (curpage_off == saved_off) { 1813 tail_page = cur_page; 1814 goto use_tail_page; 1815 } 1816 1817 if (page_off > curpage_off || curpage_off >= saved_off) 1818 goto use_tail_page; 1819 1820 if (page_off1) 1821 goto use_cur_page; 1822 1823 if (!err && !usa_error && 1824 page->rhdr.sign == NTFS_RCRD_SIGNATURE && 1825 cur_page->rhdr.lsn == page->rhdr.lsn && 1826 cur_page->record_hdr.next_record_off == 1827 page->record_hdr.next_record_off && 1828 ((page_pos == page_cnt && 1829 le16_to_cpu(page->page_pos) == 1) || 1830 (page_pos != page_cnt && 1831 le16_to_cpu(page->page_pos) == page_pos + 1 && 1832 le16_to_cpu(page->page_count) == page_cnt))) { 1833 cur_page = NULL; 1834 goto use_tail_page; 1835 } 1836 1837 page_off1 = page_off; 1838 1839 use_cur_page: 1840 1841 lsn_cur = le64_to_cpu(cur_page->rhdr.lsn); 1842 1843 if (last_ok_lsn != 1844 le64_to_cpu(cur_page->record_hdr.last_end_lsn) && 1845 ((lsn_cur >> log->file_data_bits) + 1846 ((curpage_off < 1847 (lsn_to_vbo(log, lsn_cur) & ~log->page_mask)) ? 1848 1 : 1849 0)) != expected_seq) { 1850 goto check_tail; 1851 } 1852 1853 if (!is_log_record_end(cur_page)) { 1854 tail_page = NULL; 1855 last_ok_lsn = lsn_cur; 1856 goto next_page_1; 1857 } 1858 1859 log->seq_num = expected_seq; 1860 log->l_flags &= ~NTFSLOG_NO_LAST_LSN; 1861 log->last_lsn = le64_to_cpu(cur_page->record_hdr.last_end_lsn); 1862 log->ra->current_lsn = cur_page->record_hdr.last_end_lsn; 1863 1864 if (log->record_header_len <= 1865 log->page_size - 1866 le16_to_cpu(cur_page->record_hdr.next_record_off)) { 1867 log->l_flags |= NTFSLOG_REUSE_TAIL; 1868 log->next_page = curpage_off; 1869 } else { 1870 log->l_flags &= ~NTFSLOG_REUSE_TAIL; 1871 log->next_page = nextpage_off; 1872 } 1873 1874 if (wrapped_file) 1875 log->l_flags |= NTFSLOG_WRAPPED; 1876 1877 last_ok_lsn = le64_to_cpu(cur_page->record_hdr.last_end_lsn); 1878 goto next_page_1; 1879 } 1880 1881 /* 1882 * If we are at the expected first page of a transfer check to see 1883 * if either tail copy is at this offset. 1884 * If this page is the last page of a transfer, check if we wrote 1885 * a subsequent tail copy. 1886 */ 1887 if (page_cnt == page_pos || page_cnt == page_pos + 1) { 1888 /* 1889 * Check if the offset matches either the first or second 1890 * tail copy. It is possible it will match both. 1891 */ 1892 if (curpage_off == final_off) 1893 tail_page = first_tail; 1894 1895 /* 1896 * If we already matched on the first page then 1897 * check the ending lsn's. 1898 */ 1899 if (curpage_off == second_off) { 1900 if (!tail_page || 1901 (second_tail && 1902 le64_to_cpu(second_tail->record_hdr.last_end_lsn) > 1903 le64_to_cpu(first_tail->record_hdr 1904 .last_end_lsn))) { 1905 tail_page = second_tail; 1906 } 1907 } 1908 } 1909 1910 use_tail_page: 1911 if (tail_page) { 1912 /* We have a candidate for a tail copy. */ 1913 lsn_cur = le64_to_cpu(tail_page->record_hdr.last_end_lsn); 1914 1915 if (last_ok_lsn < lsn_cur) { 1916 /* 1917 * If the sequence number is not expected, 1918 * then don't use the tail copy. 1919 */ 1920 if (expected_seq != (lsn_cur >> log->file_data_bits)) 1921 tail_page = NULL; 1922 } else if (last_ok_lsn > lsn_cur) { 1923 /* 1924 * If the last lsn is greater than the one on 1925 * this page then forget this tail. 1926 */ 1927 tail_page = NULL; 1928 } 1929 } 1930 1931 /* 1932 *If we have an error on the current page, 1933 * we will break of this loop. 1934 */ 1935 if (err || usa_error) 1936 goto check_tail; 1937 1938 /* 1939 * Done if the last lsn on this page doesn't match the previous known 1940 * last lsn or the sequence number is not expected. 1941 */ 1942 lsn_cur = le64_to_cpu(page->rhdr.lsn); 1943 if (last_ok_lsn != lsn_cur && 1944 expected_seq != (lsn_cur >> log->file_data_bits)) { 1945 goto check_tail; 1946 } 1947 1948 /* 1949 * Check that the page position and page count values are correct. 1950 * If this is the first page of a transfer the position must be 1 1951 * and the count will be unknown. 1952 */ 1953 if (page_cnt == page_pos) { 1954 if (page->page_pos != cpu_to_le16(1) && 1955 (!reuse_page || page->page_pos != page->page_count)) { 1956 /* 1957 * If the current page is the first page we are 1958 * looking at and we are reusing this page then 1959 * it can be either the first or last page of a 1960 * transfer. Otherwise it can only be the first. 1961 */ 1962 goto check_tail; 1963 } 1964 } else if (le16_to_cpu(page->page_count) != page_cnt || 1965 le16_to_cpu(page->page_pos) != page_pos + 1) { 1966 /* 1967 * The page position better be 1 more than the last page 1968 * position and the page count better match. 1969 */ 1970 goto check_tail; 1971 } 1972 1973 /* 1974 * We have a valid page the file and may have a valid page 1975 * the tail copy area. 1976 * If the tail page was written after the page the file then 1977 * break of the loop. 1978 */ 1979 if (tail_page && 1980 le64_to_cpu(tail_page->record_hdr.last_end_lsn) > lsn_cur) { 1981 /* Remember if we will replace the page. */ 1982 replace_page = true; 1983 goto check_tail; 1984 } 1985 1986 tail_page = NULL; 1987 1988 if (is_log_record_end(page)) { 1989 /* 1990 * Since we have read this page we know the sequence number 1991 * is the same as our expected value. 1992 */ 1993 log->seq_num = expected_seq; 1994 log->last_lsn = le64_to_cpu(page->record_hdr.last_end_lsn); 1995 log->ra->current_lsn = page->record_hdr.last_end_lsn; 1996 log->l_flags &= ~NTFSLOG_NO_LAST_LSN; 1997 1998 /* 1999 * If there is room on this page for another header then 2000 * remember we want to reuse the page. 2001 */ 2002 if (log->record_header_len <= 2003 log->page_size - 2004 le16_to_cpu(page->record_hdr.next_record_off)) { 2005 log->l_flags |= NTFSLOG_REUSE_TAIL; 2006 log->next_page = curpage_off; 2007 } else { 2008 log->l_flags &= ~NTFSLOG_REUSE_TAIL; 2009 log->next_page = nextpage_off; 2010 } 2011 2012 /* Remember if we wrapped the log file. */ 2013 if (wrapped_file) 2014 log->l_flags |= NTFSLOG_WRAPPED; 2015 } 2016 2017 /* 2018 * Remember the last page count and position. 2019 * Also remember the last known lsn. 2020 */ 2021 page_cnt = le16_to_cpu(page->page_count); 2022 page_pos = le16_to_cpu(page->page_pos); 2023 last_ok_lsn = le64_to_cpu(page->rhdr.lsn); 2024 2025 next_page_1: 2026 2027 if (wrapped) { 2028 expected_seq += 1; 2029 wrapped_file = 1; 2030 } 2031 2032 curpage_off = nextpage_off; 2033 kfree(page); 2034 page = NULL; 2035 reuse_page = 0; 2036 goto next_page; 2037 2038 check_tail: 2039 if (tail_page) { 2040 log->seq_num = expected_seq; 2041 log->last_lsn = le64_to_cpu(tail_page->record_hdr.last_end_lsn); 2042 log->ra->current_lsn = tail_page->record_hdr.last_end_lsn; 2043 log->l_flags &= ~NTFSLOG_NO_LAST_LSN; 2044 2045 if (log->page_size - 2046 le16_to_cpu( 2047 tail_page->record_hdr.next_record_off) >= 2048 log->record_header_len) { 2049 log->l_flags |= NTFSLOG_REUSE_TAIL; 2050 log->next_page = curpage_off; 2051 } else { 2052 log->l_flags &= ~NTFSLOG_REUSE_TAIL; 2053 log->next_page = nextpage_off; 2054 } 2055 2056 if (wrapped) 2057 log->l_flags |= NTFSLOG_WRAPPED; 2058 } 2059 2060 /* Remember that the partial IO will start at the next page. */ 2061 second_off = nextpage_off; 2062 2063 /* 2064 * If the next page is the first page of the file then update 2065 * the sequence number for log records which begon the next page. 2066 */ 2067 if (wrapped) 2068 expected_seq += 1; 2069 2070 /* 2071 * If we have a tail copy or are performing single page I/O we can 2072 * immediately look at the next page. 2073 */ 2074 if (replace_page || (log->ra->flags & RESTART_SINGLE_PAGE_IO)) { 2075 page_cnt = 2; 2076 page_pos = 1; 2077 goto check_valid; 2078 } 2079 2080 if (page_pos != page_cnt) 2081 goto check_valid; 2082 /* 2083 * If the next page causes us to wrap to the beginning of the log 2084 * file then we know which page to check next. 2085 */ 2086 if (wrapped) { 2087 page_cnt = 2; 2088 page_pos = 1; 2089 goto check_valid; 2090 } 2091 2092 cur_pos = 2; 2093 2094 next_test_page: 2095 kfree(tst_page); 2096 tst_page = NULL; 2097 2098 /* Walk through the file, reading log pages. */ 2099 err = read_log_page(log, nextpage_off, &tst_page, &usa_error); 2100 2101 /* 2102 * If we get a USA error then assume that we correctly found 2103 * the end of the original transfer. 2104 */ 2105 if (usa_error) 2106 goto file_is_valid; 2107 2108 /* 2109 * If we were able to read the page, we examine it to see if it 2110 * is the same or different Io block. 2111 */ 2112 if (err) 2113 goto next_test_page_1; 2114 2115 if (le16_to_cpu(tst_page->page_pos) == cur_pos && 2116 check_subseq_log_page(log, tst_page, nextpage_off, expected_seq)) { 2117 page_cnt = le16_to_cpu(tst_page->page_count) + 1; 2118 page_pos = le16_to_cpu(tst_page->page_pos); 2119 goto check_valid; 2120 } else { 2121 goto file_is_valid; 2122 } 2123 2124 next_test_page_1: 2125 2126 nextpage_off = next_page_off(log, curpage_off); 2127 wrapped = nextpage_off == log->first_page; 2128 2129 if (wrapped) { 2130 expected_seq += 1; 2131 page_cnt = 2; 2132 page_pos = 1; 2133 } 2134 2135 cur_pos += 1; 2136 part_io_count += 1; 2137 if (!wrapped) 2138 goto next_test_page; 2139 2140 check_valid: 2141 /* Skip over the remaining pages this transfer. */ 2142 remain_pages = page_cnt - page_pos - 1; 2143 part_io_count += remain_pages; 2144 2145 while (remain_pages--) { 2146 nextpage_off = next_page_off(log, curpage_off); 2147 wrapped = nextpage_off == log->first_page; 2148 2149 if (wrapped) 2150 expected_seq += 1; 2151 } 2152 2153 /* Call our routine to check this log page. */ 2154 kfree(tst_page); 2155 tst_page = NULL; 2156 2157 err = read_log_page(log, nextpage_off, &tst_page, &usa_error); 2158 if (!err && !usa_error && 2159 check_subseq_log_page(log, tst_page, nextpage_off, expected_seq)) { 2160 err = -EINVAL; 2161 goto out; 2162 } 2163 2164 file_is_valid: 2165 2166 /* We have a valid file. */ 2167 if (page_off1 || tail_page) { 2168 struct RECORD_PAGE_HDR *tmp_page; 2169 2170 if (sb_rdonly(log->ni->mi.sbi->sb)) { 2171 err = -EROFS; 2172 goto out; 2173 } 2174 2175 if (page_off1) { 2176 tmp_page = Add2Ptr(page_bufs, page_off1 - page_off); 2177 tails -= (page_off1 - page_off) / log->page_size; 2178 if (!tail_page) 2179 tails -= 1; 2180 } else { 2181 tmp_page = tail_page; 2182 tails = 1; 2183 } 2184 2185 while (tails--) { 2186 u64 off = hdr_file_off(log, tmp_page); 2187 2188 if (!page) { 2189 page = kmalloc(log->page_size, GFP_NOFS); 2190 if (!page) { 2191 err = -ENOMEM; 2192 goto out; 2193 } 2194 } 2195 2196 /* 2197 * Correct page and copy the data from this page 2198 * into it and flush it to disk. 2199 */ 2200 memcpy(page, tmp_page, log->page_size); 2201 2202 /* Fill last flushed lsn value flush the page. */ 2203 if (log->major_ver < 2) 2204 page->rhdr.lsn = page->record_hdr.last_end_lsn; 2205 else 2206 page->file_off = 0; 2207 2208 page->page_pos = page->page_count = cpu_to_le16(1); 2209 2210 ntfs_fix_pre_write(&page->rhdr, log->page_size); 2211 2212 err = ntfs_sb_write_run(log->ni->mi.sbi, 2213 &log->ni->file.run, off, page, 2214 log->page_size, 0); 2215 2216 if (err) 2217 goto out; 2218 2219 if (part_io_count && second_off == off) { 2220 second_off += log->page_size; 2221 part_io_count -= 1; 2222 } 2223 2224 tmp_page = Add2Ptr(tmp_page, log->page_size); 2225 } 2226 } 2227 2228 if (part_io_count) { 2229 if (sb_rdonly(log->ni->mi.sbi->sb)) { 2230 err = -EROFS; 2231 goto out; 2232 } 2233 } 2234 2235 out: 2236 kfree(second_tail); 2237 kfree(first_tail); 2238 kfree(page); 2239 kfree(tst_page); 2240 kfree(page_bufs); 2241 2242 return err; 2243 } 2244 2245 /* 2246 * read_log_rec_buf - Copy a log record from the file to a buffer. 2247 * 2248 * The log record may span several log pages and may even wrap the file. 2249 */ 2250 static int read_log_rec_buf(struct ntfs_log *log, 2251 const struct LFS_RECORD_HDR *rh, void *buffer) 2252 { 2253 int err; 2254 struct RECORD_PAGE_HDR *ph = NULL; 2255 u64 lsn = le64_to_cpu(rh->this_lsn); 2256 u32 vbo = lsn_to_vbo(log, lsn) & ~log->page_mask; 2257 u32 off = lsn_to_page_off(log, lsn) + log->record_header_len; 2258 u32 data_len = le32_to_cpu(rh->client_data_len); 2259 2260 /* 2261 * While there are more bytes to transfer, 2262 * we continue to attempt to perform the read. 2263 */ 2264 for (;;) { 2265 bool usa_error; 2266 u32 tail = log->page_size - off; 2267 2268 if (tail >= data_len) 2269 tail = data_len; 2270 2271 data_len -= tail; 2272 2273 err = read_log_page(log, vbo, &ph, &usa_error); 2274 if (err) 2275 goto out; 2276 2277 /* 2278 * The last lsn on this page better be greater or equal 2279 * to the lsn we are copying. 2280 */ 2281 if (lsn > le64_to_cpu(ph->rhdr.lsn)) { 2282 err = -EINVAL; 2283 goto out; 2284 } 2285 2286 memcpy(buffer, Add2Ptr(ph, off), tail); 2287 2288 /* If there are no more bytes to transfer, we exit the loop. */ 2289 if (!data_len) { 2290 if (!is_log_record_end(ph) || 2291 lsn > le64_to_cpu(ph->record_hdr.last_end_lsn)) { 2292 err = -EINVAL; 2293 goto out; 2294 } 2295 break; 2296 } 2297 2298 if (ph->rhdr.lsn == ph->record_hdr.last_end_lsn || 2299 lsn > le64_to_cpu(ph->rhdr.lsn)) { 2300 err = -EINVAL; 2301 goto out; 2302 } 2303 2304 vbo = next_page_off(log, vbo); 2305 off = log->data_off; 2306 2307 /* 2308 * Adjust our pointer the user's buffer to transfer 2309 * the next block to. 2310 */ 2311 buffer = Add2Ptr(buffer, tail); 2312 } 2313 2314 out: 2315 kfree(ph); 2316 return err; 2317 } 2318 2319 static int read_rst_area(struct ntfs_log *log, struct NTFS_RESTART **rst_, 2320 u64 *lsn) 2321 { 2322 int err; 2323 struct LFS_RECORD_HDR *rh = NULL; 2324 const struct CLIENT_REC *cr = 2325 Add2Ptr(log->ra, le16_to_cpu(log->ra->client_off)); 2326 u64 lsnr, lsnc = le64_to_cpu(cr->restart_lsn); 2327 u32 len; 2328 struct NTFS_RESTART *rst; 2329 2330 *lsn = 0; 2331 *rst_ = NULL; 2332 2333 /* If the client doesn't have a restart area, go ahead and exit now. */ 2334 if (!lsnc) 2335 return 0; 2336 2337 err = read_log_page(log, lsn_to_vbo(log, lsnc), 2338 (struct RECORD_PAGE_HDR **)&rh, NULL); 2339 if (err) 2340 return err; 2341 2342 rst = NULL; 2343 lsnr = le64_to_cpu(rh->this_lsn); 2344 2345 if (lsnc != lsnr) { 2346 /* If the lsn values don't match, then the disk is corrupt. */ 2347 err = -EINVAL; 2348 goto out; 2349 } 2350 2351 *lsn = lsnr; 2352 len = le32_to_cpu(rh->client_data_len); 2353 2354 if (!len) { 2355 err = 0; 2356 goto out; 2357 } 2358 2359 if (len < sizeof(struct NTFS_RESTART)) { 2360 err = -EINVAL; 2361 goto out; 2362 } 2363 2364 rst = kmalloc(len, GFP_NOFS); 2365 if (!rst) { 2366 err = -ENOMEM; 2367 goto out; 2368 } 2369 2370 /* Copy the data into the 'rst' buffer. */ 2371 err = read_log_rec_buf(log, rh, rst); 2372 if (err) 2373 goto out; 2374 2375 *rst_ = rst; 2376 rst = NULL; 2377 2378 out: 2379 kfree(rh); 2380 kfree(rst); 2381 2382 return err; 2383 } 2384 2385 static int find_log_rec(struct ntfs_log *log, u64 lsn, struct lcb *lcb) 2386 { 2387 int err; 2388 struct LFS_RECORD_HDR *rh = lcb->lrh; 2389 u32 rec_len, len; 2390 2391 /* Read the record header for this lsn. */ 2392 if (!rh) { 2393 err = read_log_page(log, lsn_to_vbo(log, lsn), 2394 (struct RECORD_PAGE_HDR **)&rh, NULL); 2395 2396 lcb->lrh = rh; 2397 if (err) 2398 return err; 2399 } 2400 2401 /* 2402 * If the lsn the log record doesn't match the desired 2403 * lsn then the disk is corrupt. 2404 */ 2405 if (lsn != le64_to_cpu(rh->this_lsn)) 2406 return -EINVAL; 2407 2408 len = le32_to_cpu(rh->client_data_len); 2409 2410 /* 2411 * Check that the length field isn't greater than the total 2412 * available space the log file. 2413 */ 2414 rec_len = len + log->record_header_len; 2415 if (rec_len >= log->total_avail) 2416 return -EINVAL; 2417 2418 /* 2419 * If the entire log record is on this log page, 2420 * put a pointer to the log record the context block. 2421 */ 2422 if (rh->flags & LOG_RECORD_MULTI_PAGE) { 2423 void *lr = kmalloc(len, GFP_NOFS); 2424 2425 if (!lr) 2426 return -ENOMEM; 2427 2428 lcb->log_rec = lr; 2429 lcb->alloc = true; 2430 2431 /* Copy the data into the buffer returned. */ 2432 err = read_log_rec_buf(log, rh, lr); 2433 if (err) 2434 return err; 2435 } else { 2436 /* If beyond the end of the current page -> an error. */ 2437 u32 page_off = lsn_to_page_off(log, lsn); 2438 2439 if (page_off + len + log->record_header_len > log->page_size) 2440 return -EINVAL; 2441 2442 lcb->log_rec = Add2Ptr(rh, sizeof(struct LFS_RECORD_HDR)); 2443 lcb->alloc = false; 2444 } 2445 2446 return 0; 2447 } 2448 2449 /* 2450 * read_log_rec_lcb - Init the query operation. 2451 */ 2452 static int read_log_rec_lcb(struct ntfs_log *log, u64 lsn, u32 ctx_mode, 2453 struct lcb **lcb_) 2454 { 2455 int err; 2456 const struct CLIENT_REC *cr; 2457 struct lcb *lcb; 2458 2459 switch (ctx_mode) { 2460 case lcb_ctx_undo_next: 2461 case lcb_ctx_prev: 2462 case lcb_ctx_next: 2463 break; 2464 default: 2465 return -EINVAL; 2466 } 2467 2468 /* Check that the given lsn is the legal range for this client. */ 2469 cr = Add2Ptr(log->ra, le16_to_cpu(log->ra->client_off)); 2470 2471 if (!verify_client_lsn(log, cr, lsn)) 2472 return -EINVAL; 2473 2474 lcb = kzalloc(sizeof(struct lcb), GFP_NOFS); 2475 if (!lcb) 2476 return -ENOMEM; 2477 lcb->client = log->client_id; 2478 lcb->ctx_mode = ctx_mode; 2479 2480 /* Find the log record indicated by the given lsn. */ 2481 err = find_log_rec(log, lsn, lcb); 2482 if (err) 2483 goto out; 2484 2485 *lcb_ = lcb; 2486 return 0; 2487 2488 out: 2489 lcb_put(lcb); 2490 *lcb_ = NULL; 2491 return err; 2492 } 2493 2494 /* 2495 * find_client_next_lsn 2496 * 2497 * Attempt to find the next lsn to return to a client based on the context mode. 2498 */ 2499 static int find_client_next_lsn(struct ntfs_log *log, struct lcb *lcb, u64 *lsn) 2500 { 2501 int err; 2502 u64 next_lsn; 2503 struct LFS_RECORD_HDR *hdr; 2504 2505 hdr = lcb->lrh; 2506 *lsn = 0; 2507 2508 if (lcb_ctx_next != lcb->ctx_mode) 2509 goto check_undo_next; 2510 2511 /* Loop as long as another lsn can be found. */ 2512 for (;;) { 2513 u64 current_lsn; 2514 2515 err = next_log_lsn(log, hdr, ¤t_lsn); 2516 if (err) 2517 goto out; 2518 2519 if (!current_lsn) 2520 break; 2521 2522 if (hdr != lcb->lrh) 2523 kfree(hdr); 2524 2525 hdr = NULL; 2526 err = read_log_page(log, lsn_to_vbo(log, current_lsn), 2527 (struct RECORD_PAGE_HDR **)&hdr, NULL); 2528 if (err) 2529 goto out; 2530 2531 if (memcmp(&hdr->client, &lcb->client, 2532 sizeof(struct CLIENT_ID))) { 2533 /*err = -EINVAL; */ 2534 } else if (LfsClientRecord == hdr->record_type) { 2535 kfree(lcb->lrh); 2536 lcb->lrh = hdr; 2537 *lsn = current_lsn; 2538 return 0; 2539 } 2540 } 2541 2542 out: 2543 if (hdr != lcb->lrh) 2544 kfree(hdr); 2545 return err; 2546 2547 check_undo_next: 2548 if (lcb_ctx_undo_next == lcb->ctx_mode) 2549 next_lsn = le64_to_cpu(hdr->client_undo_next_lsn); 2550 else if (lcb_ctx_prev == lcb->ctx_mode) 2551 next_lsn = le64_to_cpu(hdr->client_prev_lsn); 2552 else 2553 return 0; 2554 2555 if (!next_lsn) 2556 return 0; 2557 2558 if (!verify_client_lsn( 2559 log, Add2Ptr(log->ra, le16_to_cpu(log->ra->client_off)), 2560 next_lsn)) 2561 return 0; 2562 2563 hdr = NULL; 2564 err = read_log_page(log, lsn_to_vbo(log, next_lsn), 2565 (struct RECORD_PAGE_HDR **)&hdr, NULL); 2566 if (err) 2567 return err; 2568 kfree(lcb->lrh); 2569 lcb->lrh = hdr; 2570 2571 *lsn = next_lsn; 2572 2573 return 0; 2574 } 2575 2576 static int read_next_log_rec(struct ntfs_log *log, struct lcb *lcb, u64 *lsn) 2577 { 2578 int err; 2579 2580 err = find_client_next_lsn(log, lcb, lsn); 2581 if (err) 2582 return err; 2583 2584 if (!*lsn) 2585 return 0; 2586 2587 if (lcb->alloc) 2588 kfree(lcb->log_rec); 2589 2590 lcb->log_rec = NULL; 2591 lcb->alloc = false; 2592 kfree(lcb->lrh); 2593 lcb->lrh = NULL; 2594 2595 return find_log_rec(log, *lsn, lcb); 2596 } 2597 2598 bool check_index_header(const struct INDEX_HDR *hdr, size_t bytes) 2599 { 2600 __le16 mask; 2601 u32 min_de, de_off, used, total; 2602 const struct NTFS_DE *e; 2603 2604 if (hdr_has_subnode(hdr)) { 2605 min_de = sizeof(struct NTFS_DE) + sizeof(u64); 2606 mask = NTFS_IE_HAS_SUBNODES; 2607 } else { 2608 min_de = sizeof(struct NTFS_DE); 2609 mask = 0; 2610 } 2611 2612 de_off = le32_to_cpu(hdr->de_off); 2613 used = le32_to_cpu(hdr->used); 2614 total = le32_to_cpu(hdr->total); 2615 2616 if (de_off > bytes - min_de || used > bytes || total > bytes || 2617 de_off + min_de > used || used > total) { 2618 return false; 2619 } 2620 2621 e = Add2Ptr(hdr, de_off); 2622 for (;;) { 2623 u16 esize = le16_to_cpu(e->size); 2624 struct NTFS_DE *next = Add2Ptr(e, esize); 2625 2626 if (esize < min_de || PtrOffset(hdr, next) > used || 2627 (e->flags & NTFS_IE_HAS_SUBNODES) != mask) { 2628 return false; 2629 } 2630 2631 if (de_is_last(e)) 2632 break; 2633 2634 e = next; 2635 } 2636 2637 return true; 2638 } 2639 2640 static inline bool check_index_buffer(const struct INDEX_BUFFER *ib, u32 bytes) 2641 { 2642 u16 fo; 2643 const struct NTFS_RECORD_HEADER *r = &ib->rhdr; 2644 2645 if (r->sign != NTFS_INDX_SIGNATURE) 2646 return false; 2647 2648 fo = (SECTOR_SIZE - ((bytes >> SECTOR_SHIFT) + 1) * sizeof(short)); 2649 2650 if (le16_to_cpu(r->fix_off) > fo) 2651 return false; 2652 2653 if ((le16_to_cpu(r->fix_num) - 1) * SECTOR_SIZE != bytes) 2654 return false; 2655 2656 return check_index_header(&ib->ihdr, 2657 bytes - offsetof(struct INDEX_BUFFER, ihdr)); 2658 } 2659 2660 static inline bool check_index_root(const struct ATTRIB *attr, 2661 struct ntfs_sb_info *sbi) 2662 { 2663 bool ret; 2664 const struct INDEX_ROOT *root = resident_data(attr); 2665 u8 index_bits = le32_to_cpu(root->index_block_size) >= 2666 sbi->cluster_size ? 2667 sbi->cluster_bits : 2668 SECTOR_SHIFT; 2669 u8 block_clst = root->index_block_clst; 2670 2671 if (le32_to_cpu(attr->res.data_size) < sizeof(struct INDEX_ROOT) || 2672 (root->type != ATTR_NAME && root->type != ATTR_ZERO) || 2673 (root->type == ATTR_NAME && 2674 root->rule != NTFS_COLLATION_TYPE_FILENAME) || 2675 (le32_to_cpu(root->index_block_size) != 2676 (block_clst << index_bits)) || 2677 (block_clst != 1 && block_clst != 2 && block_clst != 4 && 2678 block_clst != 8 && block_clst != 0x10 && block_clst != 0x20 && 2679 block_clst != 0x40 && block_clst != 0x80)) { 2680 return false; 2681 } 2682 2683 ret = check_index_header(&root->ihdr, 2684 le32_to_cpu(attr->res.data_size) - 2685 offsetof(struct INDEX_ROOT, ihdr)); 2686 return ret; 2687 } 2688 2689 static inline bool check_attr(const struct MFT_REC *rec, 2690 const struct ATTRIB *attr, 2691 struct ntfs_sb_info *sbi) 2692 { 2693 u32 asize = le32_to_cpu(attr->size); 2694 u32 rsize = 0; 2695 u64 dsize, svcn, evcn; 2696 u16 run_off; 2697 2698 /* Check the fixed part of the attribute record header. */ 2699 if (asize >= sbi->record_size || 2700 asize + PtrOffset(rec, attr) >= sbi->record_size || 2701 (attr->name_len && 2702 le16_to_cpu(attr->name_off) + attr->name_len * sizeof(short) > 2703 asize)) { 2704 return false; 2705 } 2706 2707 /* Check the attribute fields. */ 2708 switch (attr->non_res) { 2709 case 0: 2710 rsize = le32_to_cpu(attr->res.data_size); 2711 if (rsize >= asize || 2712 le16_to_cpu(attr->res.data_off) + rsize > asize) { 2713 return false; 2714 } 2715 break; 2716 2717 case 1: 2718 dsize = le64_to_cpu(attr->nres.data_size); 2719 svcn = le64_to_cpu(attr->nres.svcn); 2720 evcn = le64_to_cpu(attr->nres.evcn); 2721 run_off = le16_to_cpu(attr->nres.run_off); 2722 2723 if (svcn > evcn + 1 || run_off >= asize || 2724 le64_to_cpu(attr->nres.valid_size) > dsize || 2725 dsize > le64_to_cpu(attr->nres.alloc_size)) { 2726 return false; 2727 } 2728 2729 if (run_off > asize) 2730 return false; 2731 2732 if (run_unpack(NULL, sbi, 0, svcn, evcn, svcn, 2733 Add2Ptr(attr, run_off), asize - run_off) < 0) { 2734 return false; 2735 } 2736 2737 return true; 2738 2739 default: 2740 return false; 2741 } 2742 2743 switch (attr->type) { 2744 case ATTR_NAME: 2745 if (fname_full_size(Add2Ptr( 2746 attr, le16_to_cpu(attr->res.data_off))) > asize) { 2747 return false; 2748 } 2749 break; 2750 2751 case ATTR_ROOT: 2752 return check_index_root(attr, sbi); 2753 2754 case ATTR_STD: 2755 if (rsize < sizeof(struct ATTR_STD_INFO5) && 2756 rsize != sizeof(struct ATTR_STD_INFO)) { 2757 return false; 2758 } 2759 break; 2760 2761 case ATTR_LIST: 2762 case ATTR_ID: 2763 case ATTR_SECURE: 2764 case ATTR_LABEL: 2765 case ATTR_VOL_INFO: 2766 case ATTR_DATA: 2767 case ATTR_ALLOC: 2768 case ATTR_BITMAP: 2769 case ATTR_REPARSE: 2770 case ATTR_EA_INFO: 2771 case ATTR_EA: 2772 case ATTR_PROPERTYSET: 2773 case ATTR_LOGGED_UTILITY_STREAM: 2774 break; 2775 2776 default: 2777 return false; 2778 } 2779 2780 return true; 2781 } 2782 2783 static inline bool check_file_record(const struct MFT_REC *rec, 2784 const struct MFT_REC *rec2, 2785 struct ntfs_sb_info *sbi) 2786 { 2787 const struct ATTRIB *attr; 2788 u16 fo = le16_to_cpu(rec->rhdr.fix_off); 2789 u16 fn = le16_to_cpu(rec->rhdr.fix_num); 2790 u16 ao = le16_to_cpu(rec->attr_off); 2791 u32 rs = sbi->record_size; 2792 2793 /* Check the file record header for consistency. */ 2794 if (rec->rhdr.sign != NTFS_FILE_SIGNATURE || 2795 fo > (SECTOR_SIZE - ((rs >> SECTOR_SHIFT) + 1) * sizeof(short)) || 2796 (fn - 1) * SECTOR_SIZE != rs || ao < MFTRECORD_FIXUP_OFFSET_1 || 2797 ao > sbi->record_size - SIZEOF_RESIDENT || !is_rec_inuse(rec) || 2798 le32_to_cpu(rec->total) != rs) { 2799 return false; 2800 } 2801 2802 /* Loop to check all of the attributes. */ 2803 for (attr = Add2Ptr(rec, ao); attr->type != ATTR_END; 2804 attr = Add2Ptr(attr, le32_to_cpu(attr->size))) { 2805 if (check_attr(rec, attr, sbi)) 2806 continue; 2807 return false; 2808 } 2809 2810 return true; 2811 } 2812 2813 static inline int check_lsn(const struct NTFS_RECORD_HEADER *hdr, 2814 const u64 *rlsn) 2815 { 2816 u64 lsn; 2817 2818 if (!rlsn) 2819 return true; 2820 2821 lsn = le64_to_cpu(hdr->lsn); 2822 2823 if (hdr->sign == NTFS_HOLE_SIGNATURE) 2824 return false; 2825 2826 if (*rlsn > lsn) 2827 return true; 2828 2829 return false; 2830 } 2831 2832 static inline bool check_if_attr(const struct MFT_REC *rec, 2833 const struct LOG_REC_HDR *lrh) 2834 { 2835 u16 ro = le16_to_cpu(lrh->record_off); 2836 u16 o = le16_to_cpu(rec->attr_off); 2837 const struct ATTRIB *attr = Add2Ptr(rec, o); 2838 2839 while (o < ro) { 2840 u32 asize; 2841 2842 if (attr->type == ATTR_END) 2843 break; 2844 2845 asize = le32_to_cpu(attr->size); 2846 if (!asize) 2847 break; 2848 2849 o += asize; 2850 attr = Add2Ptr(attr, asize); 2851 } 2852 2853 return o == ro; 2854 } 2855 2856 static inline bool check_if_index_root(const struct MFT_REC *rec, 2857 const struct LOG_REC_HDR *lrh) 2858 { 2859 u16 ro = le16_to_cpu(lrh->record_off); 2860 u16 o = le16_to_cpu(rec->attr_off); 2861 const struct ATTRIB *attr = Add2Ptr(rec, o); 2862 2863 while (o < ro) { 2864 u32 asize; 2865 2866 if (attr->type == ATTR_END) 2867 break; 2868 2869 asize = le32_to_cpu(attr->size); 2870 if (!asize) 2871 break; 2872 2873 o += asize; 2874 attr = Add2Ptr(attr, asize); 2875 } 2876 2877 return o == ro && attr->type == ATTR_ROOT; 2878 } 2879 2880 static inline bool check_if_root_index(const struct ATTRIB *attr, 2881 const struct INDEX_HDR *hdr, 2882 const struct LOG_REC_HDR *lrh) 2883 { 2884 u16 ao = le16_to_cpu(lrh->attr_off); 2885 u32 de_off = le32_to_cpu(hdr->de_off); 2886 u32 o = PtrOffset(attr, hdr) + de_off; 2887 const struct NTFS_DE *e = Add2Ptr(hdr, de_off); 2888 u32 asize = le32_to_cpu(attr->size); 2889 2890 while (o < ao) { 2891 u16 esize; 2892 2893 if (o >= asize) 2894 break; 2895 2896 esize = le16_to_cpu(e->size); 2897 if (!esize) 2898 break; 2899 2900 o += esize; 2901 e = Add2Ptr(e, esize); 2902 } 2903 2904 return o == ao; 2905 } 2906 2907 static inline bool check_if_alloc_index(const struct INDEX_HDR *hdr, 2908 u32 attr_off) 2909 { 2910 u32 de_off = le32_to_cpu(hdr->de_off); 2911 u32 o = offsetof(struct INDEX_BUFFER, ihdr) + de_off; 2912 const struct NTFS_DE *e = Add2Ptr(hdr, de_off); 2913 u32 used = le32_to_cpu(hdr->used); 2914 2915 while (o < attr_off) { 2916 u16 esize; 2917 2918 if (de_off >= used) 2919 break; 2920 2921 esize = le16_to_cpu(e->size); 2922 if (!esize) 2923 break; 2924 2925 o += esize; 2926 de_off += esize; 2927 e = Add2Ptr(e, esize); 2928 } 2929 2930 return o == attr_off; 2931 } 2932 2933 static inline void change_attr_size(struct MFT_REC *rec, struct ATTRIB *attr, 2934 u32 nsize) 2935 { 2936 u32 asize = le32_to_cpu(attr->size); 2937 int dsize = nsize - asize; 2938 u8 *next = Add2Ptr(attr, asize); 2939 u32 used = le32_to_cpu(rec->used); 2940 2941 memmove(Add2Ptr(attr, nsize), next, used - PtrOffset(rec, next)); 2942 2943 rec->used = cpu_to_le32(used + dsize); 2944 attr->size = cpu_to_le32(nsize); 2945 } 2946 2947 struct OpenAttr { 2948 struct ATTRIB *attr; 2949 struct runs_tree *run1; 2950 struct runs_tree run0; 2951 struct ntfs_inode *ni; 2952 // CLST rno; 2953 }; 2954 2955 /* 2956 * cmp_type_and_name 2957 * 2958 * Return: 0 if 'attr' has the same type and name. 2959 */ 2960 static inline int cmp_type_and_name(const struct ATTRIB *a1, 2961 const struct ATTRIB *a2) 2962 { 2963 return a1->type != a2->type || a1->name_len != a2->name_len || 2964 (a1->name_len && memcmp(attr_name(a1), attr_name(a2), 2965 a1->name_len * sizeof(short))); 2966 } 2967 2968 static struct OpenAttr *find_loaded_attr(struct ntfs_log *log, 2969 const struct ATTRIB *attr, CLST rno) 2970 { 2971 struct OPEN_ATTR_ENRTY *oe = NULL; 2972 2973 while ((oe = enum_rstbl(log->open_attr_tbl, oe))) { 2974 struct OpenAttr *op_attr; 2975 2976 if (ino_get(&oe->ref) != rno) 2977 continue; 2978 2979 op_attr = (struct OpenAttr *)oe->ptr; 2980 if (!cmp_type_and_name(op_attr->attr, attr)) 2981 return op_attr; 2982 } 2983 return NULL; 2984 } 2985 2986 static struct ATTRIB *attr_create_nonres_log(struct ntfs_sb_info *sbi, 2987 enum ATTR_TYPE type, u64 size, 2988 const u16 *name, size_t name_len, 2989 __le16 flags) 2990 { 2991 struct ATTRIB *attr; 2992 u32 name_size = ALIGN(name_len * sizeof(short), 8); 2993 bool is_ext = flags & (ATTR_FLAG_COMPRESSED | ATTR_FLAG_SPARSED); 2994 u32 asize = name_size + 2995 (is_ext ? SIZEOF_NONRESIDENT_EX : SIZEOF_NONRESIDENT); 2996 2997 attr = kzalloc(asize, GFP_NOFS); 2998 if (!attr) 2999 return NULL; 3000 3001 attr->type = type; 3002 attr->size = cpu_to_le32(asize); 3003 attr->flags = flags; 3004 attr->non_res = 1; 3005 attr->name_len = name_len; 3006 3007 attr->nres.evcn = cpu_to_le64((u64)bytes_to_cluster(sbi, size) - 1); 3008 attr->nres.alloc_size = cpu_to_le64(ntfs_up_cluster(sbi, size)); 3009 attr->nres.data_size = cpu_to_le64(size); 3010 attr->nres.valid_size = attr->nres.data_size; 3011 if (is_ext) { 3012 attr->name_off = SIZEOF_NONRESIDENT_EX_LE; 3013 if (is_attr_compressed(attr)) 3014 attr->nres.c_unit = NTFS_LZNT_CUNIT; 3015 3016 attr->nres.run_off = 3017 cpu_to_le16(SIZEOF_NONRESIDENT_EX + name_size); 3018 memcpy(Add2Ptr(attr, SIZEOF_NONRESIDENT_EX), name, 3019 name_len * sizeof(short)); 3020 } else { 3021 attr->name_off = SIZEOF_NONRESIDENT_LE; 3022 attr->nres.run_off = 3023 cpu_to_le16(SIZEOF_NONRESIDENT + name_size); 3024 memcpy(Add2Ptr(attr, SIZEOF_NONRESIDENT), name, 3025 name_len * sizeof(short)); 3026 } 3027 3028 return attr; 3029 } 3030 3031 /* 3032 * do_action - Common routine for the Redo and Undo Passes. 3033 * @rlsn: If it is NULL then undo. 3034 */ 3035 static int do_action(struct ntfs_log *log, struct OPEN_ATTR_ENRTY *oe, 3036 const struct LOG_REC_HDR *lrh, u32 op, void *data, 3037 u32 dlen, u32 rec_len, const u64 *rlsn) 3038 { 3039 int err = 0; 3040 struct ntfs_sb_info *sbi = log->ni->mi.sbi; 3041 struct inode *inode = NULL, *inode_parent; 3042 struct mft_inode *mi = NULL, *mi2_child = NULL; 3043 CLST rno = 0, rno_base = 0; 3044 struct INDEX_BUFFER *ib = NULL; 3045 struct MFT_REC *rec = NULL; 3046 struct ATTRIB *attr = NULL, *attr2; 3047 struct INDEX_HDR *hdr; 3048 struct INDEX_ROOT *root; 3049 struct NTFS_DE *e, *e1, *e2; 3050 struct NEW_ATTRIBUTE_SIZES *new_sz; 3051 struct ATTR_FILE_NAME *fname; 3052 struct OpenAttr *oa, *oa2; 3053 u32 nsize, t32, asize, used, esize, off, bits; 3054 u16 id, id2; 3055 u32 record_size = sbi->record_size; 3056 u64 t64; 3057 u16 roff = le16_to_cpu(lrh->record_off); 3058 u16 aoff = le16_to_cpu(lrh->attr_off); 3059 u64 lco = 0; 3060 u64 cbo = (u64)le16_to_cpu(lrh->cluster_off) << SECTOR_SHIFT; 3061 u64 tvo = le64_to_cpu(lrh->target_vcn) << sbi->cluster_bits; 3062 u64 vbo = cbo + tvo; 3063 void *buffer_le = NULL; 3064 u32 bytes = 0; 3065 bool a_dirty = false; 3066 u16 data_off; 3067 3068 oa = oe->ptr; 3069 3070 /* Big switch to prepare. */ 3071 switch (op) { 3072 /* ============================================================ 3073 * Process MFT records, as described by the current log record. 3074 * ============================================================ 3075 */ 3076 case InitializeFileRecordSegment: 3077 case DeallocateFileRecordSegment: 3078 case WriteEndOfFileRecordSegment: 3079 case CreateAttribute: 3080 case DeleteAttribute: 3081 case UpdateResidentValue: 3082 case UpdateMappingPairs: 3083 case SetNewAttributeSizes: 3084 case AddIndexEntryRoot: 3085 case DeleteIndexEntryRoot: 3086 case SetIndexEntryVcnRoot: 3087 case UpdateFileNameRoot: 3088 case UpdateRecordDataRoot: 3089 case ZeroEndOfFileRecord: 3090 rno = vbo >> sbi->record_bits; 3091 inode = ilookup(sbi->sb, rno); 3092 if (inode) { 3093 mi = &ntfs_i(inode)->mi; 3094 } else { 3095 /* Read from disk. */ 3096 err = mi_get(sbi, rno, &mi); 3097 if (err && op == InitializeFileRecordSegment) { 3098 mi = kzalloc(sizeof(struct mft_inode), 3099 GFP_NOFS); 3100 if (!mi) 3101 return -ENOMEM; 3102 err = mi_format_new(mi, sbi, rno, 0, false); 3103 } 3104 if (err) 3105 return err; 3106 } 3107 rec = mi->mrec; 3108 3109 if (op == DeallocateFileRecordSegment) 3110 goto skip_load_parent; 3111 3112 if (rec->rhdr.sign == NTFS_BAAD_SIGNATURE) 3113 goto dirty_vol; 3114 if (!check_lsn(&rec->rhdr, rlsn)) 3115 goto out; 3116 if (!check_file_record(rec, NULL, sbi)) 3117 goto dirty_vol; 3118 attr = Add2Ptr(rec, roff); 3119 3120 if (is_rec_base(rec) || InitializeFileRecordSegment == op) { 3121 rno_base = rno; 3122 goto skip_load_parent; 3123 } 3124 3125 rno_base = ino_get(&rec->parent_ref); 3126 inode_parent = ntfs_iget5(sbi->sb, &rec->parent_ref, NULL); 3127 if (IS_ERR(inode_parent)) 3128 goto skip_load_parent; 3129 3130 if (is_bad_inode(inode_parent)) { 3131 iput(inode_parent); 3132 goto skip_load_parent; 3133 } 3134 3135 if (ni_load_mi_ex(ntfs_i(inode_parent), rno, &mi2_child)) { 3136 iput(inode_parent); 3137 } else { 3138 if (mi2_child->mrec != mi->mrec) 3139 memcpy(mi2_child->mrec, mi->mrec, 3140 sbi->record_size); 3141 3142 if (inode) 3143 iput(inode); 3144 else 3145 mi_put(mi); 3146 3147 inode = inode_parent; 3148 mi = mi2_child; 3149 rec = mi2_child->mrec; 3150 attr = Add2Ptr(rec, roff); 3151 } 3152 3153 skip_load_parent: 3154 inode_parent = NULL; 3155 break; 3156 3157 /* 3158 * Process attributes, as described by the current log record. 3159 */ 3160 case UpdateNonresidentValue: 3161 case AddIndexEntryAllocation: 3162 case DeleteIndexEntryAllocation: 3163 case WriteEndOfIndexBuffer: 3164 case SetIndexEntryVcnAllocation: 3165 case UpdateFileNameAllocation: 3166 case SetBitsInNonresidentBitMap: 3167 case ClearBitsInNonresidentBitMap: 3168 case UpdateRecordDataAllocation: 3169 attr = oa->attr; 3170 bytes = UpdateNonresidentValue == op ? dlen : 0; 3171 lco = (u64)le16_to_cpu(lrh->lcns_follow) << sbi->cluster_bits; 3172 3173 if (attr->type == ATTR_ALLOC) { 3174 t32 = le32_to_cpu(oe->bytes_per_index); 3175 if (bytes < t32) 3176 bytes = t32; 3177 } 3178 3179 if (!bytes) 3180 bytes = lco - cbo; 3181 3182 bytes += roff; 3183 if (attr->type == ATTR_ALLOC) 3184 bytes = (bytes + 511) & ~511; // align 3185 3186 buffer_le = kmalloc(bytes, GFP_NOFS); 3187 if (!buffer_le) 3188 return -ENOMEM; 3189 3190 err = ntfs_read_run_nb(sbi, oa->run1, vbo, buffer_le, bytes, 3191 NULL); 3192 if (err) 3193 goto out; 3194 3195 if (attr->type == ATTR_ALLOC && *(int *)buffer_le) 3196 ntfs_fix_post_read(buffer_le, bytes, false); 3197 break; 3198 3199 default: 3200 WARN_ON(1); 3201 } 3202 3203 /* Big switch to do operation. */ 3204 switch (op) { 3205 case InitializeFileRecordSegment: 3206 if (roff + dlen > record_size) 3207 goto dirty_vol; 3208 3209 memcpy(Add2Ptr(rec, roff), data, dlen); 3210 mi->dirty = true; 3211 break; 3212 3213 case DeallocateFileRecordSegment: 3214 clear_rec_inuse(rec); 3215 le16_add_cpu(&rec->seq, 1); 3216 mi->dirty = true; 3217 break; 3218 3219 case WriteEndOfFileRecordSegment: 3220 attr2 = (struct ATTRIB *)data; 3221 if (!check_if_attr(rec, lrh) || roff + dlen > record_size) 3222 goto dirty_vol; 3223 3224 memmove(attr, attr2, dlen); 3225 rec->used = cpu_to_le32(ALIGN(roff + dlen, 8)); 3226 3227 mi->dirty = true; 3228 break; 3229 3230 case CreateAttribute: 3231 attr2 = (struct ATTRIB *)data; 3232 asize = le32_to_cpu(attr2->size); 3233 used = le32_to_cpu(rec->used); 3234 3235 if (!check_if_attr(rec, lrh) || dlen < SIZEOF_RESIDENT || 3236 !IS_ALIGNED(asize, 8) || 3237 Add2Ptr(attr2, asize) > Add2Ptr(lrh, rec_len) || 3238 dlen > record_size - used) { 3239 goto dirty_vol; 3240 } 3241 3242 memmove(Add2Ptr(attr, asize), attr, used - roff); 3243 memcpy(attr, attr2, asize); 3244 3245 rec->used = cpu_to_le32(used + asize); 3246 id = le16_to_cpu(rec->next_attr_id); 3247 id2 = le16_to_cpu(attr2->id); 3248 if (id <= id2) 3249 rec->next_attr_id = cpu_to_le16(id2 + 1); 3250 if (is_attr_indexed(attr)) 3251 le16_add_cpu(&rec->hard_links, 1); 3252 3253 oa2 = find_loaded_attr(log, attr, rno_base); 3254 if (oa2) { 3255 void *p2 = kmemdup(attr, le32_to_cpu(attr->size), 3256 GFP_NOFS); 3257 if (p2) { 3258 // run_close(oa2->run1); 3259 kfree(oa2->attr); 3260 oa2->attr = p2; 3261 } 3262 } 3263 3264 mi->dirty = true; 3265 break; 3266 3267 case DeleteAttribute: 3268 asize = le32_to_cpu(attr->size); 3269 used = le32_to_cpu(rec->used); 3270 3271 if (!check_if_attr(rec, lrh)) 3272 goto dirty_vol; 3273 3274 rec->used = cpu_to_le32(used - asize); 3275 if (is_attr_indexed(attr)) 3276 le16_add_cpu(&rec->hard_links, -1); 3277 3278 memmove(attr, Add2Ptr(attr, asize), used - asize - roff); 3279 3280 mi->dirty = true; 3281 break; 3282 3283 case UpdateResidentValue: 3284 nsize = aoff + dlen; 3285 3286 if (!check_if_attr(rec, lrh)) 3287 goto dirty_vol; 3288 3289 asize = le32_to_cpu(attr->size); 3290 used = le32_to_cpu(rec->used); 3291 3292 if (lrh->redo_len == lrh->undo_len) { 3293 if (nsize > asize) 3294 goto dirty_vol; 3295 goto move_data; 3296 } 3297 3298 if (nsize > asize && nsize - asize > record_size - used) 3299 goto dirty_vol; 3300 3301 nsize = ALIGN(nsize, 8); 3302 data_off = le16_to_cpu(attr->res.data_off); 3303 3304 if (nsize < asize) { 3305 memmove(Add2Ptr(attr, aoff), data, dlen); 3306 data = NULL; // To skip below memmove(). 3307 } 3308 3309 memmove(Add2Ptr(attr, nsize), Add2Ptr(attr, asize), 3310 used - le16_to_cpu(lrh->record_off) - asize); 3311 3312 rec->used = cpu_to_le32(used + nsize - asize); 3313 attr->size = cpu_to_le32(nsize); 3314 attr->res.data_size = cpu_to_le32(aoff + dlen - data_off); 3315 3316 move_data: 3317 if (data) 3318 memmove(Add2Ptr(attr, aoff), data, dlen); 3319 3320 oa2 = find_loaded_attr(log, attr, rno_base); 3321 if (oa2) { 3322 void *p2 = kmemdup(attr, le32_to_cpu(attr->size), 3323 GFP_NOFS); 3324 if (p2) { 3325 // run_close(&oa2->run0); 3326 oa2->run1 = &oa2->run0; 3327 kfree(oa2->attr); 3328 oa2->attr = p2; 3329 } 3330 } 3331 3332 mi->dirty = true; 3333 break; 3334 3335 case UpdateMappingPairs: 3336 nsize = aoff + dlen; 3337 asize = le32_to_cpu(attr->size); 3338 used = le32_to_cpu(rec->used); 3339 3340 if (!check_if_attr(rec, lrh) || !attr->non_res || 3341 aoff < le16_to_cpu(attr->nres.run_off) || aoff > asize || 3342 (nsize > asize && nsize - asize > record_size - used)) { 3343 goto dirty_vol; 3344 } 3345 3346 nsize = ALIGN(nsize, 8); 3347 3348 memmove(Add2Ptr(attr, nsize), Add2Ptr(attr, asize), 3349 used - le16_to_cpu(lrh->record_off) - asize); 3350 rec->used = cpu_to_le32(used + nsize - asize); 3351 attr->size = cpu_to_le32(nsize); 3352 memmove(Add2Ptr(attr, aoff), data, dlen); 3353 3354 if (run_get_highest_vcn(le64_to_cpu(attr->nres.svcn), 3355 attr_run(attr), &t64)) { 3356 goto dirty_vol; 3357 } 3358 3359 attr->nres.evcn = cpu_to_le64(t64); 3360 oa2 = find_loaded_attr(log, attr, rno_base); 3361 if (oa2 && oa2->attr->non_res) 3362 oa2->attr->nres.evcn = attr->nres.evcn; 3363 3364 mi->dirty = true; 3365 break; 3366 3367 case SetNewAttributeSizes: 3368 new_sz = data; 3369 if (!check_if_attr(rec, lrh) || !attr->non_res) 3370 goto dirty_vol; 3371 3372 attr->nres.alloc_size = new_sz->alloc_size; 3373 attr->nres.data_size = new_sz->data_size; 3374 attr->nres.valid_size = new_sz->valid_size; 3375 3376 if (dlen >= sizeof(struct NEW_ATTRIBUTE_SIZES)) 3377 attr->nres.total_size = new_sz->total_size; 3378 3379 oa2 = find_loaded_attr(log, attr, rno_base); 3380 if (oa2) { 3381 void *p2 = kmemdup(attr, le32_to_cpu(attr->size), 3382 GFP_NOFS); 3383 if (p2) { 3384 kfree(oa2->attr); 3385 oa2->attr = p2; 3386 } 3387 } 3388 mi->dirty = true; 3389 break; 3390 3391 case AddIndexEntryRoot: 3392 e = (struct NTFS_DE *)data; 3393 esize = le16_to_cpu(e->size); 3394 root = resident_data(attr); 3395 hdr = &root->ihdr; 3396 used = le32_to_cpu(hdr->used); 3397 3398 if (!check_if_index_root(rec, lrh) || 3399 !check_if_root_index(attr, hdr, lrh) || 3400 Add2Ptr(data, esize) > Add2Ptr(lrh, rec_len) || 3401 esize > le32_to_cpu(rec->total) - le32_to_cpu(rec->used)) { 3402 goto dirty_vol; 3403 } 3404 3405 e1 = Add2Ptr(attr, le16_to_cpu(lrh->attr_off)); 3406 3407 change_attr_size(rec, attr, le32_to_cpu(attr->size) + esize); 3408 3409 memmove(Add2Ptr(e1, esize), e1, 3410 PtrOffset(e1, Add2Ptr(hdr, used))); 3411 memmove(e1, e, esize); 3412 3413 le32_add_cpu(&attr->res.data_size, esize); 3414 hdr->used = cpu_to_le32(used + esize); 3415 le32_add_cpu(&hdr->total, esize); 3416 3417 mi->dirty = true; 3418 break; 3419 3420 case DeleteIndexEntryRoot: 3421 root = resident_data(attr); 3422 hdr = &root->ihdr; 3423 used = le32_to_cpu(hdr->used); 3424 3425 if (!check_if_index_root(rec, lrh) || 3426 !check_if_root_index(attr, hdr, lrh)) { 3427 goto dirty_vol; 3428 } 3429 3430 e1 = Add2Ptr(attr, le16_to_cpu(lrh->attr_off)); 3431 esize = le16_to_cpu(e1->size); 3432 e2 = Add2Ptr(e1, esize); 3433 3434 memmove(e1, e2, PtrOffset(e2, Add2Ptr(hdr, used))); 3435 3436 le32_sub_cpu(&attr->res.data_size, esize); 3437 hdr->used = cpu_to_le32(used - esize); 3438 le32_sub_cpu(&hdr->total, esize); 3439 3440 change_attr_size(rec, attr, le32_to_cpu(attr->size) - esize); 3441 3442 mi->dirty = true; 3443 break; 3444 3445 case SetIndexEntryVcnRoot: 3446 root = resident_data(attr); 3447 hdr = &root->ihdr; 3448 3449 if (!check_if_index_root(rec, lrh) || 3450 !check_if_root_index(attr, hdr, lrh)) { 3451 goto dirty_vol; 3452 } 3453 3454 e = Add2Ptr(attr, le16_to_cpu(lrh->attr_off)); 3455 3456 de_set_vbn_le(e, *(__le64 *)data); 3457 mi->dirty = true; 3458 break; 3459 3460 case UpdateFileNameRoot: 3461 root = resident_data(attr); 3462 hdr = &root->ihdr; 3463 3464 if (!check_if_index_root(rec, lrh) || 3465 !check_if_root_index(attr, hdr, lrh)) { 3466 goto dirty_vol; 3467 } 3468 3469 e = Add2Ptr(attr, le16_to_cpu(lrh->attr_off)); 3470 fname = (struct ATTR_FILE_NAME *)(e + 1); 3471 memmove(&fname->dup, data, sizeof(fname->dup)); // 3472 mi->dirty = true; 3473 break; 3474 3475 case UpdateRecordDataRoot: 3476 root = resident_data(attr); 3477 hdr = &root->ihdr; 3478 3479 if (!check_if_index_root(rec, lrh) || 3480 !check_if_root_index(attr, hdr, lrh)) { 3481 goto dirty_vol; 3482 } 3483 3484 e = Add2Ptr(attr, le16_to_cpu(lrh->attr_off)); 3485 3486 memmove(Add2Ptr(e, le16_to_cpu(e->view.data_off)), data, dlen); 3487 3488 mi->dirty = true; 3489 break; 3490 3491 case ZeroEndOfFileRecord: 3492 if (roff + dlen > record_size) 3493 goto dirty_vol; 3494 3495 memset(attr, 0, dlen); 3496 mi->dirty = true; 3497 break; 3498 3499 case UpdateNonresidentValue: 3500 if (lco < cbo + roff + dlen) 3501 goto dirty_vol; 3502 3503 memcpy(Add2Ptr(buffer_le, roff), data, dlen); 3504 3505 a_dirty = true; 3506 if (attr->type == ATTR_ALLOC) 3507 ntfs_fix_pre_write(buffer_le, bytes); 3508 break; 3509 3510 case AddIndexEntryAllocation: 3511 ib = Add2Ptr(buffer_le, roff); 3512 hdr = &ib->ihdr; 3513 e = data; 3514 esize = le16_to_cpu(e->size); 3515 e1 = Add2Ptr(ib, aoff); 3516 3517 if (is_baad(&ib->rhdr)) 3518 goto dirty_vol; 3519 if (!check_lsn(&ib->rhdr, rlsn)) 3520 goto out; 3521 3522 used = le32_to_cpu(hdr->used); 3523 3524 if (!check_index_buffer(ib, bytes) || 3525 !check_if_alloc_index(hdr, aoff) || 3526 Add2Ptr(e, esize) > Add2Ptr(lrh, rec_len) || 3527 used + esize > le32_to_cpu(hdr->total)) { 3528 goto dirty_vol; 3529 } 3530 3531 memmove(Add2Ptr(e1, esize), e1, 3532 PtrOffset(e1, Add2Ptr(hdr, used))); 3533 memcpy(e1, e, esize); 3534 3535 hdr->used = cpu_to_le32(used + esize); 3536 3537 a_dirty = true; 3538 3539 ntfs_fix_pre_write(&ib->rhdr, bytes); 3540 break; 3541 3542 case DeleteIndexEntryAllocation: 3543 ib = Add2Ptr(buffer_le, roff); 3544 hdr = &ib->ihdr; 3545 e = Add2Ptr(ib, aoff); 3546 esize = le16_to_cpu(e->size); 3547 3548 if (is_baad(&ib->rhdr)) 3549 goto dirty_vol; 3550 if (!check_lsn(&ib->rhdr, rlsn)) 3551 goto out; 3552 3553 if (!check_index_buffer(ib, bytes) || 3554 !check_if_alloc_index(hdr, aoff)) { 3555 goto dirty_vol; 3556 } 3557 3558 e1 = Add2Ptr(e, esize); 3559 nsize = esize; 3560 used = le32_to_cpu(hdr->used); 3561 3562 memmove(e, e1, PtrOffset(e1, Add2Ptr(hdr, used))); 3563 3564 hdr->used = cpu_to_le32(used - nsize); 3565 3566 a_dirty = true; 3567 3568 ntfs_fix_pre_write(&ib->rhdr, bytes); 3569 break; 3570 3571 case WriteEndOfIndexBuffer: 3572 ib = Add2Ptr(buffer_le, roff); 3573 hdr = &ib->ihdr; 3574 e = Add2Ptr(ib, aoff); 3575 3576 if (is_baad(&ib->rhdr)) 3577 goto dirty_vol; 3578 if (!check_lsn(&ib->rhdr, rlsn)) 3579 goto out; 3580 if (!check_index_buffer(ib, bytes) || 3581 !check_if_alloc_index(hdr, aoff) || 3582 aoff + dlen > offsetof(struct INDEX_BUFFER, ihdr) + 3583 le32_to_cpu(hdr->total)) { 3584 goto dirty_vol; 3585 } 3586 3587 hdr->used = cpu_to_le32(dlen + PtrOffset(hdr, e)); 3588 memmove(e, data, dlen); 3589 3590 a_dirty = true; 3591 ntfs_fix_pre_write(&ib->rhdr, bytes); 3592 break; 3593 3594 case SetIndexEntryVcnAllocation: 3595 ib = Add2Ptr(buffer_le, roff); 3596 hdr = &ib->ihdr; 3597 e = Add2Ptr(ib, aoff); 3598 3599 if (is_baad(&ib->rhdr)) 3600 goto dirty_vol; 3601 3602 if (!check_lsn(&ib->rhdr, rlsn)) 3603 goto out; 3604 if (!check_index_buffer(ib, bytes) || 3605 !check_if_alloc_index(hdr, aoff)) { 3606 goto dirty_vol; 3607 } 3608 3609 de_set_vbn_le(e, *(__le64 *)data); 3610 3611 a_dirty = true; 3612 ntfs_fix_pre_write(&ib->rhdr, bytes); 3613 break; 3614 3615 case UpdateFileNameAllocation: 3616 ib = Add2Ptr(buffer_le, roff); 3617 hdr = &ib->ihdr; 3618 e = Add2Ptr(ib, aoff); 3619 3620 if (is_baad(&ib->rhdr)) 3621 goto dirty_vol; 3622 3623 if (!check_lsn(&ib->rhdr, rlsn)) 3624 goto out; 3625 if (!check_index_buffer(ib, bytes) || 3626 !check_if_alloc_index(hdr, aoff)) { 3627 goto dirty_vol; 3628 } 3629 3630 fname = (struct ATTR_FILE_NAME *)(e + 1); 3631 memmove(&fname->dup, data, sizeof(fname->dup)); 3632 3633 a_dirty = true; 3634 ntfs_fix_pre_write(&ib->rhdr, bytes); 3635 break; 3636 3637 case SetBitsInNonresidentBitMap: 3638 off = le32_to_cpu(((struct BITMAP_RANGE *)data)->bitmap_off); 3639 bits = le32_to_cpu(((struct BITMAP_RANGE *)data)->bits); 3640 3641 if (cbo + (off + 7) / 8 > lco || 3642 cbo + ((off + bits + 7) / 8) > lco) { 3643 goto dirty_vol; 3644 } 3645 3646 ntfs_bitmap_set_le(Add2Ptr(buffer_le, roff), off, bits); 3647 a_dirty = true; 3648 break; 3649 3650 case ClearBitsInNonresidentBitMap: 3651 off = le32_to_cpu(((struct BITMAP_RANGE *)data)->bitmap_off); 3652 bits = le32_to_cpu(((struct BITMAP_RANGE *)data)->bits); 3653 3654 if (cbo + (off + 7) / 8 > lco || 3655 cbo + ((off + bits + 7) / 8) > lco) { 3656 goto dirty_vol; 3657 } 3658 3659 ntfs_bitmap_clear_le(Add2Ptr(buffer_le, roff), off, bits); 3660 a_dirty = true; 3661 break; 3662 3663 case UpdateRecordDataAllocation: 3664 ib = Add2Ptr(buffer_le, roff); 3665 hdr = &ib->ihdr; 3666 e = Add2Ptr(ib, aoff); 3667 3668 if (is_baad(&ib->rhdr)) 3669 goto dirty_vol; 3670 3671 if (!check_lsn(&ib->rhdr, rlsn)) 3672 goto out; 3673 if (!check_index_buffer(ib, bytes) || 3674 !check_if_alloc_index(hdr, aoff)) { 3675 goto dirty_vol; 3676 } 3677 3678 memmove(Add2Ptr(e, le16_to_cpu(e->view.data_off)), data, dlen); 3679 3680 a_dirty = true; 3681 ntfs_fix_pre_write(&ib->rhdr, bytes); 3682 break; 3683 3684 default: 3685 WARN_ON(1); 3686 } 3687 3688 if (rlsn) { 3689 __le64 t64 = cpu_to_le64(*rlsn); 3690 3691 if (rec) 3692 rec->rhdr.lsn = t64; 3693 if (ib) 3694 ib->rhdr.lsn = t64; 3695 } 3696 3697 if (mi && mi->dirty) { 3698 err = mi_write(mi, 0); 3699 if (err) 3700 goto out; 3701 } 3702 3703 if (a_dirty) { 3704 attr = oa->attr; 3705 err = ntfs_sb_write_run(sbi, oa->run1, vbo, buffer_le, bytes, 3706 0); 3707 if (err) 3708 goto out; 3709 } 3710 3711 out: 3712 3713 if (inode) 3714 iput(inode); 3715 else if (mi != mi2_child) 3716 mi_put(mi); 3717 3718 kfree(buffer_le); 3719 3720 return err; 3721 3722 dirty_vol: 3723 log->set_dirty = true; 3724 goto out; 3725 } 3726 3727 /* 3728 * log_replay - Replays log and empties it. 3729 * 3730 * This function is called during mount operation. 3731 * It replays log and empties it. 3732 * Initialized is set false if logfile contains '-1'. 3733 */ 3734 int log_replay(struct ntfs_inode *ni, bool *initialized) 3735 { 3736 int err; 3737 struct ntfs_sb_info *sbi = ni->mi.sbi; 3738 struct ntfs_log *log; 3739 3740 u64 rec_lsn, checkpt_lsn = 0, rlsn = 0; 3741 struct ATTR_NAME_ENTRY *attr_names = NULL; 3742 u32 attr_names_bytes = 0; 3743 u32 oatbl_bytes = 0; 3744 struct RESTART_TABLE *dptbl = NULL; 3745 struct RESTART_TABLE *trtbl = NULL; 3746 const struct RESTART_TABLE *rt; 3747 struct RESTART_TABLE *oatbl = NULL; 3748 struct inode *inode; 3749 struct OpenAttr *oa; 3750 struct ntfs_inode *ni_oe; 3751 struct ATTRIB *attr = NULL; 3752 u64 size, vcn, undo_next_lsn; 3753 CLST rno, lcn, lcn0, len0, clen; 3754 void *data; 3755 struct NTFS_RESTART *rst = NULL; 3756 struct lcb *lcb = NULL; 3757 struct OPEN_ATTR_ENRTY *oe; 3758 struct ATTR_NAME_ENTRY *ane; 3759 struct TRANSACTION_ENTRY *tr; 3760 struct DIR_PAGE_ENTRY *dp; 3761 u32 i, bytes_per_attr_entry; 3762 u32 vbo, tail, off, dlen; 3763 u32 saved_len, rec_len, transact_id; 3764 bool use_second_page; 3765 struct RESTART_AREA *ra2, *ra = NULL; 3766 struct CLIENT_REC *ca, *cr; 3767 __le16 client; 3768 struct RESTART_HDR *rh; 3769 const struct LFS_RECORD_HDR *frh; 3770 const struct LOG_REC_HDR *lrh; 3771 bool is_mapped; 3772 bool is_ro = sb_rdonly(sbi->sb); 3773 u64 t64; 3774 u16 t16; 3775 u32 t32; 3776 3777 log = kzalloc(sizeof(struct ntfs_log), GFP_NOFS); 3778 if (!log) 3779 return -ENOMEM; 3780 3781 log->ni = ni; 3782 log->l_size = log->orig_file_size = ni->vfs_inode.i_size; 3783 3784 /* Get the size of page. NOTE: To replay we can use default page. */ 3785 #if PAGE_SIZE >= DefaultLogPageSize && PAGE_SIZE <= DefaultLogPageSize * 2 3786 log->page_size = norm_file_page(PAGE_SIZE, &log->l_size, true); 3787 #else 3788 log->page_size = norm_file_page(PAGE_SIZE, &log->l_size, false); 3789 #endif 3790 if (!log->page_size) { 3791 err = -EINVAL; 3792 goto out; 3793 } 3794 3795 log->one_page_buf = kmalloc(log->page_size, GFP_NOFS); 3796 if (!log->one_page_buf) { 3797 err = -ENOMEM; 3798 goto out; 3799 } 3800 3801 log->page_mask = log->page_size - 1; 3802 log->page_bits = blksize_bits(log->page_size); 3803 3804 /* Look for a restart area on the disk. */ 3805 err = log_read_rst(log, true, &log->rst_info); 3806 if (err) 3807 goto out; 3808 3809 /* remember 'initialized' */ 3810 *initialized = log->rst_info.initialized; 3811 3812 if (!log->rst_info.restart) { 3813 if (log->rst_info.initialized) { 3814 /* No restart area but the file is not initialized. */ 3815 err = -EINVAL; 3816 goto out; 3817 } 3818 3819 log_init_pg_hdr(log, 1, 1); 3820 log_create(log, 0, get_random_u32(), false, false); 3821 3822 ra = log_create_ra(log); 3823 if (!ra) { 3824 err = -ENOMEM; 3825 goto out; 3826 } 3827 log->ra = ra; 3828 log->init_ra = true; 3829 3830 goto process_log; 3831 } 3832 3833 /* 3834 * If the restart offset above wasn't zero then we won't 3835 * look for a second restart. 3836 */ 3837 if (log->rst_info.vbo) 3838 goto check_restart_area; 3839 3840 err = log_read_rst(log, false, &log->rst_info2); 3841 if (err) 3842 goto out; 3843 3844 /* Determine which restart area to use. */ 3845 if (!log->rst_info2.restart || 3846 log->rst_info2.last_lsn <= log->rst_info.last_lsn) 3847 goto use_first_page; 3848 3849 use_second_page = true; 3850 3851 if (log->rst_info.chkdsk_was_run && 3852 log->page_size != log->rst_info.vbo) { 3853 struct RECORD_PAGE_HDR *sp = NULL; 3854 bool usa_error; 3855 3856 if (!read_log_page(log, log->page_size, &sp, &usa_error) && 3857 sp->rhdr.sign == NTFS_CHKD_SIGNATURE) { 3858 use_second_page = false; 3859 } 3860 kfree(sp); 3861 } 3862 3863 if (use_second_page) { 3864 kfree(log->rst_info.r_page); 3865 memcpy(&log->rst_info, &log->rst_info2, 3866 sizeof(struct restart_info)); 3867 log->rst_info2.r_page = NULL; 3868 } 3869 3870 use_first_page: 3871 kfree(log->rst_info2.r_page); 3872 3873 check_restart_area: 3874 /* 3875 * If the restart area is at offset 0, we want 3876 * to write the second restart area first. 3877 */ 3878 log->init_ra = !!log->rst_info.vbo; 3879 3880 /* If we have a valid page then grab a pointer to the restart area. */ 3881 ra2 = log->rst_info.valid_page ? 3882 Add2Ptr(log->rst_info.r_page, 3883 le16_to_cpu(log->rst_info.r_page->ra_off)) : 3884 NULL; 3885 3886 if (log->rst_info.chkdsk_was_run || 3887 (ra2 && ra2->client_idx[1] == LFS_NO_CLIENT_LE)) { 3888 bool wrapped = false; 3889 bool use_multi_page = false; 3890 u32 open_log_count; 3891 3892 /* Do some checks based on whether we have a valid log page. */ 3893 open_log_count = log->rst_info.valid_page ? 3894 le32_to_cpu(ra2->open_log_count) : 3895 get_random_u32(); 3896 3897 log_init_pg_hdr(log, 1, 1); 3898 3899 log_create(log, log->rst_info.last_lsn, open_log_count, wrapped, 3900 use_multi_page); 3901 3902 ra = log_create_ra(log); 3903 if (!ra) { 3904 err = -ENOMEM; 3905 goto out; 3906 } 3907 log->ra = ra; 3908 3909 /* Put the restart areas and initialize 3910 * the log file as required. 3911 */ 3912 goto process_log; 3913 } 3914 3915 if (!ra2) { 3916 err = -EINVAL; 3917 goto out; 3918 } 3919 3920 /* 3921 * If the log page or the system page sizes have changed, we can't 3922 * use the log file. We must use the system page size instead of the 3923 * default size if there is not a clean shutdown. 3924 */ 3925 t32 = le32_to_cpu(log->rst_info.r_page->sys_page_size); 3926 if (log->page_size != t32) { 3927 log->l_size = log->orig_file_size; 3928 log->page_size = norm_file_page(t32, &log->l_size, 3929 t32 == DefaultLogPageSize); 3930 } 3931 3932 if (log->page_size != t32 || 3933 log->page_size != le32_to_cpu(log->rst_info.r_page->page_size)) { 3934 err = -EINVAL; 3935 goto out; 3936 } 3937 3938 log->page_mask = log->page_size - 1; 3939 log->page_bits = blksize_bits(log->page_size); 3940 3941 /* If the file size has shrunk then we won't mount it. */ 3942 if (log->l_size < le64_to_cpu(ra2->l_size)) { 3943 err = -EINVAL; 3944 goto out; 3945 } 3946 3947 log_init_pg_hdr(log, le16_to_cpu(log->rst_info.r_page->major_ver), 3948 le16_to_cpu(log->rst_info.r_page->minor_ver)); 3949 3950 log->l_size = le64_to_cpu(ra2->l_size); 3951 log->seq_num_bits = le32_to_cpu(ra2->seq_num_bits); 3952 log->file_data_bits = sizeof(u64) * 8 - log->seq_num_bits; 3953 log->seq_num_mask = (8 << log->file_data_bits) - 1; 3954 log->last_lsn = le64_to_cpu(ra2->current_lsn); 3955 log->seq_num = log->last_lsn >> log->file_data_bits; 3956 log->ra_off = le16_to_cpu(log->rst_info.r_page->ra_off); 3957 log->restart_size = log->sys_page_size - log->ra_off; 3958 log->record_header_len = le16_to_cpu(ra2->rec_hdr_len); 3959 log->ra_size = le16_to_cpu(ra2->ra_len); 3960 log->data_off = le16_to_cpu(ra2->data_off); 3961 log->data_size = log->page_size - log->data_off; 3962 log->reserved = log->data_size - log->record_header_len; 3963 3964 vbo = lsn_to_vbo(log, log->last_lsn); 3965 3966 if (vbo < log->first_page) { 3967 /* This is a pseudo lsn. */ 3968 log->l_flags |= NTFSLOG_NO_LAST_LSN; 3969 log->next_page = log->first_page; 3970 goto find_oldest; 3971 } 3972 3973 /* Find the end of this log record. */ 3974 off = final_log_off(log, log->last_lsn, 3975 le32_to_cpu(ra2->last_lsn_data_len)); 3976 3977 /* If we wrapped the file then increment the sequence number. */ 3978 if (off <= vbo) { 3979 log->seq_num += 1; 3980 log->l_flags |= NTFSLOG_WRAPPED; 3981 } 3982 3983 /* Now compute the next log page to use. */ 3984 vbo &= ~log->sys_page_mask; 3985 tail = log->page_size - (off & log->page_mask) - 1; 3986 3987 /* 3988 *If we can fit another log record on the page, 3989 * move back a page the log file. 3990 */ 3991 if (tail >= log->record_header_len) { 3992 log->l_flags |= NTFSLOG_REUSE_TAIL; 3993 log->next_page = vbo; 3994 } else { 3995 log->next_page = next_page_off(log, vbo); 3996 } 3997 3998 find_oldest: 3999 /* 4000 * Find the oldest client lsn. Use the last 4001 * flushed lsn as a starting point. 4002 */ 4003 log->oldest_lsn = log->last_lsn; 4004 oldest_client_lsn(Add2Ptr(ra2, le16_to_cpu(ra2->client_off)), 4005 ra2->client_idx[1], &log->oldest_lsn); 4006 log->oldest_lsn_off = lsn_to_vbo(log, log->oldest_lsn); 4007 4008 if (log->oldest_lsn_off < log->first_page) 4009 log->l_flags |= NTFSLOG_NO_OLDEST_LSN; 4010 4011 if (!(ra2->flags & RESTART_SINGLE_PAGE_IO)) 4012 log->l_flags |= NTFSLOG_WRAPPED | NTFSLOG_MULTIPLE_PAGE_IO; 4013 4014 log->current_openlog_count = le32_to_cpu(ra2->open_log_count); 4015 log->total_avail_pages = log->l_size - log->first_page; 4016 log->total_avail = log->total_avail_pages >> log->page_bits; 4017 log->max_current_avail = log->total_avail * log->reserved; 4018 log->total_avail = log->total_avail * log->data_size; 4019 4020 log->current_avail = current_log_avail(log); 4021 4022 ra = kzalloc(log->restart_size, GFP_NOFS); 4023 if (!ra) { 4024 err = -ENOMEM; 4025 goto out; 4026 } 4027 log->ra = ra; 4028 4029 t16 = le16_to_cpu(ra2->client_off); 4030 if (t16 == offsetof(struct RESTART_AREA, clients)) { 4031 memcpy(ra, ra2, log->ra_size); 4032 } else { 4033 memcpy(ra, ra2, offsetof(struct RESTART_AREA, clients)); 4034 memcpy(ra->clients, Add2Ptr(ra2, t16), 4035 le16_to_cpu(ra2->ra_len) - t16); 4036 4037 log->current_openlog_count = get_random_u32(); 4038 ra->open_log_count = cpu_to_le32(log->current_openlog_count); 4039 log->ra_size = offsetof(struct RESTART_AREA, clients) + 4040 sizeof(struct CLIENT_REC); 4041 ra->client_off = 4042 cpu_to_le16(offsetof(struct RESTART_AREA, clients)); 4043 ra->ra_len = cpu_to_le16(log->ra_size); 4044 } 4045 4046 le32_add_cpu(&ra->open_log_count, 1); 4047 4048 /* Now we need to walk through looking for the last lsn. */ 4049 err = last_log_lsn(log); 4050 if (err) 4051 goto out; 4052 4053 log->current_avail = current_log_avail(log); 4054 4055 /* Remember which restart area to write first. */ 4056 log->init_ra = log->rst_info.vbo; 4057 4058 process_log: 4059 /* 1.0, 1.1, 2.0 log->major_ver/minor_ver - short values. */ 4060 switch ((log->major_ver << 16) + log->minor_ver) { 4061 case 0x10000: 4062 case 0x10001: 4063 case 0x20000: 4064 break; 4065 default: 4066 ntfs_warn(sbi->sb, "\x24LogFile version %d.%d is not supported", 4067 log->major_ver, log->minor_ver); 4068 err = -EOPNOTSUPP; 4069 log->set_dirty = true; 4070 goto out; 4071 } 4072 4073 /* One client "NTFS" per logfile. */ 4074 ca = Add2Ptr(ra, le16_to_cpu(ra->client_off)); 4075 4076 for (client = ra->client_idx[1];; client = cr->next_client) { 4077 if (client == LFS_NO_CLIENT_LE) { 4078 /* Insert "NTFS" client LogFile. */ 4079 client = ra->client_idx[0]; 4080 if (client == LFS_NO_CLIENT_LE) { 4081 err = -EINVAL; 4082 goto out; 4083 } 4084 4085 t16 = le16_to_cpu(client); 4086 cr = ca + t16; 4087 4088 remove_client(ca, cr, &ra->client_idx[0]); 4089 4090 cr->restart_lsn = 0; 4091 cr->oldest_lsn = cpu_to_le64(log->oldest_lsn); 4092 cr->name_bytes = cpu_to_le32(8); 4093 cr->name[0] = cpu_to_le16('N'); 4094 cr->name[1] = cpu_to_le16('T'); 4095 cr->name[2] = cpu_to_le16('F'); 4096 cr->name[3] = cpu_to_le16('S'); 4097 4098 add_client(ca, t16, &ra->client_idx[1]); 4099 break; 4100 } 4101 4102 cr = ca + le16_to_cpu(client); 4103 4104 if (cpu_to_le32(8) == cr->name_bytes && 4105 cpu_to_le16('N') == cr->name[0] && 4106 cpu_to_le16('T') == cr->name[1] && 4107 cpu_to_le16('F') == cr->name[2] && 4108 cpu_to_le16('S') == cr->name[3]) 4109 break; 4110 } 4111 4112 /* Update the client handle with the client block information. */ 4113 log->client_id.seq_num = cr->seq_num; 4114 log->client_id.client_idx = client; 4115 4116 err = read_rst_area(log, &rst, &checkpt_lsn); 4117 if (err) 4118 goto out; 4119 4120 if (!rst) 4121 goto out; 4122 4123 bytes_per_attr_entry = !rst->major_ver ? 0x2C : 0x28; 4124 4125 if (rst->check_point_start) 4126 checkpt_lsn = le64_to_cpu(rst->check_point_start); 4127 4128 /* Allocate and Read the Transaction Table. */ 4129 if (!rst->transact_table_len) 4130 goto check_dirty_page_table; /* reduce tab pressure. */ 4131 4132 t64 = le64_to_cpu(rst->transact_table_lsn); 4133 err = read_log_rec_lcb(log, t64, lcb_ctx_prev, &lcb); 4134 if (err) 4135 goto out; 4136 4137 lrh = lcb->log_rec; 4138 frh = lcb->lrh; 4139 rec_len = le32_to_cpu(frh->client_data_len); 4140 4141 if (!check_log_rec(lrh, rec_len, le32_to_cpu(frh->transact_id), 4142 bytes_per_attr_entry)) { 4143 err = -EINVAL; 4144 goto out; 4145 } 4146 4147 t16 = le16_to_cpu(lrh->redo_off); 4148 4149 rt = Add2Ptr(lrh, t16); 4150 t32 = rec_len - t16; 4151 4152 /* Now check that this is a valid restart table. */ 4153 if (!check_rstbl(rt, t32)) { 4154 err = -EINVAL; 4155 goto out; 4156 } 4157 4158 trtbl = kmemdup(rt, t32, GFP_NOFS); 4159 if (!trtbl) { 4160 err = -ENOMEM; 4161 goto out; 4162 } 4163 4164 lcb_put(lcb); 4165 lcb = NULL; 4166 4167 check_dirty_page_table: 4168 /* The next record back should be the Dirty Pages Table. */ 4169 if (!rst->dirty_pages_len) 4170 goto check_attribute_names; /* reduce tab pressure. */ 4171 4172 t64 = le64_to_cpu(rst->dirty_pages_table_lsn); 4173 err = read_log_rec_lcb(log, t64, lcb_ctx_prev, &lcb); 4174 if (err) 4175 goto out; 4176 4177 lrh = lcb->log_rec; 4178 frh = lcb->lrh; 4179 rec_len = le32_to_cpu(frh->client_data_len); 4180 4181 if (!check_log_rec(lrh, rec_len, le32_to_cpu(frh->transact_id), 4182 bytes_per_attr_entry)) { 4183 err = -EINVAL; 4184 goto out; 4185 } 4186 4187 t16 = le16_to_cpu(lrh->redo_off); 4188 4189 rt = Add2Ptr(lrh, t16); 4190 t32 = rec_len - t16; 4191 4192 /* Now check that this is a valid restart table. */ 4193 if (!check_rstbl(rt, t32)) { 4194 err = -EINVAL; 4195 goto out; 4196 } 4197 4198 dptbl = kmemdup(rt, t32, GFP_NOFS); 4199 if (!dptbl) { 4200 err = -ENOMEM; 4201 goto out; 4202 } 4203 4204 /* Convert Ra version '0' into version '1'. */ 4205 if (rst->major_ver) 4206 goto end_conv_1; /* reduce tab pressure. */ 4207 4208 dp = NULL; 4209 while ((dp = enum_rstbl(dptbl, dp))) { 4210 struct DIR_PAGE_ENTRY_32 *dp0 = (struct DIR_PAGE_ENTRY_32 *)dp; 4211 // NOTE: Danger. Check for of boundary. 4212 memmove(&dp->vcn, &dp0->vcn_low, 4213 2 * sizeof(u64) + 4214 le32_to_cpu(dp->lcns_follow) * sizeof(u64)); 4215 } 4216 4217 end_conv_1: 4218 lcb_put(lcb); 4219 lcb = NULL; 4220 4221 /* 4222 * Go through the table and remove the duplicates, 4223 * remembering the oldest lsn values. 4224 */ 4225 if (sbi->cluster_size <= log->page_size) 4226 goto trace_dp_table; /* reduce tab pressure. */ 4227 dp = NULL; 4228 while ((dp = enum_rstbl(dptbl, dp))) { 4229 struct DIR_PAGE_ENTRY *next = dp; 4230 4231 while ((next = enum_rstbl(dptbl, next))) { 4232 if (next->target_attr == dp->target_attr && 4233 next->vcn == dp->vcn) { 4234 if (le64_to_cpu(next->oldest_lsn) < 4235 le64_to_cpu(dp->oldest_lsn)) { 4236 dp->oldest_lsn = next->oldest_lsn; 4237 } 4238 4239 free_rsttbl_idx(dptbl, PtrOffset(dptbl, next)); 4240 } 4241 } 4242 } 4243 trace_dp_table: 4244 check_attribute_names: 4245 /* The next record should be the Attribute Names. */ 4246 if (!rst->attr_names_len) 4247 goto check_attr_table; /* reduce tab pressure. */ 4248 4249 t64 = le64_to_cpu(rst->attr_names_lsn); 4250 err = read_log_rec_lcb(log, t64, lcb_ctx_prev, &lcb); 4251 if (err) 4252 goto out; 4253 4254 lrh = lcb->log_rec; 4255 frh = lcb->lrh; 4256 rec_len = le32_to_cpu(frh->client_data_len); 4257 4258 if (!check_log_rec(lrh, rec_len, le32_to_cpu(frh->transact_id), 4259 bytes_per_attr_entry)) { 4260 err = -EINVAL; 4261 goto out; 4262 } 4263 4264 t32 = lrh_length(lrh); 4265 attr_names_bytes = rec_len - t32; 4266 4267 attr_names = kmemdup(Add2Ptr(lrh, t32), attr_names_bytes, GFP_NOFS); 4268 if (!attr_names) { 4269 err = -ENOMEM; 4270 goto out; 4271 } 4272 4273 lcb_put(lcb); 4274 lcb = NULL; 4275 4276 check_attr_table: 4277 /* The next record should be the attribute Table. */ 4278 if (!rst->open_attr_len) 4279 goto check_attribute_names2; /* reduce tab pressure. */ 4280 4281 t64 = le64_to_cpu(rst->open_attr_table_lsn); 4282 err = read_log_rec_lcb(log, t64, lcb_ctx_prev, &lcb); 4283 if (err) 4284 goto out; 4285 4286 lrh = lcb->log_rec; 4287 frh = lcb->lrh; 4288 rec_len = le32_to_cpu(frh->client_data_len); 4289 4290 if (!check_log_rec(lrh, rec_len, le32_to_cpu(frh->transact_id), 4291 bytes_per_attr_entry)) { 4292 err = -EINVAL; 4293 goto out; 4294 } 4295 4296 t16 = le16_to_cpu(lrh->redo_off); 4297 4298 rt = Add2Ptr(lrh, t16); 4299 oatbl_bytes = rec_len - t16; 4300 4301 if (!check_rstbl(rt, oatbl_bytes)) { 4302 err = -EINVAL; 4303 goto out; 4304 } 4305 4306 oatbl = kmemdup(rt, oatbl_bytes, GFP_NOFS); 4307 if (!oatbl) { 4308 err = -ENOMEM; 4309 goto out; 4310 } 4311 4312 log->open_attr_tbl = oatbl; 4313 4314 /* Clear all of the Attr pointers. */ 4315 oe = NULL; 4316 while ((oe = enum_rstbl(oatbl, oe))) { 4317 if (!rst->major_ver) { 4318 struct OPEN_ATTR_ENRTY_32 oe0; 4319 4320 /* Really 'oe' points to OPEN_ATTR_ENRTY_32. */ 4321 memcpy(&oe0, oe, SIZEOF_OPENATTRIBUTEENTRY0); 4322 4323 oe->bytes_per_index = oe0.bytes_per_index; 4324 oe->type = oe0.type; 4325 oe->is_dirty_pages = oe0.is_dirty_pages; 4326 oe->name_len = 0; 4327 oe->ref = oe0.ref; 4328 oe->open_record_lsn = oe0.open_record_lsn; 4329 } 4330 4331 oe->is_attr_name = 0; 4332 oe->ptr = NULL; 4333 } 4334 4335 lcb_put(lcb); 4336 lcb = NULL; 4337 4338 check_attribute_names2: 4339 if (attr_names && oatbl) { 4340 off = 0; 4341 for (;;) { 4342 /* Check we can use attribute name entry 'ane'. */ 4343 static_assert(sizeof(*ane) == 4); 4344 if (off + sizeof(*ane) > attr_names_bytes) { 4345 /* just ignore the rest. */ 4346 break; 4347 } 4348 4349 ane = Add2Ptr(attr_names, off); 4350 t16 = le16_to_cpu(ane->off); 4351 if (!t16) { 4352 /* this is the only valid exit. */ 4353 break; 4354 } 4355 4356 /* Check we can use open attribute entry 'oe'. */ 4357 if (t16 + sizeof(*oe) > oatbl_bytes) { 4358 /* just ignore the rest. */ 4359 break; 4360 } 4361 4362 /* TODO: Clear table on exit! */ 4363 oe = Add2Ptr(oatbl, t16); 4364 t16 = le16_to_cpu(ane->name_bytes); 4365 off += t16 + sizeof(*ane); 4366 if (off > attr_names_bytes) { 4367 /* just ignore the rest. */ 4368 break; 4369 } 4370 oe->name_len = t16 / sizeof(short); 4371 oe->ptr = ane->name; 4372 oe->is_attr_name = 2; 4373 } 4374 } 4375 4376 /* 4377 * If the checkpt_lsn is zero, then this is a freshly 4378 * formatted disk and we have no work to do. 4379 */ 4380 if (!checkpt_lsn) { 4381 err = 0; 4382 goto out; 4383 } 4384 4385 if (!oatbl) { 4386 oatbl = init_rsttbl(bytes_per_attr_entry, 8); 4387 if (!oatbl) { 4388 err = -ENOMEM; 4389 goto out; 4390 } 4391 } 4392 4393 log->open_attr_tbl = oatbl; 4394 4395 /* Start the analysis pass from the Checkpoint lsn. */ 4396 rec_lsn = checkpt_lsn; 4397 4398 /* Read the first lsn. */ 4399 err = read_log_rec_lcb(log, checkpt_lsn, lcb_ctx_next, &lcb); 4400 if (err) 4401 goto out; 4402 4403 /* Loop to read all subsequent records to the end of the log file. */ 4404 next_log_record_analyze: 4405 err = read_next_log_rec(log, lcb, &rec_lsn); 4406 if (err) 4407 goto out; 4408 4409 if (!rec_lsn) 4410 goto end_log_records_enumerate; 4411 4412 frh = lcb->lrh; 4413 transact_id = le32_to_cpu(frh->transact_id); 4414 rec_len = le32_to_cpu(frh->client_data_len); 4415 lrh = lcb->log_rec; 4416 4417 if (!check_log_rec(lrh, rec_len, transact_id, bytes_per_attr_entry)) { 4418 err = -EINVAL; 4419 goto out; 4420 } 4421 4422 /* 4423 * The first lsn after the previous lsn remembered 4424 * the checkpoint is the first candidate for the rlsn. 4425 */ 4426 if (!rlsn) 4427 rlsn = rec_lsn; 4428 4429 if (LfsClientRecord != frh->record_type) 4430 goto next_log_record_analyze; 4431 4432 /* 4433 * Now update the Transaction Table for this transaction. If there 4434 * is no entry present or it is unallocated we allocate the entry. 4435 */ 4436 if (!trtbl) { 4437 trtbl = init_rsttbl(sizeof(struct TRANSACTION_ENTRY), 4438 INITIAL_NUMBER_TRANSACTIONS); 4439 if (!trtbl) { 4440 err = -ENOMEM; 4441 goto out; 4442 } 4443 } 4444 4445 tr = Add2Ptr(trtbl, transact_id); 4446 4447 if (transact_id >= bytes_per_rt(trtbl) || 4448 tr->next != RESTART_ENTRY_ALLOCATED_LE) { 4449 tr = alloc_rsttbl_from_idx(&trtbl, transact_id); 4450 if (!tr) { 4451 err = -ENOMEM; 4452 goto out; 4453 } 4454 tr->transact_state = TransactionActive; 4455 tr->first_lsn = cpu_to_le64(rec_lsn); 4456 } 4457 4458 tr->prev_lsn = tr->undo_next_lsn = cpu_to_le64(rec_lsn); 4459 4460 /* 4461 * If this is a compensation log record, then change 4462 * the undo_next_lsn to be the undo_next_lsn of this record. 4463 */ 4464 if (lrh->undo_op == cpu_to_le16(CompensationLogRecord)) 4465 tr->undo_next_lsn = frh->client_undo_next_lsn; 4466 4467 /* Dispatch to handle log record depending on type. */ 4468 switch (le16_to_cpu(lrh->redo_op)) { 4469 case InitializeFileRecordSegment: 4470 case DeallocateFileRecordSegment: 4471 case WriteEndOfFileRecordSegment: 4472 case CreateAttribute: 4473 case DeleteAttribute: 4474 case UpdateResidentValue: 4475 case UpdateNonresidentValue: 4476 case UpdateMappingPairs: 4477 case SetNewAttributeSizes: 4478 case AddIndexEntryRoot: 4479 case DeleteIndexEntryRoot: 4480 case AddIndexEntryAllocation: 4481 case DeleteIndexEntryAllocation: 4482 case WriteEndOfIndexBuffer: 4483 case SetIndexEntryVcnRoot: 4484 case SetIndexEntryVcnAllocation: 4485 case UpdateFileNameRoot: 4486 case UpdateFileNameAllocation: 4487 case SetBitsInNonresidentBitMap: 4488 case ClearBitsInNonresidentBitMap: 4489 case UpdateRecordDataRoot: 4490 case UpdateRecordDataAllocation: 4491 case ZeroEndOfFileRecord: 4492 t16 = le16_to_cpu(lrh->target_attr); 4493 t64 = le64_to_cpu(lrh->target_vcn); 4494 dp = find_dp(dptbl, t16, t64); 4495 4496 if (dp) 4497 goto copy_lcns; 4498 4499 /* 4500 * Calculate the number of clusters per page the system 4501 * which wrote the checkpoint, possibly creating the table. 4502 */ 4503 if (dptbl) { 4504 t32 = (le16_to_cpu(dptbl->size) - 4505 sizeof(struct DIR_PAGE_ENTRY)) / 4506 sizeof(u64); 4507 } else { 4508 t32 = log->clst_per_page; 4509 kfree(dptbl); 4510 dptbl = init_rsttbl(struct_size(dp, page_lcns, t32), 4511 32); 4512 if (!dptbl) { 4513 err = -ENOMEM; 4514 goto out; 4515 } 4516 } 4517 4518 dp = alloc_rsttbl_idx(&dptbl); 4519 if (!dp) { 4520 err = -ENOMEM; 4521 goto out; 4522 } 4523 dp->target_attr = cpu_to_le32(t16); 4524 dp->transfer_len = cpu_to_le32(t32 << sbi->cluster_bits); 4525 dp->lcns_follow = cpu_to_le32(t32); 4526 dp->vcn = cpu_to_le64(t64 & ~((u64)t32 - 1)); 4527 dp->oldest_lsn = cpu_to_le64(rec_lsn); 4528 4529 copy_lcns: 4530 /* 4531 * Copy the Lcns from the log record into the Dirty Page Entry. 4532 * TODO: For different page size support, must somehow make 4533 * whole routine a loop, case Lcns do not fit below. 4534 */ 4535 t16 = le16_to_cpu(lrh->lcns_follow); 4536 for (i = 0; i < t16; i++) { 4537 size_t j = (size_t)(le64_to_cpu(lrh->target_vcn) - 4538 le64_to_cpu(dp->vcn)); 4539 dp->page_lcns[j + i] = lrh->page_lcns[i]; 4540 } 4541 4542 goto next_log_record_analyze; 4543 4544 case DeleteDirtyClusters: { 4545 u32 range_count = 4546 le16_to_cpu(lrh->redo_len) / sizeof(struct LCN_RANGE); 4547 const struct LCN_RANGE *r = 4548 Add2Ptr(lrh, le16_to_cpu(lrh->redo_off)); 4549 4550 /* Loop through all of the Lcn ranges this log record. */ 4551 for (i = 0; i < range_count; i++, r++) { 4552 u64 lcn0 = le64_to_cpu(r->lcn); 4553 u64 lcn_e = lcn0 + le64_to_cpu(r->len) - 1; 4554 4555 dp = NULL; 4556 while ((dp = enum_rstbl(dptbl, dp))) { 4557 u32 j; 4558 4559 t32 = le32_to_cpu(dp->lcns_follow); 4560 for (j = 0; j < t32; j++) { 4561 t64 = le64_to_cpu(dp->page_lcns[j]); 4562 if (t64 >= lcn0 && t64 <= lcn_e) 4563 dp->page_lcns[j] = 0; 4564 } 4565 } 4566 } 4567 goto next_log_record_analyze; 4568 } 4569 4570 case OpenNonresidentAttribute: 4571 t16 = le16_to_cpu(lrh->target_attr); 4572 if (t16 >= bytes_per_rt(oatbl)) { 4573 /* 4574 * Compute how big the table needs to be. 4575 * Add 10 extra entries for some cushion. 4576 */ 4577 u32 new_e = t16 / le16_to_cpu(oatbl->size); 4578 4579 new_e += 10 - le16_to_cpu(oatbl->used); 4580 4581 oatbl = extend_rsttbl(oatbl, new_e, ~0u); 4582 log->open_attr_tbl = oatbl; 4583 if (!oatbl) { 4584 err = -ENOMEM; 4585 goto out; 4586 } 4587 } 4588 4589 /* Point to the entry being opened. */ 4590 oe = alloc_rsttbl_from_idx(&oatbl, t16); 4591 log->open_attr_tbl = oatbl; 4592 if (!oe) { 4593 err = -ENOMEM; 4594 goto out; 4595 } 4596 4597 /* Initialize this entry from the log record. */ 4598 t16 = le16_to_cpu(lrh->redo_off); 4599 if (!rst->major_ver) { 4600 /* Convert version '0' into version '1'. */ 4601 struct OPEN_ATTR_ENRTY_32 *oe0 = Add2Ptr(lrh, t16); 4602 4603 oe->bytes_per_index = oe0->bytes_per_index; 4604 oe->type = oe0->type; 4605 oe->is_dirty_pages = oe0->is_dirty_pages; 4606 oe->name_len = 0; //oe0.name_len; 4607 oe->ref = oe0->ref; 4608 oe->open_record_lsn = oe0->open_record_lsn; 4609 } else { 4610 memcpy(oe, Add2Ptr(lrh, t16), bytes_per_attr_entry); 4611 } 4612 4613 t16 = le16_to_cpu(lrh->undo_len); 4614 if (t16) { 4615 oe->ptr = kmalloc(t16, GFP_NOFS); 4616 if (!oe->ptr) { 4617 err = -ENOMEM; 4618 goto out; 4619 } 4620 oe->name_len = t16 / sizeof(short); 4621 memcpy(oe->ptr, 4622 Add2Ptr(lrh, le16_to_cpu(lrh->undo_off)), t16); 4623 oe->is_attr_name = 1; 4624 } else { 4625 oe->ptr = NULL; 4626 oe->is_attr_name = 0; 4627 } 4628 4629 goto next_log_record_analyze; 4630 4631 case HotFix: 4632 t16 = le16_to_cpu(lrh->target_attr); 4633 t64 = le64_to_cpu(lrh->target_vcn); 4634 dp = find_dp(dptbl, t16, t64); 4635 if (dp) { 4636 size_t j = le64_to_cpu(lrh->target_vcn) - 4637 le64_to_cpu(dp->vcn); 4638 if (dp->page_lcns[j]) 4639 dp->page_lcns[j] = lrh->page_lcns[0]; 4640 } 4641 goto next_log_record_analyze; 4642 4643 case EndTopLevelAction: 4644 tr = Add2Ptr(trtbl, transact_id); 4645 tr->prev_lsn = cpu_to_le64(rec_lsn); 4646 tr->undo_next_lsn = frh->client_undo_next_lsn; 4647 goto next_log_record_analyze; 4648 4649 case PrepareTransaction: 4650 tr = Add2Ptr(trtbl, transact_id); 4651 tr->transact_state = TransactionPrepared; 4652 goto next_log_record_analyze; 4653 4654 case CommitTransaction: 4655 tr = Add2Ptr(trtbl, transact_id); 4656 tr->transact_state = TransactionCommitted; 4657 goto next_log_record_analyze; 4658 4659 case ForgetTransaction: 4660 free_rsttbl_idx(trtbl, transact_id); 4661 goto next_log_record_analyze; 4662 4663 case Noop: 4664 case OpenAttributeTableDump: 4665 case AttributeNamesDump: 4666 case DirtyPageTableDump: 4667 case TransactionTableDump: 4668 /* The following cases require no action the Analysis Pass. */ 4669 goto next_log_record_analyze; 4670 4671 default: 4672 /* 4673 * All codes will be explicitly handled. 4674 * If we see a code we do not expect, then we are trouble. 4675 */ 4676 goto next_log_record_analyze; 4677 } 4678 4679 end_log_records_enumerate: 4680 lcb_put(lcb); 4681 lcb = NULL; 4682 4683 /* 4684 * Scan the Dirty Page Table and Transaction Table for 4685 * the lowest lsn, and return it as the Redo lsn. 4686 */ 4687 dp = NULL; 4688 while ((dp = enum_rstbl(dptbl, dp))) { 4689 t64 = le64_to_cpu(dp->oldest_lsn); 4690 if (t64 && t64 < rlsn) 4691 rlsn = t64; 4692 } 4693 4694 tr = NULL; 4695 while ((tr = enum_rstbl(trtbl, tr))) { 4696 t64 = le64_to_cpu(tr->first_lsn); 4697 if (t64 && t64 < rlsn) 4698 rlsn = t64; 4699 } 4700 4701 /* 4702 * Only proceed if the Dirty Page Table or Transaction 4703 * table are not empty. 4704 */ 4705 if ((!dptbl || !dptbl->total) && (!trtbl || !trtbl->total)) 4706 goto end_replay; 4707 4708 sbi->flags |= NTFS_FLAGS_NEED_REPLAY; 4709 if (is_ro) 4710 goto out; 4711 4712 /* Reopen all of the attributes with dirty pages. */ 4713 oe = NULL; 4714 next_open_attribute: 4715 4716 oe = enum_rstbl(oatbl, oe); 4717 if (!oe) { 4718 err = 0; 4719 dp = NULL; 4720 goto next_dirty_page; 4721 } 4722 4723 oa = kzalloc(sizeof(struct OpenAttr), GFP_NOFS); 4724 if (!oa) { 4725 err = -ENOMEM; 4726 goto out; 4727 } 4728 4729 inode = ntfs_iget5(sbi->sb, &oe->ref, NULL); 4730 if (IS_ERR(inode)) 4731 goto fake_attr; 4732 4733 if (is_bad_inode(inode)) { 4734 iput(inode); 4735 fake_attr: 4736 if (oa->ni) { 4737 iput(&oa->ni->vfs_inode); 4738 oa->ni = NULL; 4739 } 4740 4741 attr = attr_create_nonres_log(sbi, oe->type, 0, oe->ptr, 4742 oe->name_len, 0); 4743 if (!attr) { 4744 kfree(oa); 4745 err = -ENOMEM; 4746 goto out; 4747 } 4748 oa->attr = attr; 4749 oa->run1 = &oa->run0; 4750 goto final_oe; 4751 } 4752 4753 ni_oe = ntfs_i(inode); 4754 oa->ni = ni_oe; 4755 4756 attr = ni_find_attr(ni_oe, NULL, NULL, oe->type, oe->ptr, oe->name_len, 4757 NULL, NULL); 4758 4759 if (!attr) 4760 goto fake_attr; 4761 4762 t32 = le32_to_cpu(attr->size); 4763 oa->attr = kmemdup(attr, t32, GFP_NOFS); 4764 if (!oa->attr) 4765 goto fake_attr; 4766 4767 if (!S_ISDIR(inode->i_mode)) { 4768 if (attr->type == ATTR_DATA && !attr->name_len) { 4769 oa->run1 = &ni_oe->file.run; 4770 goto final_oe; 4771 } 4772 } else { 4773 if (attr->type == ATTR_ALLOC && 4774 attr->name_len == ARRAY_SIZE(I30_NAME) && 4775 !memcmp(attr_name(attr), I30_NAME, sizeof(I30_NAME))) { 4776 oa->run1 = &ni_oe->dir.alloc_run; 4777 goto final_oe; 4778 } 4779 } 4780 4781 if (attr->non_res) { 4782 u16 roff = le16_to_cpu(attr->nres.run_off); 4783 CLST svcn = le64_to_cpu(attr->nres.svcn); 4784 4785 if (roff > t32) { 4786 kfree(oa->attr); 4787 oa->attr = NULL; 4788 goto fake_attr; 4789 } 4790 4791 err = run_unpack(&oa->run0, sbi, inode->i_ino, svcn, 4792 le64_to_cpu(attr->nres.evcn), svcn, 4793 Add2Ptr(attr, roff), t32 - roff); 4794 if (err < 0) { 4795 kfree(oa->attr); 4796 oa->attr = NULL; 4797 goto fake_attr; 4798 } 4799 err = 0; 4800 } 4801 oa->run1 = &oa->run0; 4802 attr = oa->attr; 4803 4804 final_oe: 4805 if (oe->is_attr_name == 1) 4806 kfree(oe->ptr); 4807 oe->is_attr_name = 0; 4808 oe->ptr = oa; 4809 oe->name_len = attr->name_len; 4810 4811 goto next_open_attribute; 4812 4813 /* 4814 * Now loop through the dirty page table to extract all of the Vcn/Lcn. 4815 * Mapping that we have, and insert it into the appropriate run. 4816 */ 4817 next_dirty_page: 4818 dp = enum_rstbl(dptbl, dp); 4819 if (!dp) 4820 goto do_redo_1; 4821 4822 oe = Add2Ptr(oatbl, le32_to_cpu(dp->target_attr)); 4823 4824 if (oe->next != RESTART_ENTRY_ALLOCATED_LE) 4825 goto next_dirty_page; 4826 4827 oa = oe->ptr; 4828 if (!oa) 4829 goto next_dirty_page; 4830 4831 i = -1; 4832 next_dirty_page_vcn: 4833 i += 1; 4834 if (i >= le32_to_cpu(dp->lcns_follow)) 4835 goto next_dirty_page; 4836 4837 vcn = le64_to_cpu(dp->vcn) + i; 4838 size = (vcn + 1) << sbi->cluster_bits; 4839 4840 if (!dp->page_lcns[i]) 4841 goto next_dirty_page_vcn; 4842 4843 rno = ino_get(&oe->ref); 4844 if (rno <= MFT_REC_MIRR && 4845 size < (MFT_REC_VOL + 1) * sbi->record_size && 4846 oe->type == ATTR_DATA) { 4847 goto next_dirty_page_vcn; 4848 } 4849 4850 lcn = le64_to_cpu(dp->page_lcns[i]); 4851 4852 if ((!run_lookup_entry(oa->run1, vcn, &lcn0, &len0, NULL) || 4853 lcn0 != lcn) && 4854 !run_add_entry(oa->run1, vcn, lcn, 1, false)) { 4855 err = -ENOMEM; 4856 goto out; 4857 } 4858 attr = oa->attr; 4859 if (size > le64_to_cpu(attr->nres.alloc_size)) { 4860 attr->nres.valid_size = attr->nres.data_size = 4861 attr->nres.alloc_size = cpu_to_le64(size); 4862 } 4863 goto next_dirty_page_vcn; 4864 4865 do_redo_1: 4866 /* 4867 * Perform the Redo Pass, to restore all of the dirty pages to the same 4868 * contents that they had immediately before the crash. If the dirty 4869 * page table is empty, then we can skip the entire Redo Pass. 4870 */ 4871 if (!dptbl || !dptbl->total) 4872 goto do_undo_action; 4873 4874 rec_lsn = rlsn; 4875 4876 /* 4877 * Read the record at the Redo lsn, before falling 4878 * into common code to handle each record. 4879 */ 4880 err = read_log_rec_lcb(log, rlsn, lcb_ctx_next, &lcb); 4881 if (err) 4882 goto out; 4883 4884 /* 4885 * Now loop to read all of our log records forwards, until 4886 * we hit the end of the file, cleaning up at the end. 4887 */ 4888 do_action_next: 4889 frh = lcb->lrh; 4890 4891 if (LfsClientRecord != frh->record_type) 4892 goto read_next_log_do_action; 4893 4894 transact_id = le32_to_cpu(frh->transact_id); 4895 rec_len = le32_to_cpu(frh->client_data_len); 4896 lrh = lcb->log_rec; 4897 4898 if (!check_log_rec(lrh, rec_len, transact_id, bytes_per_attr_entry)) { 4899 err = -EINVAL; 4900 goto out; 4901 } 4902 4903 /* Ignore log records that do not update pages. */ 4904 if (lrh->lcns_follow) 4905 goto find_dirty_page; 4906 4907 goto read_next_log_do_action; 4908 4909 find_dirty_page: 4910 t16 = le16_to_cpu(lrh->target_attr); 4911 t64 = le64_to_cpu(lrh->target_vcn); 4912 dp = find_dp(dptbl, t16, t64); 4913 4914 if (!dp) 4915 goto read_next_log_do_action; 4916 4917 if (rec_lsn < le64_to_cpu(dp->oldest_lsn)) 4918 goto read_next_log_do_action; 4919 4920 t16 = le16_to_cpu(lrh->target_attr); 4921 if (t16 >= bytes_per_rt(oatbl)) { 4922 err = -EINVAL; 4923 goto out; 4924 } 4925 4926 oe = Add2Ptr(oatbl, t16); 4927 4928 if (oe->next != RESTART_ENTRY_ALLOCATED_LE) { 4929 err = -EINVAL; 4930 goto out; 4931 } 4932 4933 oa = oe->ptr; 4934 4935 if (!oa) { 4936 err = -EINVAL; 4937 goto out; 4938 } 4939 attr = oa->attr; 4940 4941 vcn = le64_to_cpu(lrh->target_vcn); 4942 4943 if (!run_lookup_entry(oa->run1, vcn, &lcn, NULL, NULL) || 4944 lcn == SPARSE_LCN) { 4945 goto read_next_log_do_action; 4946 } 4947 4948 /* Point to the Redo data and get its length. */ 4949 data = Add2Ptr(lrh, le16_to_cpu(lrh->redo_off)); 4950 dlen = le16_to_cpu(lrh->redo_len); 4951 4952 /* Shorten length by any Lcns which were deleted. */ 4953 saved_len = dlen; 4954 4955 for (i = le16_to_cpu(lrh->lcns_follow); i; i--) { 4956 size_t j; 4957 u32 alen, voff; 4958 4959 voff = le16_to_cpu(lrh->record_off) + 4960 le16_to_cpu(lrh->attr_off); 4961 voff += le16_to_cpu(lrh->cluster_off) << SECTOR_SHIFT; 4962 4963 /* If the Vcn question is allocated, we can just get out. */ 4964 j = le64_to_cpu(lrh->target_vcn) - le64_to_cpu(dp->vcn); 4965 if (dp->page_lcns[j + i - 1]) 4966 break; 4967 4968 if (!saved_len) 4969 saved_len = 1; 4970 4971 /* 4972 * Calculate the allocated space left relative to the 4973 * log record Vcn, after removing this unallocated Vcn. 4974 */ 4975 alen = (i - 1) << sbi->cluster_bits; 4976 4977 /* 4978 * If the update described this log record goes beyond 4979 * the allocated space, then we will have to reduce the length. 4980 */ 4981 if (voff >= alen) 4982 dlen = 0; 4983 else if (voff + dlen > alen) 4984 dlen = alen - voff; 4985 } 4986 4987 /* 4988 * If the resulting dlen from above is now zero, 4989 * we can skip this log record. 4990 */ 4991 if (!dlen && saved_len) 4992 goto read_next_log_do_action; 4993 4994 t16 = le16_to_cpu(lrh->redo_op); 4995 if (can_skip_action(t16)) 4996 goto read_next_log_do_action; 4997 4998 /* Apply the Redo operation a common routine. */ 4999 err = do_action(log, oe, lrh, t16, data, dlen, rec_len, &rec_lsn); 5000 if (err) 5001 goto out; 5002 5003 /* Keep reading and looping back until end of file. */ 5004 read_next_log_do_action: 5005 err = read_next_log_rec(log, lcb, &rec_lsn); 5006 if (!err && rec_lsn) 5007 goto do_action_next; 5008 5009 lcb_put(lcb); 5010 lcb = NULL; 5011 5012 do_undo_action: 5013 /* Scan Transaction Table. */ 5014 tr = NULL; 5015 transaction_table_next: 5016 tr = enum_rstbl(trtbl, tr); 5017 if (!tr) 5018 goto undo_action_done; 5019 5020 if (TransactionActive != tr->transact_state || !tr->undo_next_lsn) { 5021 free_rsttbl_idx(trtbl, PtrOffset(trtbl, tr)); 5022 goto transaction_table_next; 5023 } 5024 5025 log->transaction_id = PtrOffset(trtbl, tr); 5026 undo_next_lsn = le64_to_cpu(tr->undo_next_lsn); 5027 5028 /* 5029 * We only have to do anything if the transaction has 5030 * something its undo_next_lsn field. 5031 */ 5032 if (!undo_next_lsn) 5033 goto commit_undo; 5034 5035 /* Read the first record to be undone by this transaction. */ 5036 err = read_log_rec_lcb(log, undo_next_lsn, lcb_ctx_undo_next, &lcb); 5037 if (err) 5038 goto out; 5039 5040 /* 5041 * Now loop to read all of our log records forwards, 5042 * until we hit the end of the file, cleaning up at the end. 5043 */ 5044 undo_action_next: 5045 5046 lrh = lcb->log_rec; 5047 frh = lcb->lrh; 5048 transact_id = le32_to_cpu(frh->transact_id); 5049 rec_len = le32_to_cpu(frh->client_data_len); 5050 5051 if (!check_log_rec(lrh, rec_len, transact_id, bytes_per_attr_entry)) { 5052 err = -EINVAL; 5053 goto out; 5054 } 5055 5056 if (lrh->undo_op == cpu_to_le16(Noop)) 5057 goto read_next_log_undo_action; 5058 5059 oe = Add2Ptr(oatbl, le16_to_cpu(lrh->target_attr)); 5060 oa = oe->ptr; 5061 5062 t16 = le16_to_cpu(lrh->lcns_follow); 5063 if (!t16) 5064 goto add_allocated_vcns; 5065 5066 is_mapped = run_lookup_entry(oa->run1, le64_to_cpu(lrh->target_vcn), 5067 &lcn, &clen, NULL); 5068 5069 /* 5070 * If the mapping isn't already the table or the mapping 5071 * corresponds to a hole the mapping, we need to make sure 5072 * there is no partial page already memory. 5073 */ 5074 if (is_mapped && lcn != SPARSE_LCN && clen >= t16) 5075 goto add_allocated_vcns; 5076 5077 vcn = le64_to_cpu(lrh->target_vcn); 5078 vcn &= ~(u64)(log->clst_per_page - 1); 5079 5080 add_allocated_vcns: 5081 for (i = 0, vcn = le64_to_cpu(lrh->target_vcn), 5082 size = (vcn + 1) << sbi->cluster_bits; 5083 i < t16; i++, vcn += 1, size += sbi->cluster_size) { 5084 attr = oa->attr; 5085 if (!attr->non_res) { 5086 if (size > le32_to_cpu(attr->res.data_size)) 5087 attr->res.data_size = cpu_to_le32(size); 5088 } else { 5089 if (size > le64_to_cpu(attr->nres.data_size)) 5090 attr->nres.valid_size = attr->nres.data_size = 5091 attr->nres.alloc_size = 5092 cpu_to_le64(size); 5093 } 5094 } 5095 5096 t16 = le16_to_cpu(lrh->undo_op); 5097 if (can_skip_action(t16)) 5098 goto read_next_log_undo_action; 5099 5100 /* Point to the Redo data and get its length. */ 5101 data = Add2Ptr(lrh, le16_to_cpu(lrh->undo_off)); 5102 dlen = le16_to_cpu(lrh->undo_len); 5103 5104 /* It is time to apply the undo action. */ 5105 err = do_action(log, oe, lrh, t16, data, dlen, rec_len, NULL); 5106 5107 read_next_log_undo_action: 5108 /* 5109 * Keep reading and looping back until we have read the 5110 * last record for this transaction. 5111 */ 5112 err = read_next_log_rec(log, lcb, &rec_lsn); 5113 if (err) 5114 goto out; 5115 5116 if (rec_lsn) 5117 goto undo_action_next; 5118 5119 lcb_put(lcb); 5120 lcb = NULL; 5121 5122 commit_undo: 5123 free_rsttbl_idx(trtbl, log->transaction_id); 5124 5125 log->transaction_id = 0; 5126 5127 goto transaction_table_next; 5128 5129 undo_action_done: 5130 5131 ntfs_update_mftmirr(sbi, 0); 5132 5133 sbi->flags &= ~NTFS_FLAGS_NEED_REPLAY; 5134 5135 end_replay: 5136 5137 err = 0; 5138 if (is_ro) 5139 goto out; 5140 5141 rh = kzalloc(log->page_size, GFP_NOFS); 5142 if (!rh) { 5143 err = -ENOMEM; 5144 goto out; 5145 } 5146 5147 rh->rhdr.sign = NTFS_RSTR_SIGNATURE; 5148 rh->rhdr.fix_off = cpu_to_le16(offsetof(struct RESTART_HDR, fixups)); 5149 t16 = (log->page_size >> SECTOR_SHIFT) + 1; 5150 rh->rhdr.fix_num = cpu_to_le16(t16); 5151 rh->sys_page_size = cpu_to_le32(log->page_size); 5152 rh->page_size = cpu_to_le32(log->page_size); 5153 5154 t16 = ALIGN(offsetof(struct RESTART_HDR, fixups) + sizeof(short) * t16, 5155 8); 5156 rh->ra_off = cpu_to_le16(t16); 5157 rh->minor_ver = cpu_to_le16(1); // 0x1A: 5158 rh->major_ver = cpu_to_le16(1); // 0x1C: 5159 5160 ra2 = Add2Ptr(rh, t16); 5161 memcpy(ra2, ra, sizeof(struct RESTART_AREA)); 5162 5163 ra2->client_idx[0] = 0; 5164 ra2->client_idx[1] = LFS_NO_CLIENT_LE; 5165 ra2->flags = cpu_to_le16(2); 5166 5167 le32_add_cpu(&ra2->open_log_count, 1); 5168 5169 ntfs_fix_pre_write(&rh->rhdr, log->page_size); 5170 5171 err = ntfs_sb_write_run(sbi, &ni->file.run, 0, rh, log->page_size, 0); 5172 if (!err) 5173 err = ntfs_sb_write_run(sbi, &log->ni->file.run, log->page_size, 5174 rh, log->page_size, 0); 5175 5176 kfree(rh); 5177 if (err) 5178 goto out; 5179 5180 out: 5181 kfree(rst); 5182 if (lcb) 5183 lcb_put(lcb); 5184 5185 /* 5186 * Scan the Open Attribute Table to close all of 5187 * the open attributes. 5188 */ 5189 oe = NULL; 5190 while ((oe = enum_rstbl(oatbl, oe))) { 5191 rno = ino_get(&oe->ref); 5192 5193 if (oe->is_attr_name == 1) { 5194 kfree(oe->ptr); 5195 oe->ptr = NULL; 5196 continue; 5197 } 5198 5199 if (oe->is_attr_name) 5200 continue; 5201 5202 oa = oe->ptr; 5203 if (!oa) 5204 continue; 5205 5206 run_close(&oa->run0); 5207 kfree(oa->attr); 5208 if (oa->ni) 5209 iput(&oa->ni->vfs_inode); 5210 kfree(oa); 5211 } 5212 5213 kfree(trtbl); 5214 kfree(oatbl); 5215 kfree(dptbl); 5216 kfree(attr_names); 5217 kfree(log->rst_info.r_page); 5218 5219 kfree(ra); 5220 kfree(log->one_page_buf); 5221 5222 if (err) 5223 sbi->flags |= NTFS_FLAGS_NEED_REPLAY; 5224 5225 if (err == -EROFS) 5226 err = 0; 5227 else if (log->set_dirty) 5228 ntfs_set_state(sbi, NTFS_DIRTY_ERROR); 5229 5230 kfree(log); 5231 5232 return err; 5233 } 5234