linux-master/lib/fortify_kunit.c

405 	struct fortify_padding pad = { };
406 	int i, end = sizeof(pad.buf) - 1;
409 	for (i = 0; i < sizeof(pad.buf) - 1; i++)
410 		pad.buf[i] = i + '0';
412 	KUNIT_EXPECT_EQ(test, pad.buf[end], '\0');
413 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
416 	KUNIT_EXPECT_EQ(test, strlen(pad.buf), end);
420 	pad.buf[end] = 'A';
421 	end = sizeof(pad.buf);
422 	KUNIT_EXPECT_EQ(test, strlen(pad.buf), end);
428 	struct fortify_padding pad = { };
429 	int i, end = sizeof(pad.buf) - 1;
432 	for (i = 0; i < sizeof(pad.buf) - 1; i++)
433 		pad.buf[i] = i + '0';
435 	KUNIT_EXPECT_EQ(test, pad.buf[end], '\0');
436 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
439 	KUNIT_EXPECT_EQ(test, strnlen(pad.buf, sizeof(pad.buf)), end);
442 	KUNIT_EXPECT_EQ(test, strnlen(pad.buf, sizeof(pad.buf) / 2),
443 					sizeof(pad.buf) / 2);
447 	pad.buf[end] = 'A';
448 	end = sizeof(pad.buf);
450 	KUNIT_EXPECT_EQ(test, strnlen(pad.buf, end + 1), end);
452 	KUNIT_EXPECT_EQ(test, strnlen(pad.buf, end + 2), end);
456 	KUNIT_EXPECT_EQ(test, strnlen(pad.buf, end), end);
459 	end = sizeof(pad.buf) / 2;
460 	KUNIT_EXPECT_EQ(test, strnlen(pad.buf, end), end);
466 	struct fortify_padding pad = { };
467 	char src[sizeof(pad.buf) + 1] = { };
475 	KUNIT_EXPECT_EQ(test, pad.bytes_before, 0);
476 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
477 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
478 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 3], '\0');
479 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
482 	KUNIT_ASSERT_TRUE(test, strcpy(pad.buf, src)
483 				== pad.buf);
487 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
488 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
489 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
493 	KUNIT_ASSERT_TRUE(test, strcpy(pad.buf, src)
494 				== pad.buf);
498 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
499 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
500 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
502 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
506 	KUNIT_ASSERT_TRUE(test, strcpy(pad.buf, src)
507 				== pad.buf);
515 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
516 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
517 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
519 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
524 	struct fortify_padding pad = { };
528 	KUNIT_EXPECT_EQ(test, pad.bytes_before, 0);
529 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
530 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
531 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 3], '\0');
532 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
535 	KUNIT_ASSERT_TRUE(test, strncpy(pad.buf, src,
536 					sizeof(pad.buf) + unconst - 1)
537 				== pad.buf);
540 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
541 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
542 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
545 	KUNIT_ASSERT_TRUE(test, strncpy(pad.buf, src,
546 					sizeof(pad.buf) + unconst)
547 				== pad.buf);
550 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 1], '\0');
551 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
552 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
554 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
557 	KUNIT_ASSERT_TRUE(test, strncpy(pad.buf, src,
558 					sizeof(pad.buf) + unconst + 1)
559 				== pad.buf);
562 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 1], '\0');
563 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
564 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
566 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
569 	KUNIT_ASSERT_TRUE(test, strncpy(pad.buf, src,
570 					sizeof(pad.buf) + unconst + 2)
571 				== pad.buf);
574 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 1], '\0');
575 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
576 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
578 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
583 	struct fortify_padding pad = { };
587 	KUNIT_EXPECT_EQ(test, pad.bytes_before, 0);
588 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
589 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
590 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 3], '\0');
591 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
594 	KUNIT_ASSERT_EQ(test, strscpy(pad.buf, src,
595 				      sizeof(pad.buf) + unconst - 1),
599 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
600 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
601 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
604 	KUNIT_ASSERT_EQ(test, strscpy(pad.buf, src,
605 				      sizeof(pad.buf) + unconst),
609 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
610 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
611 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
614 	KUNIT_ASSERT_EQ(test, strscpy(pad.buf, src,
615 				      sizeof(pad.buf) + unconst + 1),
619 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
620 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
621 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
623 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
626 	KUNIT_ASSERT_EQ(test, strscpy(pad.buf, src,
631 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
632 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
633 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
635 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
640 	struct fortify_padding pad = { };
641 	char src[sizeof(pad.buf) / 2] = { };
651 	KUNIT_EXPECT_EQ(test, pad.bytes_before, 0);
652 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
653 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
654 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 3], '\0');
655 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
658 	KUNIT_ASSERT_TRUE(test, strcat(pad.buf, src) == pad.buf);
661 	KUNIT_ASSERT_TRUE(test, strcat(pad.buf, src) == pad.buf);
664 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
665 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
666 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
669 	KUNIT_ASSERT_TRUE(test, strcat(pad.buf, one) == pad.buf);
672 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
673 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
674 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
677 	KUNIT_ASSERT_TRUE(test, strcat(pad.buf, one) == pad.buf);
680 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
681 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
682 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
683 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
686 	KUNIT_ASSERT_TRUE(test, strcat(pad.buf, two) == pad.buf);
689 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
690 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
691 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
692 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
697 	struct fortify_padding pad = { };
698 	char src[sizeof(pad.buf)] = { };
707 	KUNIT_EXPECT_EQ(test, pad.bytes_before, 0);
708 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
709 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
710 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 3], '\0');
711 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
714 	KUNIT_ASSERT_TRUE(test, strncat(pad.buf, src, partial) == pad.buf);
718 	KUNIT_ASSERT_TRUE(test, strncat(pad.buf, src, partial) == pad.buf);
722 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
723 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
724 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
727 	KUNIT_ASSERT_TRUE(test, strncat(pad.buf, src, 1) == pad.buf);
731 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
732 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
733 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
736 	KUNIT_ASSERT_TRUE(test, strncat(pad.buf, src, 1) == pad.buf);
740 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
741 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
742 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
743 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
746 	KUNIT_ASSERT_TRUE(test, strncat(pad.buf, src, 2) == pad.buf);
750 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
751 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
752 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
753 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
756 	pad.buf[sizeof(pad.buf) - 1] = 'A';
757 	KUNIT_ASSERT_TRUE(test, strncat(pad.buf, src, 1) == pad.buf);
761 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 1], '\0');
762 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
763 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
765 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
770 	struct fortify_padding pad = { };
771 	char src[sizeof(pad.buf)] = { };
773 	int len = sizeof(pad.buf) + unconst;
781 	KUNIT_EXPECT_EQ(test, pad.bytes_before, 0);
782 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
783 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
784 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 3], '\0');
785 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
788 	KUNIT_ASSERT_EQ(test, strlcat(pad.buf, src, len), partial);
792 	KUNIT_ASSERT_EQ(test, strlcat(pad.buf, src, len), partial * 2);
796 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
797 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 2], '\0');
798 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
801 	KUNIT_ASSERT_EQ(test, strlcat(pad.buf, "Q", len), partial * 2 + 1);
805 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
806 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
807 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
810 	KUNIT_ASSERT_EQ(test, strlcat(pad.buf, "V", len * 2), len);
814 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
815 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
816 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
817 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
820 	KUNIT_ASSERT_EQ(test, strlcat(pad.buf, "QQ", len * 2), len + 1);
824 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
825 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
826 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
827 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
830 	pad.buf[sizeof(pad.buf) - 1] = 'A';
831 	KUNIT_ASSERT_EQ(test, strlcat(pad.buf, "TT", len * 2), len + 2);
835 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 1], '\0');
836 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 2], '\0');
837 	KUNIT_EXPECT_NE(test, pad.buf[sizeof(pad.buf) - 3], '\0');
839 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
843 	pad.buf[sizeof(pad.buf) - 1] = '\0';
844 	KUNIT_ASSERT_EQ(test, strlcat(pad.buf, src, len * 3), len - 1 + sizeof(src));
848 	KUNIT_EXPECT_EQ(test, pad.buf[sizeof(pad.buf) - 1], '\0');
850 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);
864 	struct fortify_padding pad = { };			\
865 	char srcA[sizeof(pad.buf) + 2];				\
866 	char srcB[sizeof(pad.buf) + 2];				\
867 	size_t len = sizeof(pad.buf) + unconst;			\
874 	memfunc(pad.buf, srcA, 0 + unconst);			\
875 	KUNIT_EXPECT_EQ(test, pad.buf[0], '\0');		\
878 	memfunc(pad.buf + 1, srcB, 1 + unconst);		\
879 	KUNIT_EXPECT_EQ(test, pad.buf[0], '\0');		\
880 	KUNIT_EXPECT_EQ(test, pad.buf[1], 'B');			\
881 	KUNIT_EXPECT_EQ(test, pad.buf[2], '\0');		\
884 	memfunc(pad.buf, srcA, 1 + unconst);			\
885 	KUNIT_EXPECT_EQ(test, pad.buf[0], 'A');			\
886 	KUNIT_EXPECT_EQ(test, pad.buf[1], 'B');			\
889 	memfunc(pad.buf, srcA, len - 1);			\
890 	KUNIT_EXPECT_EQ(test, pad.buf[1], 'A');			\
891 	KUNIT_EXPECT_EQ(test, pad.buf[len - 1], '\0');		\
894 	memfunc(pad.buf, srcA, len);				\
895 	KUNIT_EXPECT_EQ(test, pad.buf[1], 'A');			\
896 	KUNIT_EXPECT_EQ(test, pad.buf[len - 1], 'A');		\
897 	KUNIT_EXPECT_EQ(test, pad.bytes_after, 0);		\
900 	memfunc(pad.buf, srcA, len + 1);			\
903 	memfunc(pad.buf + 1, srcB, len);			\