From a8f3cc9e254122906318531ef39b5cae89829ef4 Mon Sep 17 00:00:00 2001
From: Adam Chlipala <adamc@hcoop.net>
Date: Tue, 10 Mar 2009 11:18:01 -0400
Subject: React demo

---
 src/scriptcheck.sml | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'src/scriptcheck.sml')

diff --git a/src/scriptcheck.sml b/src/scriptcheck.sml
index bfe87766..fd4f4cd9 100644
--- a/src/scriptcheck.sml
+++ b/src/scriptcheck.sml
@@ -38,8 +38,10 @@ structure IS = IntBinarySet
 val csBasis = SS.addList (SS.empty,
                           ["new_client_source",
                            "get_client_source",
-                           "set_client_source",
-                           "alert"])
+                           "set_client_source"])
+
+val scriptWords = ["<script",
+                   " onclick="]
 
 fun classify (ds, ps) =
     let
@@ -54,7 +56,7 @@ fun classify (ds, ps) =
             let
                 fun hasClient e =
                     case #1 e of
-                        EPrim (Prim.String s) => inString {needle = "<script", haystack = s}
+                        EPrim (Prim.String s) => List.exists (fn n => inString {needle = n, haystack = s}) scriptWords
                       | EPrim _ => false
                       | ERel _ => false
                       | ENamed n => IS.member (csids, n)
-- 
cgit v1.2.3