mirror of https://github.com/postgres/postgres
Vadim B. Mikheev
28 years ago
parent
9db2992640
commit
7bff4c5078
@ -0,0 +1,205 @@ |
|||||||
|
/*-------------------------------------------------------------------------
|
||||||
|
* |
||||||
|
* proclang.c-- |
||||||
|
* PostgreSQL PROCEDURAL LANGUAGE support code. |
||||||
|
* |
||||||
|
*------------------------------------------------------------------------- |
||||||
|
*/ |
||||||
|
#include <ctype.h> |
||||||
|
#include <string.h> |
||||||
|
#include "postgres.h" |
||||||
|
|
||||||
|
#include "access/heapam.h" |
||||||
|
#include "catalog/catname.h" |
||||||
|
#include "catalog/pg_user.h" |
||||||
|
#include "catalog/pg_proc.h" |
||||||
|
#include "catalog/pg_language.h" |
||||||
|
#include "utils/syscache.h" |
||||||
|
#include "commands/proclang.h" |
||||||
|
#include "fmgr.h" |
||||||
|
|
||||||
|
|
||||||
|
static void |
||||||
|
case_translate_language_name(const char *input, char *output) |
||||||
|
{ |
||||||
|
/*-------------------------------------------------------------------------
|
||||||
|
Translate the input language name to lower case, except if it's C, |
||||||
|
translate to upper case. |
||||||
|
--------------------------------------------------------------------------*/ |
||||||
|
int i; |
||||||
|
|
||||||
|
for (i = 0; i < NAMEDATALEN && input[i] != '\0'; ++i) |
||||||
|
output[i] = tolower(input[i]); |
||||||
|
|
||||||
|
output[i] = '\0'; |
||||||
|
|
||||||
|
if (strcmp(output, "c") == 0) |
||||||
|
output[0] = 'C'; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/* ---------------------------------------------------------------------
|
||||||
|
* CREATE PROCEDURAL LANGUAGE |
||||||
|
* --------------------------------------------------------------------- |
||||||
|
*/ |
||||||
|
void |
||||||
|
CreateProceduralLanguage(CreatePLangStmt * stmt) |
||||||
|
{ |
||||||
|
char languageName[NAMEDATALEN]; |
||||||
|
HeapTuple langTup; |
||||||
|
HeapTuple procTup; |
||||||
|
|
||||||
|
Oid typev[8]; |
||||||
|
char nulls[Natts_pg_language]; |
||||||
|
Datum values[Natts_pg_language]; |
||||||
|
Relation rdesc; |
||||||
|
HeapTuple tup; |
||||||
|
TupleDesc tupDesc; |
||||||
|
|
||||||
|
int i; |
||||||
|
|
||||||
|
/* ----------------
|
||||||
|
* Check permission |
||||||
|
* ---------------- |
||||||
|
*/ |
||||||
|
if (!superuser()) |
||||||
|
{ |
||||||
|
elog(WARN, "Only users with Postgres superuser privilege are " |
||||||
|
"permitted to create procedural languages"); |
||||||
|
} |
||||||
|
|
||||||
|
/* ----------------
|
||||||
|
* Translate the language name and check that |
||||||
|
* this language doesn't already exist |
||||||
|
* ---------------- |
||||||
|
*/ |
||||||
|
case_translate_language_name(stmt->plname, languageName); |
||||||
|
|
||||||
|
langTup = SearchSysCacheTuple(LANNAME, |
||||||
|
PointerGetDatum(languageName), |
||||||
|
0, 0, 0); |
||||||
|
if (HeapTupleIsValid(langTup)) |
||||||
|
{ |
||||||
|
elog(WARN, "Language %s already exists", languageName); |
||||||
|
} |
||||||
|
|
||||||
|
/* ----------------
|
||||||
|
* Lookup the PL handler function and check that it is |
||||||
|
* of return type Opaque |
||||||
|
* ---------------- |
||||||
|
*/ |
||||||
|
memset(typev, 0, sizeof(typev)); |
||||||
|
procTup = SearchSysCacheTuple(PRONAME, |
||||||
|
PointerGetDatum(stmt->plhandler), |
||||||
|
UInt16GetDatum(0), |
||||||
|
PointerGetDatum(typev), |
||||||
|
0); |
||||||
|
if (!HeapTupleIsValid(procTup)) |
||||||
|
{ |
||||||
|
elog(WARN, "PL handler function %s() doesn't exist", |
||||||
|
stmt->plhandler); |
||||||
|
} |
||||||
|
if (((Form_pg_proc) GETSTRUCT(procTup))->prorettype != InvalidOid) |
||||||
|
{ |
||||||
|
elog(WARN, "PL handler function %s() isn't of return type Opaque", |
||||||
|
stmt->plhandler); |
||||||
|
} |
||||||
|
|
||||||
|
/* ----------------
|
||||||
|
* Insert the new language into pg_language |
||||||
|
* ---------------- |
||||||
|
*/ |
||||||
|
for (i = 0; i < Natts_pg_language; i++) |
||||||
|
{ |
||||||
|
nulls[i] = ' '; |
||||||
|
values[i] = (Datum) NULL; |
||||||
|
} |
||||||
|
|
||||||
|
i = 0; |
||||||
|
values[i++] = PointerGetDatum(languageName); |
||||||
|
values[i++] = Int8GetDatum((bool) 1); |
||||||
|
values[i++] = Int8GetDatum(stmt->pltrusted); |
||||||
|
values[i++] = ObjectIdGetDatum(procTup->t_oid); |
||||||
|
values[i++] = (Datum) fmgr(TextInRegProcedure, stmt->plcompiler); |
||||||
|
|
||||||
|
rdesc = heap_openr(LanguageRelationName); |
||||||
|
|
||||||
|
tupDesc = rdesc->rd_att; |
||||||
|
tup = heap_formtuple(tupDesc, values, nulls); |
||||||
|
|
||||||
|
heap_insert(rdesc, tup); |
||||||
|
|
||||||
|
heap_close(rdesc); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/* ---------------------------------------------------------------------
|
||||||
|
* DROP PROCEDURAL LANGUAGE |
||||||
|
* --------------------------------------------------------------------- |
||||||
|
*/ |
||||||
|
void |
||||||
|
DropProceduralLanguage(DropPLangStmt * stmt) |
||||||
|
{ |
||||||
|
char languageName[NAMEDATALEN]; |
||||||
|
HeapTuple langTup; |
||||||
|
|
||||||
|
Relation rdesc; |
||||||
|
HeapScanDesc scanDesc; |
||||||
|
ScanKeyData scanKeyData; |
||||||
|
HeapTuple tup; |
||||||
|
|
||||||
|
/* ----------------
|
||||||
|
* Check permission |
||||||
|
* ---------------- |
||||||
|
*/ |
||||||
|
if (!superuser()) |
||||||
|
{ |
||||||
|
elog(WARN, "Only users with Postgres superuser privilege are " |
||||||
|
"permitted to drop procedural languages"); |
||||||
|
} |
||||||
|
|
||||||
|
/* ----------------
|
||||||
|
* Translate the language name, check that |
||||||
|
* this language exist and is a PL |
||||||
|
* ---------------- |
||||||
|
*/ |
||||||
|
case_translate_language_name(stmt->plname, languageName); |
||||||
|
|
||||||
|
langTup = SearchSysCacheTuple(LANNAME, |
||||||
|
PointerGetDatum(languageName), |
||||||
|
0, 0, 0); |
||||||
|
if (!HeapTupleIsValid(langTup)) |
||||||
|
{ |
||||||
|
elog(WARN, "Language %s doesn't exist", languageName); |
||||||
|
} |
||||||
|
|
||||||
|
if (!((Form_pg_language) GETSTRUCT(langTup))->lanispl) |
||||||
|
{ |
||||||
|
elog(WARN, "Language %s isn't a created procedural language", |
||||||
|
languageName); |
||||||
|
} |
||||||
|
|
||||||
|
/* ----------------
|
||||||
|
* Now scan pg_language and delete the PL tuple |
||||||
|
* ---------------- |
||||||
|
*/ |
||||||
|
rdesc = heap_openr(LanguageRelationName); |
||||||
|
|
||||||
|
ScanKeyEntryInitialize(&scanKeyData, 0, Anum_pg_language_lanname, |
||||||
|
F_NAMEEQ, PointerGetDatum(languageName)); |
||||||
|
|
||||||
|
scanDesc = heap_beginscan(rdesc, 0, NowTimeQual, 1, &scanKeyData); |
||||||
|
|
||||||
|
tup = heap_getnext(scanDesc, 0, (Buffer *) NULL); |
||||||
|
|
||||||
|
if (!HeapTupleIsValid(tup)) |
||||||
|
{ |
||||||
|
elog(WARN, "Language with name '%s' not found", languageName); |
||||||
|
} |
||||||
|
|
||||||
|
heap_delete(rdesc, &(tup->t_ctid)); |
||||||
|
|
||||||
|
heap_endscan(scanDesc); |
||||||
|
heap_close(rdesc); |
||||||
|
} |
||||||
@ -0,0 +1,17 @@ |
|||||||
|
/*-------------------------------------------------------------------------
|
||||||
|
* |
||||||
|
* proclang.h-- |
||||||
|
* prototypes for proclang.c. |
||||||
|
* |
||||||
|
* |
||||||
|
*------------------------------------------------------------------------- |
||||||
|
*/ |
||||||
|
#ifndef PROCLANG_H |
||||||
|
#define PROCLANG_H |
||||||
|
|
||||||
|
#include <nodes/parsenodes.h> |
||||||
|
|
||||||
|
extern void CreateProceduralLanguage(CreatePLangStmt * stmt); |
||||||
|
extern void DropProceduralLanguage(DropPLangStmt * stmt); |
||||||
|
|
||||||
|
#endif /* PROCLANG_H */ |
||||||
Loading…
Reference in new issue