一个 NAND flash写函数

1. 
   
2. 一个刚看到时有点晕晕的flash的写函数，后来仔细看了后明白了；附上我的草稿图 
   
3. 
   
4. int tls_fls_write(u32 addr, u8 *buf, u32 len)  
5. {  
6.     u8 *cache;  
7.     int err;  
8.     u32 sector_addr;  
9.     u32 sector_num;  
10.     u32 write_bytes;  
11.     u32 i;  
12.     struct tls_fls_drv *drv;  
13.   
14.     if (spi_fls == NULL)  
15.     {  
16.         TLS_DBGPRT_ERR("flash driver module not beed installed!\n");  
17.         return TLS_FLS_STATUS_EPERM;  
18.     }  
19.       
20.     if (spi_fls->current_drv == NULL)  
21.     {  
22.         TLS_DBGPRT_ERR("the current spi flash driver not installed!\n");  
23.         return TLS_FLS_STATUS_ENODRV;  
24.     }  
25.       
26.     if ((addr >= spi_fls->current_drv->total_size) || (len == 0)  
27.         || (buf == NULL))  
28.     {  
29.         return TLS_FLS_STATUS_EINVAL;  
30.     }  
31.     tls_os_sem_acquire(spi_fls->fls_lock, 0);  
32.     drv = spi_fls->current_drv;  
33.     write_bytes =  
34.         ((addr + len) > drv->total_size) ? (drv->total_size - addr) : len;  
35.     sector_addr = addr/drv->sector_size;  
36.     sector_num = (addr + write_bytes - 1)/drv->sector_size - sector_addr + 1;  
37.   
38.     TLS_DBGPRT_FLASH("write to flash: sector address - %d, sectors - %d.\n", sector_addr, sector_num);  
39.       
40.     err = TLS_FLS_STATUS_OK;  
41.   
42.     cache = tls_mem_alloc(drv->sector_size);  
43.     if (cache == NULL)  
44.     {  
45.         tls_os_sem_release(spi_fls->fls_lock);  
46.         TLS_DBGPRT_ERR("allocate sector cache memory(%dB) fail!\n", drv->sector_size);  
47.         return TLS_FLS_STATUS_ENOMEM;  
48.     }  
49.   
50.     for (i = 0; i < sector_num; i++)  
51.     {  
52.         TLS_DBGPRT_FLASH("firstly, read the sector - %d to cache.\n", sector_addr + i);  
53.         err = drv->read((sector_addr + i) * drv->sector_size, cache, drv->sector_size);  
54.         if (err != TLS_FLS_STATUS_OK)  
55.         {  
56.             tls_os_sem_release(spi_fls->fls_lock);  
57.             TLS_DBGPRT_ERR("flash read fail(sector %d)!\n", (sector_addr + i));  
58.             break;  
59.         }  
60.   
61.         if (1 == sector_num){/*flash write only in one sector*/  
62.             MEMCPY(cache + (addr%drv->sector_size), buf, write_bytes);     
63.             buf += write_bytes;  
64.             write_bytes = 0;              
65.         }else{/*flash write through some sectors*/  
66.             if (0 == i) {  
67.                 MEMCPY(cache+(addr%drv->sector_size), buf, drv->sector_size - (addr%drv->sector_size));  
68.                 buf += drv->sector_size - (addr%drv->sector_size);  
69.                 write_bytes -= drv->sector_size - (addr%drv->sector_size);  
70.             } else if (i == (sector_num - 1)) {  
71.                 MEMCPY(cache, buf, write_bytes);  
72.                 buf += write_bytes;  
73.                 write_bytes = 0;  
74.             } else {  
75.                 MEMCPY(cache, buf, drv->sector_size);  
76.                 buf += drv->sector_size;  
77.                 write_bytes -= drv->sector_size;  
78.             }  
79.         }  
80.   
81.         TLS_DBGPRT_FLASH("second, erase the sector - %d.\n", sector_addr + i);  
82.         err = drv->erase(sector_addr + i);  
83.         if (err != TLS_FLS_STATUS_OK)  
84.         {  
85.             tls_os_sem_release(spi_fls->fls_lock);  
86.             TLS_DBGPRT_ERR("flash erase fail(sector %d)!\n", (sector_addr + i));  
87.             break;  
88.         }  
89.   
90.         TLS_DBGPRT_FLASH("finnaly, write the data in cache to the sector - %d.\n", sector_addr + i);  
91.         err = tls_fls_page_write((sector_addr +i) * (drv->sector_size / drv->page_size),  
92.                             cache, drv->sector_size / drv->page_size);  
93.         if (err != TLS_FLS_STATUS_OK)  
94.         {  
95.             tls_os_sem_release(spi_fls->fls_lock);  
96.             TLS_DBGPRT_ERR("flash write fail(sector %d)!\n", (sector_addr + i));  
97.             break;  
98.         }  
99.     }  
100.   
101.     tls_mem_free(cache);  
102.     tls_os_sem_release(spi_fls->fls_lock);  
103.     return err;  
104. }