static void defineFileFields(HttpQueue *q, Upload *up)
{
HttpConn *conn;
HttpUploadFile *file;
char *key;
conn = q->conn;
if (conn->tx->handler == conn->http->ejsHandler) {
/*
Ejscript manages this for itself
*/
return;
}
up = q->queueData;
file = up->currentFile;
key = sjoin("FILE_CLIENT_FILENAME_", up->name, NULL);
httpSetParam(conn, key, file->clientFilename);
key = sjoin("FILE_CONTENT_TYPE_", up->name, NULL);
httpSetParam(conn, key, file->contentType);
key = sjoin("FILE_FILENAME_", up->name, NULL);
httpSetParam(conn, key, file->filename);
key = sjoin("FILE_SIZE_", up->name, NULL);
httpSetIntParam(conn, key, (int) file->size);
}
/*
Initialize the upload filter for a new request
*/
static int openUpload(HttpQueue *q)
{
HttpConn *conn;
HttpRx *rx;
Upload *up;
cchar *uploadDir;
char *boundary;
conn = q->conn;
rx = conn->rx;
if ((up = mprAllocObj(Upload, manageUpload)) == 0) {
return MPR_ERR_MEMORY;
}
q->queueData = up;
up->contentState = HTTP_UPLOAD_BOUNDARY;
rx->autoDelete = rx->route->autoDelete;
rx->renameUploads = rx->route->renameUploads;
uploadDir = getUploadDir(rx->route);
httpSetParam(conn, "UPLOAD_DIR", uploadDir);
if ((boundary = strstr(rx->mimeType, "boundary=")) != 0) {
boundary += 9;
up->boundary = sjoin("--", boundary, NULL);
up->boundaryLen = strlen(up->boundary);
}
if (up->boundaryLen == 0 || *up->boundary == '\0') {
httpError(conn, HTTP_CODE_BAD_REQUEST, "Bad boundary");
return MPR_ERR_BAD_ARGS;
}
return 0;
}
/*
Process the content data.
Returns < 0 on error
== 0 when more data is needed
== 1 when data successfully written
*/
static int processUploadData(HttpQueue *q)
{
HttpConn *conn;
HttpPacket *packet;
MprBuf *content;
Upload *up;
ssize size, dataLen;
bool pureData;
char *data, *bp, *key;
conn = q->conn;
up = q->queueData;
content = q->first->content;
packet = 0;
size = mprGetBufLength(content);
if (size < up->boundaryLen) {
/* Incomplete boundary. Return and get more data */
return 0;
}
bp = getBoundary(mprGetBufStart(content), size, up->boundary, up->boundaryLen, &pureData);
if (bp == 0) {
if (up->clientFilename) {
/*
No signature found yet. probably more data to come. Must handle split boundaries.
*/
data = mprGetBufStart(content);
dataLen = pureData ? size : (size - (up->boundaryLen - 1));
if (dataLen > 0) {
if (writeToFile(q, mprGetBufStart(content), dataLen) < 0) {
return MPR_ERR_CANT_WRITE;
}
}
mprAdjustBufStart(content, dataLen);
return 0; /* Get more data */
}
}
data = mprGetBufStart(content);
dataLen = (bp) ? (bp - data) : mprGetBufLength(content);
if (dataLen > 0) {
mprAdjustBufStart(content, dataLen);
/*
This is the CRLF before the boundary
*/
if (dataLen >= 2 && data[dataLen - 2] == '\r' && data[dataLen - 1] == '\n') {
dataLen -= 2;
}
if (up->clientFilename) {
/*
Write the last bit of file data and add to the list of files and define environment variables
*/
if (writeToFile(q, data, dataLen) < 0) {
return MPR_ERR_CANT_WRITE;
}
defineFileFields(q, up);
} else {
/*
Normal string form data variables
*/
data[dataLen] = '\0';
#if KEEP
httpTrace(conn, "request.upload.variables", "context", "'%s':'%s'", up->name, data);
#endif
key = mprUriDecode(up->name);
data = mprUriDecode(data);
httpSetParam(conn, key, data);
if (packet == 0) {
packet = httpCreatePacket(ME_MAX_BUFFER);
}
if (httpGetPacketLength(packet) > 0) {
/*
Need to add www-form-urlencoding separators
*/
mprPutCharToBuf(packet->content, '&');
} else {
conn->rx->mimeType = sclone("application/x-www-form-urlencoded");
}
mprPutToBuf(packet->content, "%s=%s", up->name, data);
}
}
if (up->clientFilename) {
/*
Now have all the data (we've seen the boundary)
*/
mprCloseFile(up->file);
up->file = 0;
up->clientFilename = 0;
}
if (packet) {
httpPutPacketToNext(q, packet);
}
up->contentState = HTTP_UPLOAD_BOUNDARY;
return 0;
}
void espSetParam(HttpConn *conn, cchar *var, cchar *value)
{
httpSetParam(conn, var, value);
}
PUBLIC void espAddParam(HttpConn *conn, cchar *var, cchar *value)
{
if (!httpGetParam(conn, var, 0)) {
httpSetParam(conn, var, value);
}
}
