在内核中分配内存，最后要通过伙伴系统接口进行实际物理页面的分配，一个重要的接口便是alloc_page.本文介绍下alloc_page的主要流程，各个部分的执行。主要包含正常分配流程，当页面不足的时候的处理方式。先定位到核心调用
#define alloc_page(gfp_mask) alloc_pages(gfp_mask, 0)
order是分配页面的阶，即2的指数个页面
#define alloc_pages(gfp_mask, order) alloc_pages_node(numa_node_id(), gfp_mask, order)
nid指定了从哪个NUMA节点分配页面，如果没有指定节点，则默认从当前节点分配
static inline struct page *alloc_pages_node(int nid, gfp_t gfp_mask, unsigned int order) { /* Unknown node is current node */ /*如果没有指定node ，则从当前node分配*/ if (nid < 0) nid = numa_node_id(); return __alloc_pages(gfp_mask, order, node_zonelist(nid, gfp_mask)); }
zonelist是一组zone的列表，有两种，局部的和全局的。局部zonelist只包含本节点的zone，全局的包含所有节点的zone。下篇文章会详细介绍这些数据结构
static inline struct page * __alloc_pages(gfp_t gfp_mask, unsigned int order, struct zonelist *zonelist) { return __alloc_pages_nodemask(gfp_mask, order, zonelist, NULL); }
到了__alloc_pages_nodemask就进入了比较正式的流程了，拨开来讲，本函数主要包含两步：
1、直接分配
2、分配失败选择另一种方式即slowpath继续处理.
首次尝试分配是调用了static struct page *get_page_from_freelist(gfp_t gfp_mask, nodemask_t *nodemask, unsigned int order,struct zonelist *zonelist, int high_zoneidx, int alloc_flags,struct zone *preferred_zone, int migratetype)，核心机制就是遍历zonelist上的zone，找到一个page.函数代码倒是没什么可说的，比较容易理解，这里主要看下涉及到的几个机制。该函数主要实现功能：1、在zonelist中找到一个合适的zone 2、从zone中分配页面。前者由一个循环体完成，后者由static inline struct page *buffered_rmqueue(struct zone *preferred_zone,struct zone *zone, int order, gfp_t gfp_flags,int migratetype)完成。
在选定zone的阶段，在正常情况下需要进行一系列的验证，保证当前zone有足够的可用页面供分配。那么什么是非正常情况呢？即使携带ALLOC_NO_WATERMARKS标识的，所以这里就分为两种情况。这里涉及到一个watermark,俗称分配水位，水位有三种
- #define ALLOC_WMARK_MIN WMARK_MIN
- #define ALLOC_WMARK_LOW WMARK_LOW
- #define ALLOC_WMARK_HIGH WMARK_HIGH
在分配之前一般会指定满足那个水位才允许分配，或者不管水位直接分配，这就对应ALLOC_NO_WATERMARKS标识。在zone结构中，有vm_stat字段，是一个数组，记录各个状态的页面的数量，其中就包含空闲页面，对应NR_FREE_PAGES，携带watermark标识的分配，需要验证空闲页面是否大于对应的水位，只有在大于水位了才允许分配，否则需要根据情况对页面进行回收reclaim，如果无法回收或者回收后仍然不满足条件，则直接返回了。在一些急迫的事务中，可以指定ALLOC_NO_WATERMARKS，这样会不会对水位进行验证，直接调用buffered_rmqueue分配页面。
buffered_rmqueue并不直接从伙伴系统分配，为了加速分配流程，每个CPU也会维护页框高速缓存,通过per_cpu_pageset管理
struct per_cpu_pageset { struct per_cpu_pages pcp; #ifdef CONFIG_NUMA s8 expire; #endif #ifdef CONFIG_SMP s8 stat_threshold; s8 vm_stat_diff[NR_VM_ZONE_STAT_ITEMS]; #endif };
其中pcp维护了各种性质的页面链表，性质基本是根据可移动性来决定的。
struct per_cpu_pages { int count; /* number of pages in the list */ int high; /* high watermark, emptying needed */ int batch; /* chunk size for buddy add/remove */ /* Lists of pages, one per migrate type stored on the pcp-lists */ /*链表数组，每个迁移类型维护一个数组*/ /*MIGRATE_UNMOVABLE, MIGRATE_RECLAIMABLE, MIGRATE_MOVABLE, MIGRATE_PCPTYPES, // the number of types on the pcp lists MIGRATE_RESERVE*/ struct list_head lists[MIGRATE_PCPTYPES]; };
count链表中所有页面的数量，high和清空相关，而batch是当缓存不足时，每次从伙伴系统申请多少页面填充进来。
核心分配逻辑如下：
if (likely(order == 0)) { struct per_cpu_pages *pcp; struct list_head *list; local_irq_save(flags); /*页框高速缓存*/ pcp = &this_cpu_ptr(zone->pageset)->pcp; /*获取缓存链表*/ list = &pcp->lists[migratetype]; /*如果链表为空*/ if (list_empty(list)) { /*尝试从伙伴系统填充链表*/ pcp->count += rmqueue_bulk(zone, 0, pcp->batch, list, migratetype, cold); /*如果依然为空，则失败*/ if (unlikely(list_empty(list))) goto failed; } /*list是双链表，如果cold为真就从表尾部分配，否则从表头分配*/ if (cold) page = list_entry(list->prev, struct page, lru); else page = list_entry(list->next, struct page, lru); /*页面从链表删除*/ list_del(&page->lru); pcp->count--; }
slow alloc path 见__alloc_pages_slowpath函数（page_alloc.c）
static inline struct page * __alloc_pages_slowpath(gfp_t gfp_mask, unsigned int order, struct zonelist *zonelist, enum zone_type high_zoneidx, nodemask_t *nodemask, struct zone *preferred_zone, int migratetype) { const gfp_t wait = gfp_mask & __GFP_WAIT; struct page *page = NULL; int alloc_flags; unsigned long pages_reclaimed = 0; unsigned long did_some_progress; bool sync_migration = false; bool deferred_compaction = false; bool contended_compaction = false; /* * In the slowpath, we sanity check order to avoid ever trying to * reclaim >= MAX_ORDER areas which will never succeed. Callers may * be using allocators in order of preference for an area that is * too large. */ if (order >= MAX_ORDER) { WARN_ON_ONCE(!(gfp_mask & __GFP_NOWARN)); return NULL; } /* * GFP_THISNODE (meaning __GFP_THISNODE, __GFP_NORETRY and * __GFP_NOWARN set) should not cause reclaim since the subsystem * (f.e. slab) using GFP_THISNODE may choose to trigger reclaim * using a larger set of nodes after it has established that the * allowed per node queues are empty and that nodes are * over allocated. */ if (IS_ENABLED(CONFIG_NUMA) && (gfp_mask & GFP_THISNODE) == GFP_THISNODE) goto nopage; /*到这一步如果允许启用kswapd线程，则唤醒所有的kswapd*/ restart: if (!(gfp_mask & __GFP_NO_KSWAPD)) wake_all_kswapd(order, zonelist, high_zoneidx, zone_idx(preferred_zone)); /* * OK, we're below the kswapd watermark and have kicked background * reclaim. Now things get more complex, so set up alloc_flags according * to how we want to proceed. */ alloc_flags = gfp_to_alloc_flags(gfp_mask); /* * Find the true preferred zone if the allocation is unconstrained by * cpusets. */ /*如果没哟没有指定cpuset，则选取最优的zonelist*/ if (!(alloc_flags & ALLOC_CPUSET) && !nodemask) first_zones_zonelist(zonelist, high_zoneidx, NULL, &preferred_zone); rebalance: /* This is the last chance, in general, before the goto nopage. */ /*尝试分配页面*/ page = get_page_from_freelist(gfp_mask, nodemask, order, zonelist, high_zoneidx, alloc_flags & ~ALLOC_NO_WATERMARKS, preferred_zone, migratetype); if (page) goto got_pg; /*如果允许不管区域的watermark*/ /* Allocate without watermarks if the context allows */ if (alloc_flags & ALLOC_NO_WATERMARKS) { /* * Ignore mempolicies if ALLOC_NO_WATERMARKS on the grounds * the allocation is high priority and these type of * allocations are system rather than user orientated */ zonelist = node_zonelist(numa_node_id(), gfp_mask); /*高优先级的内存分配，遍历整个zonelist*/ page = __alloc_pages_high_priority(gfp_mask, order, zonelist, high_zoneidx, nodemask, preferred_zone, migratetype); if (page) { goto got_pg; } } /* Atomic allocations - we can't balance anything */ if (!wait) goto nopage; /* Avoid recursion of direct reclaim */ if (current->flags & PF_MEMALLOC) goto nopage; /* Avoid allocations with no watermarks from looping endlessly */ if (test_thread_flag(TIF_MEMDIE) && !(gfp_mask & __GFP_NOFAIL)) goto nopage; /* * Try direct compaction. The first pass is asynchronous. Subsequent * attempts after direct reclaim are synchronous */ page = __alloc_pages_direct_compact(gfp_mask, order, zonelist, high_zoneidx, nodemask, alloc_flags, preferred_zone, migratetype, sync_migration, &contended_compaction, &deferred_compaction, &did_some_progress); if (page) goto got_pg; sync_migration = true; /* * If compaction is deferred for high-order allocations, it is because * sync compaction recently failed. In this is the case and the caller * requested a movable allocation that does not heavily disrupt the * system then fail the allocation instead of entering direct reclaim. */ if ((deferred_compaction || contended_compaction) && (gfp_mask & __GFP_NO_KSWAPD)) goto nopage; /* Try direct reclaim and then allocating */ /*如果压缩后仍然没哟u可用的，就对页面进行回收*/ page = __alloc_pages_direct_reclaim(gfp_mask, order, zonelist, high_zoneidx, nodemask, alloc_flags, preferred_zone, migratetype, &did_some_progress); if (page) goto got_pg; /* * If we failed to make any progress reclaiming, then we are * running out of options and have to consider going OOM */ if (!did_some_progress) { if ((gfp_mask & __GFP_FS) && !(gfp_mask & __GFP_NORETRY)) { if (oom_killer_disabled) goto nopage; /* Coredumps can quickly deplete all memory reserves */ if ((current->flags & PF_DUMPCORE) && !(gfp_mask & __GFP_NOFAIL)) goto nopage; page = __alloc_pages_may_oom(gfp_mask, order, zonelist, high_zoneidx, nodemask, preferred_zone, migratetype); if (page) goto got_pg; if (!(gfp_mask & __GFP_NOFAIL)) { /* * The oom killer is not called for high-order * allocations that may fail, so if no progress * is being made, there are no other options and * retrying is unlikely to help. */ if (order > PAGE_ALLOC_COSTLY_ORDER) goto nopage; /* * The oom killer is not called for lowmem * allocations to prevent needlessly killing * innocent tasks. */ if (high_zoneidx < ZONE_NORMAL) goto nopage; } goto restart; } } /* Check if we should retry the allocation */ pages_reclaimed += did_some_progress; if (should_alloc_retry(gfp_mask, order, did_some_progress, pages_reclaimed)) { /* Wait for some write requests to complete then retry */ wait_iff_congested(preferred_zone, BLK_RW_ASYNC, HZ/50); goto rebalance; } else { /* * High-order allocations do not necessarily loop after * direct reclaim and reclaim/compaction depends on compaction * being called after reclaim so call directly if necessary */ page = __alloc_pages_direct_compact(gfp_mask, order, zonelist, high_zoneidx, nodemask, alloc_flags, preferred_zone, migratetype, sync_migration, &contended_compaction, &deferred_compaction, &did_some_progress); if (page) goto got_pg; } nopage: warn_alloc_failed(gfp_mask, order, NULL); return page; got_pg: if (kmemcheck_enabled) kmemcheck_pagealloc_alloc(page, order, gfp_mask); return page; }
该函数主要是在首次分配失败的情况下，采取的补救解决方案。由于首次分配的失败，要么是缓存不足，要么是真没有剩余的空闲页面了。那么我们还能怎么办？大家可能都能想到，回收内存呗！！不错，回收内存的确是当前最主要的解决思路，看下代码，在__alloc_pages_slowpath函数中，首次的调用便是wake_all_kswapd，kswapd都晓得伐，负责物理页面的换出的，定期执行以保证物理页面的实时可用性，但是毕竟是定期执行，不能保证任何时刻都有充足的内存使用，所以这里尝试唤醒所有的kswapd守护进程进行物理页面的回收。而物理页面的回收涉及了更多的东西，本文不打算深入，后续有空在单独分析。到这里已经唤醒了后台的回收线程，Ok，重新尝试下分配，继续调用get_page_from_freelist，不过此时着重设置了一个参数，alloc_flags & ~ALLOC_NO_WATERMARKS即本次尝试一定要符合水位的情况，不要求特殊处理。这样如果分配得到页面，就大功告成。如果还是没有分配到，好吧，降低要求，允许忽略水位限制，调用__alloc_pages_high_priority进行分配。到这里，如果还没有分配到怎么办？判断wait标识，如果允许等待，则接着往下走；否则返回分配失败吧！没办法
接下来既然回收页面也不行，那么我不换出页面，把内存中的页面进行适当压缩总可以吧！！恩~总算中和了双方，调用了__alloc_pages_direct_compact，额，压缩页面是最后一个解决方案了，如果在不行可真是山穷水尽了，这样里OOM就不远了……
该函数主要分两步：
1、压缩区域内的页面try_to_compact_pages
2、尝试分配页面 get_page_from_freelist
分配页面前面已经介绍，这里就不用在多说了，重点看try_to_compact_pages，该函数会按照zonelist中的顺序对各个zone进行压缩，核心逻辑看下
for_each_zone_zonelist_nodemask(zone, z, zonelist, high_zoneidx, nodemask) { int status; status = compact_zone_order(zone, order, gfp_mask, sync, contended); rc = max(status, rc); /* If a normal allocation would succeed, stop compacting */ if (zone_watermark_ok(zone, order, low_wmark_pages(zone), 0, alloc_flags)) break; }
主要是compact_zone_order
static unsigned long compact_zone_order(struct zone *zone, int order, gfp_t gfp_mask, bool sync, bool *contended) { unsigned long ret; /*压缩控制器*/ struct compact_control cc = { .nr_freepages = 0, .nr_migratepages = 0, .order = order, .migratetype = allocflags_to_migratetype(gfp_mask), .zone = zone, .sync = sync, }; INIT_LIST_HEAD(&cc.freepages); INIT_LIST_HEAD(&cc.migratepages); /*执行压缩*/ ret = compact_zone(zone, &cc); VM_BUG_ON(!list_empty(&cc.freepages)); VM_BUG_ON(!list_empty(&cc.migratepages)); *contended = cc.contended; return ret; }
这里有一个结构，姑且称之为压缩控制器，记录压缩过程中需要的一些参数，设置好后又调用了compact_zone，该函数不在列举，感兴趣的参考源代码compaction.c,在该函数中，首先调用了compaction_suitable检查下当前zone是否有压缩的潜质，即通过压缩是否可以达到要求，如果可以就执行压缩，否则，呵呵……忽略吧！该函数返回三种值：
COMPACT_SKIPPED - If there are too few free pages for compaction
COMPACT_PARTIAL - If the allocation would succeed without compaction
COMPACT_CONTINUE - If compaction should run now*/
解释的比较明确，我就不多说了，唯一的一点是COMPACT_PARTIAL既然表示在不压缩的情况下也可以分配成功，为何不直接返回成功，然后尝试分配呢？还要走到下面的流程，真实捉急……根据上面的解释可能大家也都明白了，这里压缩并不是把几个页面压缩成一个页面，本质还是整理碎片，把可移动的页面尽量安排在一处，腾出来比较大的连续空间，这样增加满足需求的可能性。接下来就该执行压缩了，在此之前设置了压缩控制器的一些参数，所以有必要看下该结构
struct compact_control { struct list_head freepages; /* List of free pages to migrate to */ struct list_head migratepages; /* List of pages being migrated */ unsigned long nr_freepages; /* Number of isolated free pages */ unsigned long nr_migratepages; /* Number of pages to migrate */ unsigned long free_pfn; /* isolate_freepages search base */ unsigned long migrate_pfn; /* isolate_migratepages search base */ bool sync; /* Synchronous migration */ bool ignore_skip_hint; /* Scan blocks even if marked skip */ bool finished_update_free; /* True when the zone cached pfns are * no longer being updated */ bool finished_update_migrate; int order; /* order a direct compactor needs */ int migratetype; /* MOVABLE, RECLAIMABLE etc */ struct zone *zone; bool contended; /* True if a lock was contended */ };
如果看字段中的注释有点蒙，建议看下结构之前的注释
/* * compact_control is used to track pages being migrated and the free pages * they are being migrated to during memory compaction. The free_pfn starts * at the end of a zone and migrate_pfn begins at the start. Movable pages * are moved to the end of a zone during a compaction run and the run * completes when free_pfn <= migrate_pfn */
英语不行，简要翻译下。compact_control用于在内存压缩过程中追踪正在被移动的页面和被移动的目的页面，free_pfn起始于zone的末尾，而migrate_pfn起始于zone的起始位置。在压缩过程中，可移动的页面被移动到zone的尾部。大致就这个意思，说白了，就是free_pfn从后往前扫描寻找空闲页面，migrate_pfn从前往后扫描，寻找可移动的页面，当free_pfn <= migrate_pfn，压缩就结束了！
在执行压缩之前，对压缩控制器参数的调整
cc->migrate_pfn = zone->compact_cached_migrate_pfn; cc->free_pfn = zone->compact_cached_free_pfn; if (cc->free_pfn < start_pfn || cc->free_pfn > end_pfn) { cc->free_pfn = end_pfn & ~(pageblock_nr_pages-1); zone->compact_cached_free_pfn = cc->free_pfn; } if (cc->migrate_pfn < start_pfn || cc->migrate_pfn > end_pfn) { cc->migrate_pfn = start_pfn; zone->compact_cached_migrate_pfn = cc->migrate_pfn; }
在设置好压缩控制器之后，就调用了migrate_prep_local函数，该函数中主要调用lru_add_drain把当前LRU缓存移动到LRU链表。下面的一个while循环时压缩的主体，我么看下源码
while ((ret = compact_finished(zone, cc)) == COMPACT_CONTINUE) { unsigned long nr_migrate, nr_remaining; int err; switch (isolate_migratepages(zone, cc)) { case ISOLATE_ABORT: ret = COMPACT_PARTIAL; putback_movable_pages(&cc->migratepages); cc->nr_migratepages = 0; goto out; case ISOLATE_NONE: continue; case ISOLATE_SUCCESS: ; } nr_migrate = cc->nr_migratepages; err = migrate_pages(&cc->migratepages, compaction_alloc, (unsigned long)cc, cc->sync ? MIGRATE_SYNC_LIGHT : MIGRATE_ASYNC, MR_COMPACTION); update_nr_listpages(cc); nr_remaining = cc->nr_migratepages; trace_mm_compaction_migratepages(nr_migrate - nr_remaining, nr_remaining); /* Release isolated pages not migrated */ if (err) { putback_movable_pages(&cc->migratepages); cc->nr_migratepages = 0; if (err == -ENOMEM) { ret = COMPACT_PARTIAL; goto out; } } }
循环的判断条件函数是compact_finished，主体就是判断压缩是否完成，主要包含两部分
1、是否有fatal信号处于pending状态，如果有则立刻退返回COMPACT_PARTIAL，否则转入2。
2、检查压缩流程是否走完，即cc->free_pfn <= cc->migrate_pfn是否满足，如果满足则返回COMPACT_COMPLETE，否则转入3。
3、cc->order == -1，返回COMPACT_CONTINUE表示继续压缩，否则转入4
4、检查zone的的freepage是否满足low 的watermark，如果不满足则返回COMPACT_CONTINUE，否则转入5
5、从cc->order开始，依次检查zone->free_area数组，对各个长度的连续页面集合中，检查符合cc->migratetype的链表是否为空，如果不为空，返回COMPACT_PARTIAL，否则转入6
6、返回COMPACT_CONTINUE
循环体内部调用isolate_migratepages，该函数是压缩的主体函数，其中调用了isolate_migratepages_range，该函数实现把特定的isolate开，即从LRU链表中摘除，添加到cc->migratepages链表，而下面执行移动的是migrate_pages，该函数实现了吧刚才隔离开的页面，移动到新的地方，核心即使就是对当前撤销映射，然后对新的页面建立映射。接着就调用update_nr_listpages对控制器做了更新，但是如果出错了就会调用putback_movable_pages把隔离开的页面重新添加回去，保证系统正常运行。
压缩函数返回后，会再次尝试获取页面，如果获取到了，OK，一切照旧；如果没获取到，那也没办法，向上峰报告吧，该做什么做什么……
以马内利
参考：linux3.10.1源码