aboutsummaryrefslogtreecommitdiff
path: root/src/archive.c
blob: 48fa55dda50a455bc4547494f8baa5add5b4bec1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#include "tar.h"

int	archive_create(int archive_fd, char **files)
{
	char		file_name[PATH_MAX];
	struct stat	root_statbuf;

	for (; *files != NULL; files++)
	{
		bzero(file_name, PATH_MAX);
		strcpy(file_name, *files);
		file_write(archive_fd, file_name);
	}
	if (stat("/", &root_statbuf) == -1)
	{
		perror("achive_write");
		return -1;
	}
	if (record_write_blank(archive_fd, 2 + (2 * root_statbuf.st_blksize) / RECORD_SIZE) == -1)
		return -1;
	return 0;
}

int			archive_extract(int archive_fd)
{
	char		record[RECORD_SIZE];
	t_header	header;
	int			fd;
	struct stat	statbuf;

	while (true)
	{
		if (record_read(archive_fd, record) == -1)
			return -1;
		if (record_is_blank(record))
		{
			if (record_read(archive_fd, record) == -1)
				return -1;
			if (record_is_blank(record))
				break;
			else
				return -1;
		}
		if (header_parse(record, &statbuf, &header) == -1)
			return -1;

		switch (header.file_type[0])
		{
			case '5':
				if (mkdir(header.file_name, statbuf.st_mode) == -1)
				{
					perror(NULL);
					return -1;
				}
				break;
			case '0':
				fd = open(header.file_name, O_WRONLY | O_CREAT | O_TRUNC, statbuf.st_mode);
				if (fd == -1)
				{
					perror(NULL);
					return -1;
				}
				char *content = malloc(statbuf.st_size);
				if (content == NULL ||
					read(archive_fd, content, statbuf.st_size) == -1 ||
					lseek(archive_fd, RECORD_SIZE - statbuf.st_size % RECORD_SIZE, SEEK_CUR) == -1 ||
					write(fd, content, statbuf.st_size) == -1)
				{
					perror(NULL);
					return -1;
				}
				break;
			default:
				return -1;
		}
	}
	return 0;
}

int	archive_get_fd(t_args *args)
{
	int		fd;
	bool	is_read;

	is_read = args->action == ACTION_DIFF ||
			  args->action == ACTION_LIST ||
			  args->action == ACTION_EXTRACT;
	if (args->archive_name == NULL)
		return is_read ? STDIN_FILENO : STDOUT_FILENO;
	if (is_read)
		fd = open(args->archive_name, O_RDONLY);
	else
		fd = open(args->archive_name, O_WRONLY | O_CREAT | O_TRUNC, DEFAULT_UMASK);
	if (fd == -1)
	{
		perror("archive fd");
		return -1;
	}
	return fd;
}

int	archive_dispatch_action(int archive_fd, t_args *args)
{
	switch (args->action)
	{
		case ACTION_CREATE:
			return archive_create(archive_fd, args->files);
		case ACTION_EXTRACT:
			return archive_extract(archive_fd);
		default:
			return -1;
	}
	return 0;
}