1 /*
2  * Copyright © 2010 Intel Corporation
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a
5  * copy of this software and associated documentation files (the "Software"),
6  * to deal in the Software without restriction, including without limitation
7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8  * and/or sell copies of the Software, and to permit persons to whom the
9  * Software is furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice (including the next
12  * paragraph) shall be included in all copies or substantial portions of the
13  * Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21  * DEALINGS IN THE SOFTWARE.
22  */
23 
24 #include <stdio.h>
25 #include <string.h>
26 #include <errno.h>
27 #include <getopt.h>
28 
29 #include "glcpp.h"
30 #include "main/mtypes.h"
31 #include "main/shaderobj.h"
32 #include "util/strtod.h"
33 
34 extern int glcpp_parser_debug;
35 
36 void
_mesa_reference_shader(struct gl_context * ctx,struct gl_shader ** ptr,struct gl_shader * sh)37 _mesa_reference_shader(struct gl_context *ctx, struct gl_shader **ptr,
38                        struct gl_shader *sh)
39 {
40    (void) ctx;
41    *ptr = sh;
42 }
43 
44 /* Read from fp until EOF and return a string of everything read.
45  */
46 static char *
load_text_fp(void * ctx,FILE * fp)47 load_text_fp (void *ctx, FILE *fp)
48 {
49 #define CHUNK 4096
50 	char *text = NULL;
51 	size_t text_size = 0;
52 	size_t total_read = 0;
53 	size_t bytes;
54 
55 	while (1) {
56 		if (total_read + CHUNK + 1 > text_size) {
57 			text_size = text_size ? text_size * 2 : CHUNK + 1;
58 			text = reralloc_size (ctx, text, text_size);
59 			if (text == NULL) {
60 				fprintf (stderr, "Out of memory\n");
61 				return NULL;
62 			}
63 		}
64 		bytes = fread (text + total_read, 1, CHUNK, fp);
65 		total_read += bytes;
66 
67 		if (bytes < CHUNK) {
68 			break;
69 		}
70 	}
71 
72 	text[total_read] = '\0';
73 
74 	return text;
75 }
76 
77 static char *
load_text_file(void * ctx,const char * filename)78 load_text_file(void *ctx, const char *filename)
79 {
80 	char *text;
81 	FILE *fp;
82 
83 	if (filename == NULL || strcmp (filename, "-") == 0)
84 		return load_text_fp (ctx, stdin);
85 
86 	fp = fopen (filename, "r");
87 	if (fp == NULL) {
88 		fprintf (stderr, "Failed to open file %s: %s\n",
89 			 filename, strerror (errno));
90 		return NULL;
91 	}
92 
93 	text = load_text_fp (ctx, fp);
94 
95 	fclose(fp);
96 
97 	return text;
98 }
99 
100 /* Initialize only those things that glcpp cares about.
101  */
102 static void
init_fake_gl_context(struct gl_context * gl_ctx)103 init_fake_gl_context (struct gl_context *gl_ctx)
104 {
105 	memset(gl_ctx, 0, sizeof(*gl_ctx));
106 	gl_ctx->API = API_OPENGL_COMPAT;
107 	gl_ctx->Const.DisableGLSLLineContinuations = false;
108 }
109 
110 static void
usage(void)111 usage (void)
112 {
113 	fprintf (stderr,
114 		 "Usage: glcpp [OPTIONS] [--] [<filename>]\n"
115 		 "\n"
116 		 "Pre-process the given filename (stdin if no filename given).\n"
117 		 "The following options are supported:\n"
118 		 "    --disable-line-continuations      Do not interpret lines ending with a\n"
119 		 "                                      backslash ('\\') as a line continuation.\n");
120 }
121 
122 enum {
123 	DISABLE_LINE_CONTINUATIONS_OPT = CHAR_MAX + 1
124 };
125 
126 static const struct option
127 long_options[] = {
128 	{"disable-line-continuations", no_argument, 0, DISABLE_LINE_CONTINUATIONS_OPT },
129         {"debug",                      no_argument, 0, 'd'},
130 	{0,                            0,           0, 0 }
131 };
132 
133 int
main(int argc,char * argv[])134 main (int argc, char *argv[])
135 {
136 	char *filename = NULL;
137 	void *ctx = ralloc(NULL, void*);
138 	char *info_log = ralloc_strdup(ctx, "");
139 	const char *shader;
140 	int ret;
141 	struct gl_context gl_ctx;
142 	int c;
143 
144 	init_fake_gl_context (&gl_ctx);
145 
146 	while ((c = getopt_long(argc, argv, "d", long_options, NULL)) != -1) {
147 		switch (c) {
148 		case DISABLE_LINE_CONTINUATIONS_OPT:
149 			gl_ctx.Const.DisableGLSLLineContinuations = true;
150 			break;
151                 case 'd':
152 			glcpp_parser_debug = 1;
153 			break;
154 		default:
155 			usage ();
156 			exit (1);
157 		}
158 	}
159 
160 	if (optind + 1 < argc) {
161 		printf ("Unexpected argument: %s\n", argv[optind+1]);
162 		usage ();
163 		exit (1);
164 	}
165 	if (optind < argc) {
166 		filename = argv[optind];
167 	}
168 
169 	shader = load_text_file (ctx, filename);
170 	if (shader == NULL)
171 	   return 1;
172 
173 	_mesa_locale_init();
174 
175 	ret = glcpp_preprocess(ctx, &shader, &info_log, NULL, NULL, &gl_ctx);
176 
177 	printf("%s", shader);
178 	fprintf(stderr, "%s", info_log);
179 
180 	ralloc_free(ctx);
181 
182 	return ret;
183 }
184