검색어는 데이터로, 정렬은 허용 목록에서

목표

고객 검색과 개수 API 의 SQL 을 매개변수 질의와 정렬 허용 목록으로 고치고, 그것을 무엇이 DB 로 갔는지 보는 시험으로 고정합니다.

왜 중요한가

값을 문장에 이어 붙이면 값의 글자가 문법으로 읽힙니다. 이름에 작은따옴표가 든 고객부터 500 이 나고, 문장의 구조가 바뀌는 길이 열립니다. 위험한 글자를 바꾸는 방식은 DB 문법을 우리가 다 알아야 성립하므로, 값과 문장을 아예 섞지 않습니다.

단계

/opt/app/appsec/az-sql-lab/ 의 파일을 /root/appsec/az-sql-lab/ 로 복사하고 python3 -m pytest -q test_functional.py 로 기존 기능 시험이 통과하는지 봅니다. REQUIREMENTS.md 를 끝까지 읽습니다.
/root/appsec/az-sql-lab/test_security.py 에 요구 A(q·city 는 자리표시자로만, LIKE 의 %·_ 까지 글자 그대로)를 확인하는 시험을 test_a_ 로 시작하는 이름으로 씁니다. 지금 코드에서 실패하고, 올바르게 고친 코드에서 통과해야 합니다.
service.py 를 고쳐 요구 A 를 지킵니다. 채점기는 숨은 요구 A 시험과 여러분의 test_a_ 시험을 함께 돌립니다.
요구 B(sort 는 name·-name·created_at·-created_at 가운데 하나, 나머지는 422)를 확인하는 시험을 test_b_ 로 시작하는 이름으로 더합니다. 판정은 2단계와 같습니다.
service.py 를 고쳐 요구 B 도 지킵니다.
test_functional.py 가 그대로 통과하는지 봅니다. 채점기는 숨은 기능 시험도 함께 돌립니다.
REQUIREMENTS.md 의 '감사 로그' 표대로 audit(...) 을 부릅니다(검색은 customers.searched, 정렬 거절은 input.rejected, 검색어는 남기지 않음).
채점기가 그럴듯한 오답을 하나씩 끼워 여러분의 test_security.py 를 돌립니다. 오답마다 시험이 하나 이상 실패해야 합니다.

참고

두 함수는 conn.execute 만 씁니다. REQUIREMENTS.md 의 Recorder 를 시험에 그대로 옮겨 쓰면 SQL 과 값을 따로 볼 수 있습니다.
흔한 실수: 검색만 고치고 개수 함수를 남기는 것, 작은따옴표를 두 번 쓰는 것, LIKE 의 % 를 그대로 두는 것.
시험은 cd /root/appsec/az-sql-lab && python3 -m pytest -q test_security.py 로 돌립니다. -k test_a_ 를 붙이면 요구 A 시험만 돕니다.
정답 보기의 python3 /opt/app/appsec/apply_fix.py service.py 는 조각에 든 함수만 이름으로 찾아 바꿉니다. 직접 고친 다른 함수는 그대로 남습니다.
실습 파드에는 볼륨이 없어 세션이 끝나면 /root/appsec 가 사라집니다. 남기고 싶은 시험은 세션 안에서 옮겨 두세요.

재료를 복사하고 기능 시험을 돌린다

/opt/app/appsec/az-sql-lab/ 의 파일을 모두 /root/appsec/az-sql-lab/ 로 복사하고, 그 디렉터리에서 python3 -m pytest -q test_functional.py 로 기존 기능 시험이 통과하는 것을 본다. REQUIREMENTS.md 를 끝까지 읽는다.

cp 로 디렉터리의 파일을 통째로 옮깁니다. 고치기 전에 지금 무엇이 되는지부터 확인해 두면, 나중에 무엇을 깨뜨렸는지 가릴 수 있습니다.

요구 A 를 시험으로 쓴다

/root/appsec/az-sql-lab/test_security.py 에 요구 A(q·city 는 자리표시자로만, LIKE 의 %·_ 까지 글자 그대로)를 확인하는 시험을 test_a_ 로 시작하는 이름으로 쓴다. 이 시험은 지금 코드에서 실패하고, 올바르게 고친 코드에서 통과해야 한다.

Recorder 로 search_customers 와 count_by_city 를 불러 execute 에 넘어간 SQL 문자열에 검색어나 도시 이름이 들어 있는지 확인하세요. 작은따옴표가 든 이름과 검색어 % 의 결과도 좋은 시험입니다.

요구 A 를 지키도록 고친다

/root/appsec/az-sql-lab/service.py 를 고쳐 요구 A 를 지킨다. 채점기는 숨은 요구 A 시험과 여러분의 test_a_ 시험을 고친 코드에 돌린다.

문장에는 ? 를 두고 값은 execute 의 두 번째 인자(튜플)로 넘깁니다. LIKE 는 like ? escape '\' 로 쓰고, 검색어의 역슬래시를 먼저, 그다음 % 와 _ 를 이스케이프합니다.

요구 B 를 시험으로 쓴다

/root/appsec/az-sql-lab/test_security.py 에 요구 B(sort 는 name·-name·created_at·-created_at 가운데 하나, 나머지는 422)를 확인하는 시험을 test_b_ 로 시작하는 이름으로 더한다. 02 단계처럼 지금 코드에서 실패하고 올바른 코드에서 통과해야 한다.

city, name desc, id 처럼 목록에 없는 정렬값이 422 인지, -name 이 정말 이름 내림차순인지 확인하세요.

요구 B 를 지키도록 고친다

/root/appsec/az-sql-lab/service.py 를 고쳐 요구 B 도 지킨다. 채점기는 숨은 요구 B 시험과 여러분의 test_b_ 시험을 고친 코드에 돌린다.

정렬값 → SQL 조각 사전(ORDER_BY)을 두고, 사전에 없으면 ValueError 를 냅니다(경로가 422 로 바꿉니다). 같은 값끼리는 id asc 로 순서를 고정합니다.

기존 기능이 그대로인지 본다

고친 service.py 로 python3 -m pytest -q test_functional.py 가 그대로 통과하는지 본다. 채점기는 재료의 원본 기능 시험에 숨은 기능 시험을 더해 돌린다.

보안 수정이 정상 사용자를 막으면 그 수정은 되돌려집니다. 실패한 기능 시험의 이름이 어떤 정상 동작을 막았는지 알려 줍니다. 막을 것만 막았는지 요구사항의 '그대로 된다' 문장과 견주어 보세요.

감사 로그를 남긴다

REQUIREMENTS.md 의 '감사 로그' 표대로 audit(...) 을 부른다(검색은 customers.searched, 정렬 거절은 input.rejected, 검색어는 남기지 않음). 남기지 말라고 한 값은 어떤 항목에도 넣지 않는다.

검색이 끝난 뒤 audit('customers.searched', sort=..., results=...) 를, 정렬을 거절할 때 audit('input.rejected', action=..., field='sort') 를 부릅니다. 검색어는 어디에도 넣지 않습니다.

그럴듯한 오답을 시험이 붙잡는지 본다

채점기가 기준 코드에 그럴듯한 오답을 하나씩 끼워 /root/appsec/az-sql-lab/test_security.py 를 돌린다. 오답마다 시험이 하나 이상 실패하도록 필요한 시험을 더한다.

채점 메시지가 어떤 오답이 빠져나갔는지 알려 줍니다. 한 경로만 고친 수정이면 다른 경로를, 비교 방법이 틀린 수정이면 그 비교를 확인하는 시험을 더하세요. 요구사항의 문장 하나하나가 시험 하나가 될 수 있습니다.