From 4aecbfe56727c983ff8ec06076994fd61613f0fd Mon Sep 17 00:00:00 2001
From: baldurk <baldurk@baldurk.org>
Date: Mon, 9 Nov 2015 23:01:30 +0100
Subject: [PATCH] Handle whitespace in the middle of #version, etc directives.
 Refs #168

---
 renderdoc/driver/gl/gl_shader_refl.cpp | 39 ++++++++++++++++++++------
 1 file changed, 30 insertions(+), 9 deletions(-)

diff --git a/renderdoc/driver/gl/gl_shader_refl.cpp b/renderdoc/driver/gl/gl_shader_refl.cpp
index 8ff9c68c6..3e996cdf5 100644
--- a/renderdoc/driver/gl/gl_shader_refl.cpp
+++ b/renderdoc/driver/gl/gl_shader_refl.cpp
@@ -285,12 +285,36 @@ GLuint MakeSeparableShaderProgram(const GLHookSet &gl, GLenum type, vector<strin
 				string src = strings[i];
 
 				size_t len = src.length();
-				size_t it = src.find("#version");
+
+				// find if this source contains a #version, accounting for whitespace
+				size_t it = 0;
+				
+				while(it != string::npos)
+				{
+					it = src.find("#", it);
+
+					if(it == string::npos)
+						break;
+
+					// advance past the #
+					++it;
+
+					// skip whitespace
+					while(it < len && (src[it] == ' ' || src[it] == '\t'))
+						++it;
+					
+					if(it+7 < len && !strncmp(&src[it], "version", 7))
+					{
+						it += sizeof("version")-1;
+						break;
+					}
+				}
+
+				// no #version found
 				if(it == string::npos)
 					continue;
 
-				// skip #version
-				it += sizeof("#version")-1;
+				// it now points after the #version
 
 				// skip whitespace
 				while(it < len && (src[it] == ' ' || src[it] == '\t'))
@@ -308,7 +332,7 @@ GLuint MakeSeparableShaderProgram(const GLHookSet &gl, GLenum type, vector<strin
 				if(!strncmp(&src[it], "compatibility", 13)) it += sizeof("compatibility")-1;
 				if(!strncmp(&src[it], "es"           ,  2)) it += sizeof("es")-1;
 
-				// now skip past comments, and any #extension directives
+				// now skip past comments, and any #directives
 				while(it < len)
 				{
 					// skip whitespace
@@ -326,11 +350,8 @@ GLuint MakeSeparableShaderProgram(const GLHookSet &gl, GLenum type, vector<strin
 						continue;
 					}
 
-					// skip extension directives
-					const char extDirective[] = "#extension";
-					if(!strncmp(src.c_str()+it, extDirective, sizeof(extDirective)-1) &&
-						it+sizeof(extDirective)-1 < len &&
-						(src[it+sizeof(extDirective)-1] == ' ' || src[it+sizeof(extDirective)-1] == '\t'))
+					// skip preprocessor directives
+					if(src[it] == '#')
 					{
 						// keep going until the next newline
 						while(it < len && src[it] != '\r' && src[it] != '\n')