File: core_fileio.c

package info (click to toggle)
shntool 3.0.10-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye, buster, sid, stretch
  • size: 1,220 kB
  • ctags: 1,118
  • sloc: ansic: 8,515; sh: 3,761; makefile: 36
file content (145 lines) | stat: -rw-r--r-- 4,646 bytes parent folder | download
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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
/*  core_fileio.c - file I/O functions
 *  Copyright (C) 2000-2009  Jason Jordan <shnutils@freeshell.org>
 *
 *  This program is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU General Public License
 *  as published by the Free Software Foundation; either version 2
 *  of the License, or (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

#include <string.h>
#include "shntool.h"

CVSID("$Id: core_fileio.c,v 1.44 2009/03/11 17:18:01 jason Exp $")

int read_n_bytes(FILE *in,unsigned char *buf,int num,progress_info *proginfo)
/* reads the specified number of bytes from the file descriptor 'in' into buf */
{
  int read;

  if ((read = fread(buf,1,num,in)) != num) {
    st_debug1("tried to read %d bytes, but only read %d -- possible truncated/corrupt file",num,read);
  }

  if (proginfo) {
    proginfo->bytes_written += read;
    prog_update(proginfo);
  }

  return read;
}

int write_n_bytes(FILE *out,unsigned char *buf,int num,progress_info *proginfo)
/* writes the specified number of bytes from buf into the file descriptor 'out' */
{
  int wrote;

  if ((wrote = fwrite(buf,1,num,out)) != num) {
    st_debug1("tried to write %d bytes, but only wrote %d -- make sure that:\n"
               "+ there is enough disk space\n"
               "+ the specified output directory exists\n"
               "+ you have permission to create files in the output directory\n"
               "+ the output format's encoder is installed and in your PATH",num,wrote);
  }

  if (proginfo) {
    proginfo->bytes_written += wrote;
    prog_update(proginfo);
  }

  return wrote;
}

unsigned long transfer_n_bytes_internal(FILE *in,FILE *out1,FILE *out2,unsigned long bytes,progress_info *proginfo)
/* transfers 'bytes' bytes from file descriptor 'in' to file descriptor 'out' */
{
  unsigned char buf[XFER_SIZE];
  int bytes_to_xfer,
      actual_bytes_read,
      actual_bytes_written1,
      actual_bytes_written2;
  unsigned long total_bytes_to_xfer = bytes,
                total_bytes_xfered = 0;

  while (total_bytes_to_xfer > 0) {
    bytes_to_xfer = min(total_bytes_to_xfer,XFER_SIZE);
    actual_bytes_read = read_n_bytes(in,buf,bytes_to_xfer,NULL);
    actual_bytes_written1 = write_n_bytes(out1,buf,actual_bytes_read,proginfo);
    actual_bytes_written2 = (out2) ? write_n_bytes(out2,buf,actual_bytes_read,NULL) : 0;
    total_bytes_xfered += (unsigned long)actual_bytes_written1;
    if (actual_bytes_read != bytes_to_xfer || actual_bytes_written1 != bytes_to_xfer || (out2 && actual_bytes_written2 != bytes_to_xfer))
      break;
    total_bytes_to_xfer -= bytes_to_xfer;
  }

  return total_bytes_xfered;
}

int write_padding(FILE *out,int bytes,progress_info *proginfo)
/* writes the specified number of zero bytes to the file descriptor given */
{
  unsigned char silence[CD_BLOCK_SIZE];

  if (bytes >= CD_BLOCK_SIZE) {
    /* padding should always be less than the size of the block being padded */
    return 0;
  }

  memset((void *)silence,0,CD_BLOCK_SIZE);

  return write_n_bytes(out,silence,bytes,proginfo);
}

bool read_value_long(FILE *file,unsigned long *be_val,unsigned long *le_val,unsigned char *tag_val)
/* reads an unsigned long in big- and/or little-endian format from a file descriptor */
{
  unsigned char buf[5];

  if (fread(buf, 1, 4, file) != 4)
    return FALSE;

  buf[4] = 0;

  if (be_val)
    *be_val = (((unsigned long)buf[0]) << 24)
              | (((unsigned long)buf[1]) << 16)
              | (((unsigned long)buf[2]) << 8)
              | ((unsigned long)buf[3]);

  if (le_val)
    *le_val = (((unsigned long)buf[3]) << 24)
              | (((unsigned long)buf[2]) << 16)
              | (((unsigned long)buf[1]) << 8)
              | ((unsigned long)buf[0]);

  if (tag_val)
    tagcpy(tag_val,buf);

  return TRUE;
}

bool read_value_short(FILE *file,unsigned short *be_val,unsigned short *le_val)
/* reads an unsigned short in big- and/or little-endian format from a file descriptor */
{
  unsigned char buf[2];

  if (fread(buf, 1, 2, file) != 2)
    return FALSE;

  if (be_val)
    *be_val = (buf[0] << 8) | buf[1];

  if (le_val)
    *le_val = (buf[1] << 8) | buf[0];

  return TRUE;
}