source: pwntcha/trunk/src/linuxfr.c @ 419 Tweet

/*
 * linuxfr.c: decode linuxfr.org captchas
 * $Id: linuxfr.c 419 2005-01-04 23:10:43Z sam $
 *
 * Copyright: (c) 2005 Sam Hocevar <sam@zoy.org>
 *   This program is free software; you can redistribute it and/or
 *   modify it under the terms of the Do What The Fuck You Want To
 *   Public License as published by Banlu Kemiyatorn. See
 *   http://sam.zoy.org/projects/COPYING.WTFPL for more details.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <limits.h>

#include "config.h"
#include "common.h"

#define FONTNAME "share/font_linuxfr.png"

/* Main function */
char *decode_linuxfr(struct image *img)
{
    char all[] = "abcdefghijklmnopqrstuvwxyz"
                 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
                 "0123456789";
    char *result;
    struct image *tmp, *font;
    int x, y, r, g, b, i, j, c;
    int stats[40];

    font = image_load(FONTNAME);
    if(!font)
    {
        fprintf(stderr, "cannot load font %s\n", FONTNAME);
        exit(-1);
    }

    /* linuxfr captchas have 7 characters */
    result = malloc(8 * sizeof(char));
    memset(result, '\0', 8);

    tmp = filter_equalize(img, 150);

    for(y = 0; y < img->height; y++)
    {
        int count = 0;
        for(x = 0; x < img->width; x++)
        {
            getpixel(tmp, x, y, &r, &g, &b);
            if(r == 0)
                count++;
        }
        stats[y] = count;
    }

    /* Find 7 consecutive lines that have at least 14 pixels; they're
     * baseline candidates */
    for(y = 0; y < img->height - 11; y++)
    {
        int ycan = 1;
        for(j = 3; j < 10; j++)
        {
            if(stats[y + j] < 14)
            {
                ycan = 0;
                y = y + j - 3;
                break;
            }
        }
        if(!ycan)
            continue;

        /* Find 7 consecutive cells that have at least 2 pixels on
         * each line; they're base column candidates */
        for(x = 0; x < img->width - 9 * 7 + 1; x++)
        {
            int xcan = 1;
            for(c = 0; c < 7; c++, xcan)
            {
                for(j = 3; j < 10; j++)
                {
                    int count = 0;
                    for(i = 0; i < 8; i++)
                    {
                        getpixel(tmp, x + c * 9 + i, y + j, &r, &g, &b);
                        if(r == 0)
                        {
                            count++;
                            if(count == 2)
                                break;
                        }
                    }
                    if(count < 2)
                    {
                        xcan = 0;
                        break;
                    }
                }
            }
            if(!xcan)
                continue;

            /* Now we have an (x,y) candidate - try to fit 7 characters */
            for(c = 0; c < 7; c++, xcan)
            {
                int r2, g2, b2, test, tcan;
                for(test = 0; test < 62; test++)
                {
                    tcan = 1;
                    for(j = 0; j < 12; j++, tcan)
                        for(i = 0; i < 8; i++)
                        {
                            getpixel(tmp, x + c * 9 + i, y + j, &r, &g, &b);
                            getpixel(font, test * 9 + i, j, &r2, &g2, &b2);
                            /* Only die if font is black and image is white */
                            if(r > r2)
                            {
                                tcan = 0;
                                break;
                            }
                        }
                    if(tcan)
                    {
                        result[c] = all[test];
                        result[c+1] = '\0';
                        break;
                    }
                }
                if(!tcan)
                {
                    xcan = 0;
                    break;
                }
            }
            if(xcan)
            {
                break;
            }
        }
    }

    image_free(tmp);
    image_free(font);

    if(strlen(result) != 7)
    {
        free(result);
        return NULL;
    }

    return result;
}