/
assemble.c
83 lines (72 loc) · 2.03 KB
/
assemble.c
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
/**
* MojoShader; generate shader programs from bytecode of compiled
* Direct3D shaders.
*
* Please see the file LICENSE.txt in the source's root directory.
*
* This file written by Ryan C. Gordon.
*/
#include <stdio.h>
#include <stdlib.h>
#include "mojoshader.h"
static int assemble(const char *buf, const char *outfile)
{
FILE *io = fopen(outfile, "wb");
if (io == NULL)
{
printf(" ... fopen('%s') failed.\n", outfile);
return 0;
} // if
const MOJOSHADER_parseData *pd;
int retval = 0;
pd = MOJOSHADER_assemble(buf, NULL, 0, NULL, 0, NULL, NULL, NULL);
if (pd->error != NULL)
printf("ERROR: (line %d) %s\n", pd->error_position, pd->error);
else
{
if (pd->output != NULL)
{
if (fwrite(pd->output, pd->output_len, 1, io) != 1)
printf(" ... fwrite('%s') failed.\n", outfile);
else if (fclose(io) == EOF)
printf(" ... fclose('%s') failed.\n", outfile);
else
retval = 1;
} // if
} // else
MOJOSHADER_freeParseData(pd);
return retval;
} // assemble
int main(int argc, char **argv)
{
int retval = 1;
if (argc != 3)
printf("\n\nUSAGE: %s <d3dasmfile> <outputfile>\n\n", argv[0]);
else
{
const char *infile = argv[1];
const char *outfile = argv[2];
FILE *io = fopen(infile, "rb");
if (io == NULL)
printf(" ... fopen('%s') failed.\n", infile);
else
{
char *buf = (char *) malloc(1000000);
int rc = fread(buf, 1, 1000000-1, io);
fclose(io);
if (rc == EOF)
printf(" ... fread('%s') failed.\n", infile);
else
{
buf[rc] = '\0';
if (assemble(buf, outfile))
retval = 0;
else
remove(outfile);
free(buf);
} // else
} // for
} // else
return retval;
} // main
// end of assemble.c ...