last_pkmap_nr：記錄上次被分配的頁表項在pkmap_page_table裡的位置，初始值為０，所以第一次分配的時候last_pkmap_nr等於１。

接下來判斷什麼時候last_pkmap_nr等於０，等於０就表示1023（LAST_PKMAP(1024)-1）個頁表項已經被分配了,這時候就需要呼叫flush_all_zero_pkmaps()函式,把所有pkmap_count[]計數為1的頁表項在TLB裡面的entry給flush掉，並重置為0，這就表示該頁表項又可以用了，可能會有疑惑為什麼不在把pkmap_count置為1的時候也就是解除對映的同時把TLB也flush呢？個人感覺有可能是為了效率的問題吧，畢竟等到不夠的時候再重新整理，效率要好點吧。

再判斷pkmap_count[last_pkmap_nr]是否為0，0的話就表示這個頁表項是可用的，那麼就跳出迴圈了到下面了。

PKMAP_ADDR(last_pkmap_nr)返回這個頁表項對應的線性地址vaddr.

#define PKMAP_ADDR(nr) (PKMAP_BASE + ((nr) << PAGE_SHIFT))

set_pte_at(mm, addr, ptep, pte)函式在NON-PAE i386上的實現其實很簡單，其實就等同於下面的程式碼：

static inline void native_set_pte(pte_t *ptep , pte_t pte)
{
        *ptep = pte;
}
我們已經知道頁表的線性起始地址存放在pkmap_page_table裡面，那麼相應的可用的頁表項的地址就是&pkmap_page_table[last_pkmap_nr],得到了頁表項的地址，只要把相應的pte填寫進去，那麼整個對映不就完成了嗎？
pte由兩部分組成：高20位表示實體地址，低12位表示頁的描述資訊。
怎麼通過page查詢對應的實體地址呢（參考page_address()一文）？其實很簡單，用(page - mem_map) 再移PAGE_SHIFT位就可以了。
低12位的頁描述資訊是固定的:kmap_prot=(_PAGE_PRESENT | _PAGE_RW | _PAGE_DIRTY | _PAGE_ACCESSED | _PAGE_GLOBAL).
下面的程式碼就是做了這些事情：

mk_pte(page, kmap_prot));
#define mk_pte(page, pgprot) pfn_pte(page_to_pfn(page), (pgprot))
#define page_to_pfn __page_to_pfn
#define __page_to_pfn(page) ((unsigned long)((page) - mem_map) + \
                                 ARCH_PFN_OFFSET)
static inline pte_t pfn_pte(unsigned long page_nr, pgprot_t pgprot)
{
        return __pte(((phys_addr_t)page_nr << PAGE_SHIFT) |
                     massage_pgprot(pgprot));
}

接下來把pkmap_count[last_pkmap_nr]置為1，1不是表示不可用嗎，既然對映已經建立好了，應該賦值為2呀，其實這個操作是在他的上層函式kmap_high裡面完成的(pkmap_count[PKMAP_NR(vaddr)]++).
到此為止，整個對映就完成了，再把page和對應的線性地址加入到page_address_htable雜湊連結串列裡面就可以了（參考page_address一文）。


我們繼續看所有的頁表項都已經用了的情況下，也就是1024個頁表項全已經映射了記憶體了，如何處理。此時count==0,於是就進入了下面的程式碼:

/*
                 * Sleep for somebody else to unmap their entries
                 */
                {
                        DECLARE_WAITQUEUE(wait, current);
                
                        __set_current_state(TASK_UNINTERRUPTIBLE);
                        add_wait_queue(&pkmap_map_wait, &wait);
                        unlock_kmap();
                        schedule();
                        remove_wait_queue(&pkmap_map_wait, &wait);
                        lock_kmap();

                        /* Somebody else might have mapped it while we slept */
                        if (page_address(page))
                                return (unsigned long)page_address(page);

                        /* Re-start */
                        goto start;
                }
這段程式碼其實很簡單，就是把當前任務加入到等待佇列pkmap_map_wait，當有其他任務喚醒這個佇列時，再繼續goto start，重新整個過程。這裡就是上面說的呼叫kmap函式有可能阻塞的原因。
那麼什麼時候會喚醒pkmap_map_wait佇列呢？當呼叫kunmap_high函式，來釋放掉一個對映的時候。
kunmap_high函式其實頁很簡單，就是把要釋放的頁表項的計數減1，如果等於1的時候，表示有可用的頁表項了，再喚醒pkmap_map_wait佇列

/**
* kunmap_high - map a highmem page into memory
* @page: &struct page to unmap
*
* If ARCH_NEEDS_KMAP_HIGH_GET is not defined then this may be called
* only from user context.
*/
void kunmap_high(struct page *page)
{
        unsigned long vaddr;
        unsigned long nr;
        unsigned long flags;
        int need_wakeup;

        lock_kmap_any(flags);
        vaddr = (unsigned long)page_address(page);
        BUG_ON(!vaddr);
        nr = PKMAP_NR(vaddr);

        /*
         * A count must never go down to zero
         * without a TLB flush!
         */
        need_wakeup = 0;
        switch (--pkmap_count[nr]) {//減一
        case 0:
                BUG();
        case 1:
                /*
                 * Avoid an unnecessary wake_up() function call.
                 * The common case is pkmap_count[] == 1, but
                 * no waiters.
                 * The tasks queued in the wait-queue are guarded
                 * by both the lock in the wait-queue-head and by
                 * the kmap_lock. As the kmap_lock is held here,
                 * no need for the wait-queue-head's lock. Simply
                 * test if the queue is empty.
                 */
                need_wakeup = waitqueue_active(&pkmap_map_wait);
        }
        unlock_kmap_any(flags);

        /* do wake-up, if needed, race-free outside of the spin lock */
        if (need_wakeup)
                wake_up(&pkmap_map_wait);
}

轉自： http://bbs.chinaunix.net/thread-1938084-1-1.html