1// SPDX-License-Identifier: GPL-2.0-only
 2#ifndef _GNU_SOURCE
 3#define _GNU_SOURCE
 4#endif
 5#include <link.h>
 6#include <stdio.h>
 7#include <stdlib.h>
 
 8
 9struct Statistics {
10	unsigned long long load_address;
11	unsigned long long alignment;
12};
13
14int ExtractStatistics(struct dl_phdr_info *info, size_t size, void *data)
15{
16	struct Statistics *stats = (struct Statistics *) data;
17	int i;
18
19	if (info->dlpi_name != NULL && info->dlpi_name[0] != '\0') {
20		// Ignore headers from other than the executable.
21		return 2;
22	}
23
24	stats->load_address = (unsigned long long) info->dlpi_addr;
25	stats->alignment = 0;
26
27	for (i = 0; i < info->dlpi_phnum; i++) {
28		if (info->dlpi_phdr[i].p_type != PT_LOAD)
29			continue;
30
31		if (info->dlpi_phdr[i].p_align > stats->alignment)
32			stats->alignment = info->dlpi_phdr[i].p_align;
33	}
34
35	return 1;  // Terminate dl_iterate_phdr.
36}
37
38int main(int argc, char **argv)
39{
40	struct Statistics extracted;
41	unsigned long long misalign;
42	int ret;
43
 
 
 
44	ret = dl_iterate_phdr(ExtractStatistics, &extracted);
45	if (ret != 1) {
46		fprintf(stderr, "FAILED\n");
47		return 1;
48	}
49
50	if (extracted.alignment == 0) {
51		fprintf(stderr, "No alignment found\n");
52		return 1;
53	} else if (extracted.alignment & (extracted.alignment - 1)) {
54		fprintf(stderr, "Alignment is not a power of 2\n");
55		return 1;
56	}
57
58	misalign = extracted.load_address & (extracted.alignment - 1);
59	if (misalign) {
60		printf("alignment = %llu, load_address = %llu\n",
61			extracted.alignment, extracted.load_address);
62		fprintf(stderr, "FAILED\n");
63		return 1;
64	}
65
66	fprintf(stderr, "PASS\n");
67	return 0;
68}

 1// SPDX-License-Identifier: GPL-2.0-only
 2#ifndef _GNU_SOURCE
 3#define _GNU_SOURCE
 4#endif
 5#include <link.h>
 6#include <stdio.h>
 7#include <stdlib.h>
 8#include "../kselftest.h"
 9
10struct Statistics {
11	unsigned long long load_address;
12	unsigned long long alignment;
13};
14
15int ExtractStatistics(struct dl_phdr_info *info, size_t size, void *data)
16{
17	struct Statistics *stats = (struct Statistics *) data;
18	int i;
19
20	if (info->dlpi_name != NULL && info->dlpi_name[0] != '\0') {
21		// Ignore headers from other than the executable.
22		return 2;
23	}
24
25	stats->load_address = (unsigned long long) info->dlpi_addr;
26	stats->alignment = 0;
27
28	for (i = 0; i < info->dlpi_phnum; i++) {
29		if (info->dlpi_phdr[i].p_type != PT_LOAD)
30			continue;
31
32		if (info->dlpi_phdr[i].p_align > stats->alignment)
33			stats->alignment = info->dlpi_phdr[i].p_align;
34	}
35
36	return 1;  // Terminate dl_iterate_phdr.
37}
38
39int main(int argc, char **argv)
40{
41	struct Statistics extracted;
42	unsigned long long misalign;
43	int ret;
44
45	ksft_print_header();
46	ksft_set_plan(1);
47
48	ret = dl_iterate_phdr(ExtractStatistics, &extracted);
49	if (ret != 1)
50		ksft_exit_fail_msg("FAILED: dl_iterate_phdr\n");
 
 
51
52	if (extracted.alignment == 0)
53		ksft_exit_fail_msg("FAILED: No alignment found\n");
54	else if (extracted.alignment & (extracted.alignment - 1))
55		ksft_exit_fail_msg("FAILED: Alignment is not a power of 2\n");
 
 
 
56
57	misalign = extracted.load_address & (extracted.alignment - 1);
58	if (misalign)
59		ksft_exit_fail_msg("FAILED: alignment = %llu, load_address = %llu\n",
60				   extracted.alignment, extracted.load_address);
 
 
 
61
62	ksft_test_result_pass("Completed\n");
63	ksft_finished();
64}