# iExploder - Generates bad HTML files to perform QA for web browsers.

# Developed for the Mozilla Foundation.

#####################

#

# Copyright (c) 2006 Thomas Stromberg <thomas%stromberg.org>

# 

# This software is provided 'as-is', without any express or implied warranty.

# In no event will the authors be held liable for any damages arising from the

# use of this software.

# 

# Permission is granted to anyone to use this software for any purpose,

# including commercial applications, and to alter it and redistribute it

# freely, subject to the following restrictions:

# 

# 1. The origin of this software must not be misrepresented; you must not

# claim that you wrote the original software. If you use this software in a

# product, an acknowledgment in the product documentation would be appreciated

# but is not required.

# 

# 2. Altered source versions must be plainly marked as such, and must not be

# misrepresented as being the original software.

# 

# 3. This notice may not be removed or altered from any source distribution.

$VERSION="1.3.2"

class IExploder

    attr_accessor :test_num, :subtest_num, :lookup_mode, :random_mode, :url

    attr_accessor :offset, :lines, :stop_num

    def initialize(max_tags, max_attrs, max_props)

        @htmlMaxTags = max_tags

        @htmlMaxAttrs = max_attrs

        @cssMaxProps = max_props

        @mangledTagTotal = 0

        @stop_num = 0

    end

    def setRandomSeed

        if @test_num > 0

            srand(@test_num)

        else

            srand

        end

    end

    def readTagFiles

        # These if statements are so that mod_ruby doesn't have to reload the files

        # each time

        if (! @cssTags)

            @cssTags = readTagFile('cssproperties.in');

        end

        if (! @htmlTags)

            @htmlTags = readTagFile('htmltags.in');

        end

        if (! @htmlAttr)

            @htmlAttr = readTagFile('htmlattrs.in');

        end

        if (! @htmlValues)

            @htmlValues = readTagFile('htmlvalues.in');

        end

        if (! @cssValues)

            @cssValues = readTagFile('cssvalues.in');

        end

    end

    def readTagFile(filename)

        list = Array.new

        File.new(filename).readlines.each { |line|

            line.chop!

            # Don't include comments.

            if (line !~ /^# /) && (line.length > 0)

                list << line

            end

        }

        return  list

    end

    # based on make_up_value, essentially.

    def inventValue

        value = rand(19);

        case value

        when 1..3 then return (@htmlValues[rand(@htmlValues.length)])

        when 4..5 then return (@htmlValues[rand(@htmlValues.length)] + inventValue())

        when 6 then return (@htmlValues[rand(@htmlValues.length)] + "//" + inventValue())

        when 7 then return ''

            # this may return negative argument?

        when 8..10 then return rand(255).chr * (rand(256)+8)

        when 11 then return rand(255).chr * (rand(2048)+8)

        when 12 then return "#" + rand(999999).to_s

        when 13 then return rand(999999).to_s + "%"

        when 14..15 then return "&" + rand(999999).to_s + ";"

            # filters

        when 16 then

            return inventValue() + "=" + inventValue()

            # this my return undefined method + for nil:NilClass

        when 17 then return inventValue() + "," + inventValue()

        else

            if rand(5) > 3

                return "-" + rand(999999).to_s

            else

                return rand(999999).to_s

            end

        end

    end

    # based on make_up_value, essentially.

    def inventCssValue(tag)

        value = rand(23);

        case value

        when 1..10 then return @cssValues[rand(@cssValues.length)]

        when 11 then return ''

        when 12 then return rand(255).chr * (rand(8192)+8)

        when 13

            length = rand(1024) + 8

            return (rand(255).chr * length) + " " + (rand(255).chr * length) + " " + (rand(255).chr * length)

        when 14 then return (rand(255).chr * (rand(1024)+3)) + "px"

        when 15 then return (rand(255).chr * (rand(1024)+3)) + "em"

        when 16 then return "url(" + inventValue() + ")"

        when 17..18 then return "#" + rand(999999999).to_s

        when 19 then return "-" + rand(99999999).to_s

        else return rand(99999999).to_s;

        end

    end

    def mangleTag(tag)

        @mangledTagTotal += 1

        out = ''

        # 20% chance of closing a tag instead of opening it. This 

        # still counts against @mangledTagTotal, however.

        if rand(10) > 8

            out = "</" + tag + ">"

            return out

        end

        # we're opening it.

        out = "<" + tag

        # forgot the space between the tag and the attributes

        if rand(15) > 1

            out << ' '

        end

        attrNum = rand(@htmlMaxAttrs) + 1

        1.upto(attrNum) {

            attr = @htmlAttr[rand(@htmlAttr.length)]

            out << attr

            # 7.5% of the time we skip the = sign. Don't prefix it

            # if the attribute ends with a ( however.

            if rand(15) > 1

                out << '='

            end

            # sometimes quote it, sometimes not. I doubt the importance

            # of this test, but mangleme-1.2 added it, and adding more

            # random-ness never hurt anything but time. I'll do it less often.

            quote = rand(2)

            if (quote > 1)

                out << "\""

            end

            out << inventValue()

            # end the quote when you are done

            if (quote > 1)

                out << "\" "

            end

            # 5% chance we skip the space at the end of the name

            if rand(20) > 1

                out << ' '

            end

        }

        # CSS styles!

        if rand(4) > 1

            out << " style=\""

            1.upto(rand(@cssMaxProps)+1) {

                out << @cssTags[rand(@cssTags.length)]

                # very small chance we let the tag run on.

                if rand(50) > 1

                    out << ": "

                end

                out << inventCssValue(tag)

                # we almost always put the ; there.

                if rand(50) > 1

                    out << '; '

                end

            }

            out << "\""

        end

        out << ">\n"

        # support our local troops!

        if (@subtest_num > 0) && filterSubTest() 

            if tag =~ /html|body|head/

                return '<' + tag + '>'

            else

                return "<x-#@mangledTagTotal>\n"

            end

        else

            return out

        end    

    end

    #end

    def filterSubTest()

        result = 1

        if (@mangledTagTotal >= @offset) && (@mangledTagTotal < (@offset + @lines))

            result = nil

        end

        return result

    end

    def nextTestNum()

        if random_mode

            n = rand(99999999)

        else

            if @test_num

                n = @test_num  + 1

            else

                n = 1

            end

        end

        return n

    end

    # If we are at line 30 with 8 extra lines, there is no point to try line 31

    # with 8 lines as well.. skip back to 1 and bump up the line count.

    def nextSubTestNum()    

        if (@offset + @lines) > @htmlMaxTags

            nextNum = ((@lines * 2 -1)) * @htmlMaxTags

        else

            nextNum = @subtest_num + 1

        end      

        return nextNum

    end

    def buildPage

        if (! @test_num) || (@test_num < 1)

            @test_num = 1

        end

        next_num=nextTestNum()

        @lines = @subtest_num.div(@htmlMaxTags) + 1

        @offset = @subtest_num.modulo(@htmlMaxTags)

        # building the HTML

        bodyText = mangleTag('html')

        bodyText << "\n<head>\n"

        # Only do redirects if lookup=1 has not been specified.

        if (! @lookup_mode) && (@lines <= @htmlMaxTags) && (@stop_num != @test_num)

            newpage = @url + "?"

            if @subtest_num > 0

                newpage << "test=" << @test_num.to_s << "&subtest=" << nextSubTestNum().to_s

            else

                newpage << "test=" << next_num.to_s

            end

            if @random_mode

                newpage << "&random=1"

            end

            if @stop_num > 0

                newpage << "&stop=" << @stop_num.to_s

            end

            bodyText << "\t<META HTTP-EQUIV=\"Refresh\" content=\"0;URL=#{newpage}\">\n" 

            # use both techniques, because you never know how you might be corrupting yourself.

            bodyText << "\t<script language=\"javascript\">setTimeout('window.location=\"#{newpage}\"', 1000);</script>\n" 

        end

        bodyText << "\t" << mangleTag('meta')

        bodyText << "\t" <<  mangleTag('meta')

        bodyText << "\t" <<  mangleTag('link')

        bodyText << "\t<title>[#@test_num] iExploder #{$VERSION} - #{inventValue()}</title>\n"

        bodyText << "</head>\n\n"

        # What tags will we be messing with ######################

        tagList = [ 'body']

        # we already have 5 tags?

        1.upto(@htmlMaxTags - 5 ) { tagList << @htmlTags[rand(@htmlTags.length)] }

        tagList.each { |tag|

            bodyText << mangleTag(tag)

            bodyText << inventValue() + "\n"

        }

        bodyText << "</body>\n</html>"

    end

end

if $0 == __FILE__

    max=ARGV[0].to_i

    puts "testing #{max} tags"

    test = IExploder.new(max, 5, 5)

    test.readTagFiles()

    test.test_num=1

    test.subtest_num=1

    counter=0

    test.lines=0

    while test.lines < max

        test.lines = test.subtest_num.div(max) + 1

        test.offset = test.subtest_num.modulo(max)

        test.subtest_num=test.nextSubTestNum

        counter = counter + 1

	puts "[#{counter}] subtest #{test.subtest_num} is #{test.lines} lines with #{test.offset} offset"

    end

    puts "for #{max} tests, you will have #{counter} iterations until #{test.subtest_num}"

end