comparison mercurial/hgweb/hgweb_mod.py @ 5963:5be210afe1b8

hgweb: explicitly check if requested command exists

5962:0011316fbe0e

        # actually process the request

        try:

            cmd = req.form.get('cmd', [''])[0]
            if hasattr(protocol, cmd):
                method = getattr(protocol, cmd)
                method(self, req)
            else:

                tmpl = self.templater(req)
                if cmd == '':
                    req.form['cmd'] = [tmpl.cache['default']]
                    cmd = req.form['cmd'][0]

                if cmd == 'file' and 'raw' in req.form.get('style', []):
                    webcommands.rawfile(self, req, tmpl)
                else:
                    getattr(webcommands, cmd)(self, req, tmpl)

                del tmpl

        except (hg.RepoError, revlog.RevlogError), inst:
            req.respond('500 Internal Server Error',
                        tmpl('error', error=str(inst)))
        except ErrorResponse, inst:
            req.respond(inst.code, tmpl('error', error=inst.message))
        except AttributeError:
            req.respond(400, tmpl('error', error='No such method: ' + cmd))

    def templater(self, req):

        # determine scheme, port and server name
        # this is needed to create absolute urls

5963:5be210afe1b8

        # actually process the request

        try:

            cmd = req.form.get('cmd', [''])[0]
            if cmd in protocol.__all__:
                method = getattr(protocol, cmd)
                method(self, req)
            else:
                tmpl = self.templater(req)
                if cmd == '':
                    req.form['cmd'] = [tmpl.cache['default']]
                    cmd = req.form['cmd'][0]

                if cmd not in webcommands.__all__:
                    raise ErrorResponse(400, 'No such method: ' + cmd)
                elif cmd == 'file' and 'raw' in req.form.get('style', []):
                    webcommands.rawfile(self, req, tmpl)
                else:
                    getattr(webcommands, cmd)(self, req, tmpl)

                del tmpl

        except (hg.RepoError, revlog.RevlogError), inst:
            req.respond('500 Internal Server Error',
                        tmpl('error', error=str(inst)))
        except ErrorResponse, inst:
            req.respond(inst.code, tmpl('error', error=inst.message))

    def templater(self, req):

        # determine scheme, port and server name
        # this is needed to create absolute urls