LK · source analysis

lkboot의 network image 수신과 chain loading

LK의 lkboot 앱이 이미지를 수신하고 진입 주소로 제어권을 넘기는 과정을 살펴봅니다.

표의 버전은 소스 분석 기준이며 최신 보안 버전 권고가 아닙니다. TF-A LTS 2.14.3 이후에는 보안 수정이 있습니다. 제품 적용 시 공식 변경 기록과 사용 플랫폼의 errata를 확인하세요.

Series
14 / 20
Baseline
upstream 1eb9917
Commit
1eb99171efae
Source
app/lkboot/lkboot.c:263-337 (발췌 범위)

lkboot의 network image 수신과 chain loading 단계에서 실제로 바뀌는 상태는 무엇인가?

여기서 다루는 LK의 lkboot는 과거 Android aboot와 다르다. 프로토콜 명령, 이미지 메모리를 관리하는 주체, 이미지를 실행하기 전 하드웨어 동작을 정지시키는 절차를 구분해 살펴본다.

수신 길이와 허용 memory range를 검증한 image만 실행 가능한 entry로 사용해야 한다. 이 문장을 기준으로 코드를 위에서 아래로 읽으면, 함수 이름을 외우는 대신 어느 시점에 어떤 상태를 신뢰할 수 있는지 판단할 수 있다.

객체와 주소가 놓이는 구조

그림 1. lkboot의 network image 수신과 chain loading에서 입력, 내부 상태, 출력의 경계
입력과 전제내부 상태외부로 공개되는 결과
01protocol session
02download buffer
03image metadata
04target entry

lkb_t / downloaded image를 중심에 놓고 왼쪽의 입력이 어떤 검사를 거쳐 오른쪽 결과로 공개되는지 표시했다. 실제 디버깅에서는 각 블록의 주소와 크기를 로그에 대입한다.

실행 흐름

그림 2. 정상 경로의 주요 호출과 상태 전달
network/USB transport
lkboot protocol
image buffer
validate command
chain load

화살표는 단순 호출 순서만 뜻하지 않는다. 각 단계가 성공을 반환할 때 다음 단계가 읽을 수 있는 상태가 무엇인지 함께 확인한다. 오류 반환이 발생하면 바로 다음 화살표로 진행하지 않고 해당 단계의 정리 경로를 따라간다.

단계별 입력과 출력

각 단계의 입력과 처리 후 바뀌는 상태를 정리한 표다. 사용 중인 소스 버전에서 다른 필드를 변경한다면 그에 맞춰 표도 수정한다.

#단계진입 시 신뢰할 상태성공 뒤 남아야 할 상태다음 소비자
01network/USB transportlkboot app threadprotocol sessionlkboot protocol
02lkboot protocolnetwork/USB transport 완료 상태download bufferimage buffer
03image bufferlkboot protocol 완료 상태image metadatavalidate command
04validate commandimage buffer 완료 상태target entrychain load
05chain loadvalidate command 완료 상태target entry이 단계의 처리 결과를 사용하는 코드

분석 범위의 전제와 비교해, 다음 단계에 잘못된 입력을 넘긴 부분부터 조사한다.

주소와 객체의 이동을 그림으로 읽기

아래 그림은 주소, 객체를 사용하는 기간, 제어권을 넘기는 시점을 중심으로 정리했다. 실제 주소와 크기는 사용 중인 보드의 bdinfo, 맵 파일과 링커 심벌로 확인한다.

그림 3. lkboot transport에서 다음 image로 제어권을 넘기는 과정

호출 이름만 나열하지 않고, 각 단계가 다음 단계에 넘기는 상태를 기준으로 실행 순서를 표시한다.

01transport command
02image metadata 수신
03payload read
04range·hash 검증
05destination 배치
06cache·device 정리
07entry branch
그림 4. chain-loading 직전 RAM의 대표 배치

크기는 개념 비율이다. 실제 분석에서는 map, symbol과 runtime 출력의 시작·끝 주소를 대입한다.

높은 주소↓ 주소 감소
LK text·data현재 실행 image
LK stack·heaphandoff 완료 전 유지
transport bufferUSB/UART input
next-stage metadatasize·entry·flags
next-stage image검증된 destination
unused RAMsource와 destination 비중첩
낮은 주소예약 cursor 진행 방향

원본 코드

아래 코드는 LK upstream 1eb9917의 app/lkboot/lkboot.c에서 263-337줄을 그대로 가져온 것이다. 설명을 위해 실제 코드를 가짜 의사 코드로 바꾸지 않았다.

commit1eb99171efae98867d4be4a5bd6479ea03f43ac9anchor · L287static void lkboot_task(const struct app_descriptor *app
263            printf("lkboot: handled command, staying in server loop\n");
264        }
265
266        /* see if we need to drop out and try to direct boot */
267        if (timeout != INFINITE_TIME && (current_time() - t >= timeout)) {
268            break;
269        }
270    }
271
272#if WITH_LIB_MINIP
273    tcp_close(listen_socket);
274#endif
275
276    printf("lkboot: server timed out\n");
277
278    return ERR_TIMED_OUT;
279}
280
281/* platform code can override this to conditionally abort autobooting from flash */
282bool platform_abort_autoboot(void);
283__WEAK bool platform_abort_autoboot(void) {
284    return false;
285}
286
287static void lkboot_task(const struct app_descriptor *app, void *args) {
288    /* read a few sysparams to decide if we're going to autoboot */
289    uint8_t autoboot = 1;
290    sysparam_read("lkboot.autoboot", &autoboot, sizeof(autoboot));
291
292    /* let platform code have a shot at disabling the autoboot behavior */
293    if (platform_abort_autoboot())
294        autoboot = 0;
295
296#if !LKBOOT_AUTOBOOT
297    autoboot = 0;
298#endif
299
300    /* if we're going to autoobot, read the timeout value */
301    lk_time_t autoboot_timeout;
302    if (!autoboot) {
303        autoboot_timeout = INFINITE_TIME;
304    } else {
305        autoboot_timeout = LKBOOT_AUTOBOOT_TIMEOUT;
306        sysparam_read("lkboot.autoboot_timeout", &autoboot_timeout, sizeof(autoboot_timeout));
307    }
308
309    TRACEF("autoboot %u autoboot_timeout %u\n", autoboot, (uint)autoboot_timeout);
310
311#if LKBOOT_WITH_SERVER
312    lkboot_server(autoboot_timeout);
313#else
314    if (autoboot_timeout != INFINITE_TIME) {
315        TRACEF("waiting for %u milliseconds before autobooting\n", (uint)autoboot_timeout);
316        thread_sleep(autoboot_timeout);
317    }
318#endif
319
320    if (autoboot_timeout != INFINITE_TIME) {
321        TRACEF("trying to boot from flash...\n");
322        status_t err = do_flash_boot();
323        TRACEF("do_flash_boot returns %d\n", err);
324    }
325
326#if LKBOOT_WITH_SERVER
327    TRACEF("restarting server\n");
328    lkboot_server(INFINITE_TIME);
329#endif
330
331    TRACEF("nothing to do, exiting\n");
332}
333
334APP_START(lkboot)
335.entry = lkboot_task,
336.flags = 0,
337APP_END

발췌 범위와 읽기 기준

원문 263-337줄에는 앞뒤 문맥이 포함됩니다. 기준 구문 static void lkboot_task(const struct app_descriptor *app은 L287에 있습니다. 발췌 첫 줄을 함수 정의의 시작으로 해석하지 않습니다.

여기서 다루는 LK의 lkboot는 과거 Android aboot와 다르다. 프로토콜 명령, 이미지 메모리를 관리하는 주체, 이미지를 실행하기 전 하드웨어 동작을 정지시키는 절차를 구분해 살펴본다.

함께 읽어야 하는 원본 코드

첫 코드 조각만으로 동작이 완성되지 않는 경우 호출 매크로, 객체 정의와 실제 실행 목록을 같은 페이지에 묶었다. 각 조각은 같은 기준 commit에서 가져왔다.

01. transport packet을 lkboot command로 해석하는 함수

app/lkboot/lkboot.c 169-251줄이다.

fileapp/lkboot/lkboot.canchor · L179status_t lkboot_process_command
169        len -= lkb->avail;
170        lkb->avail = 0;
171    }
172    return 0;
173
174fail:
175    lkb->state = STATE_ERROR;
176    return -1;
177}
178
179status_t lkboot_process_command(lkb_t *lkb) {
180    msg_hdr_t hdr;
181    char cmd[128];
182    char *arg;
183    int err;
184    const char *result;
185    unsigned len;
186
187    if (lkb->read(lkb->cookie, &hdr, sizeof(hdr))) goto fail;
188    if (hdr.opcode != MSG_CMD) goto fail;
189    if (hdr.length > 127) goto fail;
190    if (lkb->read(lkb->cookie, cmd, hdr.length)) goto fail;
191    cmd[hdr.length] = 0;
192
193    TRACEF("recv '%s'\n", cmd);
194
195    if (!(arg = strchr(cmd, ':'))) goto fail;
196    *arg++ = 0;
197    len = atoul(arg);
198    if (!(arg = strchr(arg, ':'))) goto fail;
199    arg++;
200
201    err = lkb_handle_command(lkb, cmd, arg, len, &result);
202    if (err >= 0) {
203        lkb_okay(lkb);
204    } else {
205        lkb_fail(lkb, result);
206    }
207
208    TRACEF("command handled with success\n");
209    return NO_ERROR;
210
211fail:
212    TRACEF("command failed\n");
213    return ERR_IO;
214}
215
216static status_t lkboot_server(lk_time_t timeout) {
217    lkboot_dcc_init();
218
219#if WITH_LIB_MINIP
220    /* open the server's socket */
221    tcp_socket_t *listen_socket = NULL;
222    if (tcp_open_listen(&listen_socket, 1023) < 0) {
223        printf("lkboot: error opening listen socket\n");
224        return ERR_NO_MEMORY;
225    }
226#endif
227
228    /* run the main lkserver loop */
229    printf("lkboot: starting server\n");
230    lk_time_t t = current_time(); /* remember when we started */
231    for (;;) {
232        bool handled_command = false;
233
234        lkb_t *lkb;
235
236#if WITH_LIB_MINIP
237        /* wait for a new connection */
238        lk_time_t sock_timeout = 100;
239        tcp_socket_t *s;
240        if (tcp_accept_timeout(listen_socket, &s, sock_timeout) >= 0) {
241            DEBUG_ASSERT(s);
242
243            /* handle the command and close it */
244            lkb = lkboot_tcp_opened(s);
245            lkboot_process_command(lkb);
246            free(lkb);
247            tcp_close(s);
248            handled_command = true;
249        }
250#endif
251

내부 동작을 더 깊게 읽기

01

진입 조건을 먼저 고정한다

network/USB transport에서 들어온 실행은 lkboot app thread에 놓여 있다. 이때 interrupt, MMU/cache, stack, heap 중 무엇이 이미 준비되었는지 소스의 호출자까지 올라가 확인한다. 같은 함수라도 SPL, relocation 전후, app thread처럼 호출 문맥이 달라지면 허용되는 API와 지연 시간이 달라진다.

여기서 다루는 LK의 lkboot는 과거 Android aboot와 다르다. 프로토콜 명령, 이미지 메모리를 관리하는 주체, 이미지를 실행하기 전 하드웨어 동작을 정지시키는 절차를 구분해 살펴본다.

02

중심 객체의 생성과 공개를 나눈다

이 글의 중심 객체는 lkb_t / downloaded image다. 메모리를 확보한 시점, 필드를 채운 시점, 전역 list나 다른 subsystem에 공개한 시점을 구분한다. 공개 뒤 오류가 발생한다면 목록에서 제거하고 child, buffer, reference를 역순으로 정리하는지 확인한다.

빌드 산출물 관점에서는 최종 부트로더 이미지 안에 해당 symbol과 section이 실제로 포함되었는지도 map과 objdump로 검증한다.

03

주소, 크기와 정렬을 계산한다

부트 코드에서는 주소 계산 오류를 주의해야 한다. 원본과 목적지의 주소 범위, 헤더에 기록된 데이터 크기, 블록·페이지 단위를 표로 정리하고 각 구간의 끝 주소를 계산한다. 끝 주소가 start + size - 1처럼 마지막 바이트를 가리키는지, 구간 바로 다음 주소를 가리키는지는 API 규약에서 확인한다.

04

성공 flag와 실제 완료 시점을 맞춘다

lkboot protocol → image buffer → validate command 구간에서는 부분 초기화 상태가 생길 수 있다. flag, list insertion, callback 등록, storage write 완료 중 무엇이 성공의 기준인지 찾는다. hardware write나 DMA가 포함되면 함수 반환과 장치 완료가 같은 시점인지도 확인한다.

다른 CPU, 인터럽트 처리기, USB 완료 콜백, 셸 명령에서도 이 상태를 읽는다면 메모리 접근 순서와 잠금 범위를 함께 확인한다.

05

제어권을 넘기기 전에 확인할 조건을 정리한다

정상 경로는 chain load에서 끝난다. 이 단계가 함수 반환인지, scheduler 전환인지, 다른 image로의 비복귀 분기인지 구분한다. 비복귀 handoff라면 cache clean/invalidate, interrupt disable, 장치 quiesce, argument register와 FDT 또는 image address가 최종 점검 항목이다.

호출한 함수로 돌아오는 경로라면 오류와 일부 처리만 완료된 상태를 구분하는지 확인한다. 이어서 다음 후보를 시도할지, 복구 절차를 실행할지 살펴본다.

구현을 읽을 때 놓치기 쉬운 부분

01

network 수신 buffer와 실행 image의 소유권을 나눈다

protocol layer가 packet을 조립하는 buffer와 loader가 최종 image로 보관하는 buffer는 수명이 다르다. zero-copy를 쓴다면 network callback이 buffer를 재사용하기 전에 chain loader가 ownership을 넘겨받아야 한다.

선언 size, 실제 received length와 allocation size를 각각 기록한다.

02

image parser는 load address를 신뢰하기 전에 범위를 검사한다

ELF segment나 custom header가 지정한 destination이 LK text, heap, stack, page table과 겹치지 않는지 확인한다. 여러 segment의 filesz/memsz와 BSS zero 구간도 overflow 없이 계산한다.

entry point가 실제 executable segment 안에 있는지 검증한다.

03

chain loading은 kernel을 종료하는 절차다

새 image로 분기하기 전에 network DMA, interrupt, timer와 cache를 정리하고 다른 CPU를 멈춰야 한다. 단순 function pointer call은 기존 scheduler와 stack을 남긴다.

마지막 송신 ACK와 실제 handoff 사이 reset이 발생할 때 host가 성공으로 오해하지 않도록 protocol 완료 시점을 정의한다.

객체와 수명

대상만들어지는 시점유효 범위확인할 조건
lkb_t / downloaded imagelkboot protocolchain load 또는 오류 정리 완료까지진입 전제와 상태 계약
입력 buffer / metadatanetwork/USB transportparse와 검증이 끝날 때까지길이, 정렬, 소유권, 변조 가능성
등록된 list / descriptorimage buffer하위 시스템을 종료하거나 다음 이미지에 제어권을 넘길 때까지중복 등록, 해제된 객체를 가리키는 포인터, 오류 후 자원 정리
hardware 또는 persistent state실제 write/probe가 완료된 뒤reset 또는 명시적 해제·갱신까지쓰기 완료, 시간 초과, 전원 차단, 이전 상태 복원
최종 부트로더 이미지link/image 생성 시점다음 stage가 새 image로 교체할 때까지적재 주소, 진입점, 섹션, 헤더의 일치 여부

실패 지점과 증상

#조건관찰되는 증상먼저 볼 단계
01length overflowLK memory overwritelkboot protocol
02entry 검증 누락임의 주소 분기image buffer
03device quiesce 누락다음 image IRQ 유입validate command

로그가 끊긴 함수뿐 아니라 그 함수에 주소, 크기, 플래그를 전달한 앞 단계도 조사한다. 오류 처리 후 등록된 객체나 하드웨어 설정이 그대로 남아 있는지도 확인한다.

소스 밖에서 확인할 증거

소스에서 파악한 동작을 실제 빌드 결과와 대상 보드의 로그로 확인한다. 아래 명령의 보드 이름과 툴체인 접두사는 사용 중인 빌드 환경에 맞게 바꾼다.

#목적명령 또는 계측판정 기준
01빌드 구성make <project> -j1platform, target, app 선택과 linker 입력을 로그로 남긴다.
02symbol과 주소nm -n build-*/lk.elf | grep 'static void lkboot_task'init hook, app descriptor와 함수가 어느 주소에 놓였는지 확인한다.
03어셈블리 대조objdump -drS build-*/lk.elfthread 전환과 interrupt disable 구간의 실제 명령을 확인한다.
04초기화 순서init level 전후에 current_time_hires()와 함수 이름 기록level 내부 순서와 지연을 serial log로 남긴다.
05문맥 확인current_thread, interrupt state, stack pointer 기록함수가 bootstrap thread인지 scheduler 이후 thread인지 구분한다.

직접 확인할 실험

  1. 01
    protocol packet 길이 경계 시험

    lkboot protocol 진입 전후에 protocol session의 주소·크기·반환값과 timestamp를 함께 남긴다.

  2. 02
    download buffer와 LK image range 비교

    image buffer 진입 전후에 download buffer의 주소·크기·반환값과 timestamp를 함께 남긴다.

  3. 03
    jump 직전 interrupt/cache 상태 기록

    validate command 진입 전후에 image metadata의 주소·크기·반환값과 timestamp를 함께 남긴다.

원문과 다음 글